Adding spec tests for node_groups() function

Author: WhatsARanjit

.fixtures.yml

@@ -0,0 +1,4 @@
+---
+fixtures:
+  symlinks:
+  'node_manager': "#{source_dir}"

.rspec

@@ -0,0 +1,2 @@
+--format documentation
+--color

spec/functions/node_groups_spec.rb

@@ -0,0 +1,98 @@
+require 'spec_helper'
+require 'puppet/util/node_groups'
+require 'webmock/rspec'
+
+describe 'node_groups' do
+  unless Puppet.features.puppetclassify?
+    skip "puppetclassify gem not installed"
+  else
+    describe 'input validation' do
+      it { is_expected.to run.with_params('', '', [], '', 'extra').and_raise_error(ArgumentError, /Function accepts a single String/i) }
+      it { is_expected.to run.with_params([]).and_raise_error(ArgumentError, /Function accepts a single String/i) }
+    end
+
+    groups_response = <<-EOS
+    [
+      {
+          "classes": {},
+          "environment": "production",
+          "environment_trumps": false,
+          "id": "00000000-0000-4000-8000-000000000000",
+          "name": "All Nodes",
+          "parent": "00000000-0000-4000-8000-000000000000",
+          "rule": [
+              "and",
+              [
+                  "~",
+                  "name",
+                  ".*"
+              ]
+          ],
+          "variables": {}
+      },
+      {
+          "classes": {},
+          "environment": "production",
+          "environment_trumps": false,
+          "id": "7233f964-951e-4a7f-88ea-72676ed3104d",
+          "name": "Production environment",
+          "parent": "00000000-0000-4000-8000-000000000000",
+          "rule": [
+              "and",
+              [
+                  "~",
+                  "name",
+                  ".*"
+              ]
+           ],
+          "variables": {}
+      }
+    ]
+    EOS
+
+    hashified = {
+      'All Nodes'            => {
+        'classes'            => {},
+        'environment'        => 'production',
+        'environment_trumps' => false,
+        'id'                 => '00000000-0000-4000-8000-000000000000',
+        'name'               => 'All Nodes',
+        'parent'             => '00000000-0000-4000-8000-000000000000',
+        'rule'               => ['and', ['~', 'name', '.*']],
+        'variables'          => {},
+      },
+      'Production environment' => {
+        'classes'              => {},
+        'environment'          => 'production',
+        'environment_trumps'   => false,
+        'id'                   => '7233f964-951e-4a7f-88ea-72676ed3104d',
+        'name'                 => 'Production environment',
+        'parent'               => '00000000-0000-4000-8000-000000000000',
+        'rule'                 => ['and', ['~', 'name', '.*']],
+        'variables'            => {},
+      }
+    }
+
+    before do
+      YAML.stubs(:load_file).returns({
+        'server' => 'stubserver',
+        'port'   => '8080',
+      })
+      stub_request(
+        :get,
+        'https://stubserver:8080/classifier-api/v1/groups',
+      ).to_return(
+        :status => 200,
+        :body   => groups_response
+      )
+    end
+
+    describe 'without an argument' do
+      it { should run.with_params().and_return(hashified) }
+    end
+
+    describe 'with 1 String argument' do
+      it { should run.with_params('All Nodes').and_return({ 'All Nodes' => hashified['All Nodes']}) }
+    end
+  end
+end

spec/spec_helper.rb

@@ -0,0 +1 @@
+require 'puppetlabs_spec_helper/module_spec_helper'