moose-inventory 0.1.0

data/spec/lib/moose/inventory/cli/group_rmvar_spec.rb

@@ -0,0 +1,202 @@
+require 'spec_helper'
+
+# TODO: the usual respond_to? method doesn't seem to work on Thor objects.
+# Why not? For now, we'll check against instance_methods.
+
+RSpec.describe Moose::Inventory::Cli::Group do
+  before(:all) do
+    # Set up the configuration object
+    @mockarg_parts = {
+      config:  File.join(spec_root, 'config/config.yml'),
+      format:  'yaml',
+      env:     'test'
+    }
+
+    @mockargs = []
+    @mockarg_parts.each do |key, val|
+      @mockargs << "--#{key}"
+      @mockargs << val
+    end
+
+    @config = Moose::Inventory::Config
+    @config.init(@mockargs)
+
+    @db = Moose::Inventory::DB
+    @db.init if @db.db.nil?
+
+    @console = Moose::Inventory::Cli::Formatter
+    @group = Moose::Inventory::Cli::Group
+    @app = Moose::Inventory::Cli::Application
+  end
+
+  before(:each) do
+    @db.reset
+  end
+
+  describe 'rmvar' do
+    it 'should be responsive' do
+      result = @group.instance_methods(false).include?(:rmvar)
+      expect(result).to eq(true)
+    end
+    
+    #-----------------
+     it '<missing args> ... should abort with an error' do
+       actual = runner  do
+         @app.start(%w(group rmvar))
+       end
+ 
+       # Check output
+       desired = { aborted: true}
+       desired[:STDERR] = "ERROR: Wrong number of arguments, 0 for 2 or more.\n"
+       expected(actual, desired)
+     end
+ 
+     #------------------------
+     it 'GROUP key=value ... should abort if the group does not exist' do
+       group_name = "does-not-exist"
+       var_name = "foo=bar"
+       actual = runner do
+         @app.start(%W(group rmvar #{group_name} #{var_name}))
+       end
+ 
+       # Check output
+       desired = { aborted: true}
+       desired[:STDOUT] =
+         "Remove variable(s) '#{var_name}' from group '#{group_name}':\n"\
+         "  - retrieve group '#{group_name}'...\n"
+       desired[:STDERR] = 
+         "An error occurred during a transaction, any changes have been rolled back.\n"\
+         "ERROR: The group '#{group_name}' does not exist.\n"
+       expected(actual, desired)
+     end
+ 
+     #------------------------
+     it '<malformed> ... should abort with an error' do
+        # 1. Should add the var to the db
+        # 2. Should associate the host with the var
+        
+        group_name = 'test1'
+        @db.models[:group].create(name: group_name)
+  
+        var  = {name: 'foo', value: "bar"}
+        cases = %W(
+          =bar 
+          foo=bar=
+          =foo=bar
+          foo=bar=extra
+        )      
+
+        cases.each do |args| 
+          actual = runner do
+            @app.start(%W(group rmvar #{group_name} #{args} ))
+          end
+          #@console.out(actual,'p')
+    
+          desired = { aborted: true}
+          desired[:STDOUT] = 
+            "Remove variable(s) '#{args}' from group '#{group_name}':\n"\
+            "  - retrieve group '#{group_name}'...\n"\
+            "    - OK\n"\
+            "  - remove variable '#{args}'...\n"
+          desired[:STDERR] = 
+            "An error occurred during a transaction, any changes have been rolled back.\n"\
+            "ERROR: Incorrect format in {#{args}}. Expected 'key' or 'key=value'.\n"
+            
+          expected(actual, desired)
+        end
+      end
+     
+    #------------------------
+    it 'GROUP <valid args> ... should remove the group variable' do
+       group_name = 'group_test'
+       var  = {name: 'foo', value: "bar"}
+       cases = %W(
+         #{var[:name]}
+         #{var[:name]}=
+         #{var[:name]}=#{var[:value]}
+       )
+       cases.each do |example|
+         # reset the db
+         @db.reset
+         
+         # Add an initial group and groupvar
+         @db.models[:group].create(name: group_name)
+         runner do
+           @app.start(%W(group addvar #{group_name} #{var[:name]}=#{var[:value]} ))
+         end
+         
+         # Try to remove the groupvar using the case example valid args 
+         actual = runner do
+           @app.start(%W(group rmvar #{group_name} #{example}))
+         end
+         #@console.out(actual,'p')
+         
+         # Check the output
+         desired = { aborted: false}
+         desired[:STDOUT] =
+           "Remove variable(s) '#{example}' from group '#{group_name}':\n"\
+           "  - retrieve group '#{group_name}'...\n"\
+           "    - OK\n"\
+           "  - remove variable '#{example}'...\n"\
+           "    - OK\n"\
+           "  - all OK\n"\
+           "Succeeded.\n"
+           
+          #@console.out(desired,'p')  
+         expected(actual, desired)
+   
+         # Check the db
+         group = @db.models[:group].find(name: group_name)
+         groupvars = group.groupvars_dataset
+         expect(groupvars.count).to eq(0)
+         
+         groupvars = @db.models[:groupvar].all
+         expect(groupvars.count).to eq(0)
+       end
+    end
+    
+    #------------------------
+    it 'GROUP key1=value1 key2=value2 ... should remove multiple key/value pairs' do
+      group_name = 'test_group'
+      varsarray  = [
+        {name: 'var1', value: "val1"},
+        {name: 'var2', value: "val2"}
+      ]
+          
+      vars = []
+      varsarray.each do |var|
+        vars << "#{var[:name]}=#{var[:value]}"
+      end
+        
+      @db.models[:group].create(name: group_name)
+      actual = runner do
+        @app.start(%W(group addvar #{group_name}) + vars )
+      end
+      
+      actual = runner do
+        @app.start(%W(group rmvar #{group_name}) + vars )
+      end
+      #@console.out(actual,'y')
+      
+      desired = { aborted: false}
+      desired[:STDOUT] = 
+        "Remove variable(s) '#{vars.join(',')}' from group '#{group_name}':\n"\
+        "  - retrieve group '#{group_name}'...\n"\
+        "    - OK\n"
+      vars.each do |var|
+        desired[:STDOUT] =  desired[:STDOUT] + 
+          "  - remove variable '#{var}'...\n"\
+          "    - OK\n"
+      end
+      desired[:STDOUT] =  desired[:STDOUT] + 
+        "  - all OK\n"\
+        "Succeeded.\n"
+      expected(actual, desired)
+    
+      # We should have the correct hostvar associations
+      group = @db.models[:group].find(name: group_name)
+      groupvars = group.groupvars_dataset
+      expect(groupvars.count).to eq(0)
+    end        
+  end
+end

data/spec/lib/moose/inventory/cli/group_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+require_relative 'group_add_spec'
+#require_relative 'group_get_spec'
+#require_relative 'group_list_spec'
+#require_relative 'group_rm_spec'
+#
+#require_relative 'group_addchild_spec'
+#require_relative 'grpi_rmchild_spec'
+#
+#require_relative 'group_addhost_spec'
+#require_relative 'group_rmhost_spec'
+#
+#require_relative 'group_addvar_spec'
+#require_relative 'group_rmvar_spec'

data/spec/lib/moose/inventory/cli/host_add_spec.rb

@@ -0,0 +1,330 @@
+require 'spec_helper'
+
+# TODO: the usual respond_to? method doesn't seem to work on Thor objects.
+# Why not? For now, we'll check against instance_methods.
+
+RSpec.describe Moose::Inventory::Cli::Host do
+  before(:all) do
+    # Set up the configuration object
+    @mockarg_parts = {
+      config:  File.join(spec_root, 'config/config.yml'),
+      format:  'yaml',
+      env:     'test'
+    }
+
+    @mockargs = []
+    @mockarg_parts.each do |key, val|
+      @mockargs << "--#{key}"
+      @mockargs << val
+    end
+
+    @console = Moose::Inventory::Cli::Formatter
+    @config = Moose::Inventory::Config
+    @config.init(@mockargs)
+
+    @db = Moose::Inventory::DB
+    @db.init if @db.db.nil?
+
+    @host = Moose::Inventory::Cli::Host
+    @app = Moose::Inventory::Cli::Application
+  end
+
+  before(:each) do
+    @db.reset
+  end
+
+  # ============================
+  describe 'add' do
+    # --------------------
+    it 'Host.add() method should be responsive' do
+      result = @host.instance_methods(false).include?(:add)
+      expect(result).to eq(true)
+    end
+
+    # --------------------
+    it '<no arguments> ... should bail with an error' do
+      actual = runner { @app.start(%w(host add)) }
+ 
+      desired = { aborted: true}
+      desired[:STDERR] = "ERROR: Wrong number of arguments, 0 for 1 or more.\n"
+      expected(actual, desired)
+    end
+
+    # --------------------
+    it 'HOST ... should add a host to the db' do
+      name = 'test-host-add'
+
+      actual = runner { @app.start(%W(host add #{name})) }
+      #@console.out(actual,'y')
+      
+      # Check output
+      desired = {}
+      desired[:STDOUT] =
+        "Add host '#{name}':\n"\
+        "  - Creating host '#{name}'...\n"\
+        "    - OK\n"\
+        "  - Adding automatic association {host:#{name} <-> group:ungrouped}...\n"\
+        "    - OK\n"\
+        "  - All OK\n"\
+        "Succeeded\n"
+
+      expected(actual, desired)
+
+      # Check db
+      host = @db.models[:host].find(name: name)
+      expect(host[:name]).to eq(name)
+
+      groups = host.groups_dataset
+      expect(groups.count).to eq(1)
+      expect(groups[name: 'ungrouped']).not_to be_nil
+    end
+
+    # --------------------
+    it 'HOST ... should skip HOST creation if HOST already exists' do
+      name = 'test-host-add'
+      runner { @app.start(%W(host add #{name})) }
+
+      actual = runner { @app.start(%W(host add #{name})) }
+
+      # Check output
+      desired = {}
+      desired[:STDOUT] = 
+        "Add host '#{name}':\n"\
+        "  - Creating host '#{name}'...\n"\
+        "    - OK\n"\
+        "  - All OK\n"\
+        "Succeeded\n"
+          
+      desired[:STDERR] =
+        "WARNING: The host '#{name}' already exists, skipping creation.\n"
+
+      expected(actual, desired)
+    end
+
+    # --------------------
+    it 'HOST1 HOST2 HOST3 ... should add multiple hosts' do
+      names = %w(test1 test2 test3)
+
+      actual = runner { @app.start(%w(host add) + names) }
+
+      # Check output
+      desired = {STDOUT: ''}
+      names.each do |name|
+        desired[:STDOUT] =  desired[:STDOUT] +
+          "Add host '#{name}':\n"\
+          "  - Creating host '#{name}'...\n"\
+          "    - OK\n"\
+          "  - Adding automatic association {host:#{name} <-> group:ungrouped}...\n"\
+          "    - OK\n"\
+          "  - All OK\n"
+      end
+      desired[:STDOUT] = desired[:STDOUT] + "Succeeded\n"
+
+      expected(actual, desired)
+
+      # Check db
+      names.each do |name|
+        host = @db.models[:host].find(name: name)
+        expect(host[:name]).to eq(name)
+
+        groups = host.groups_dataset
+        expect(groups.count).to eq(1)
+        expect(groups[name: 'ungrouped']).not_to be_nil
+      end
+    end
+
+    # --------------------
+    it 'HOST1 --groups GROUP ... should add the '\
+      'host and associate it with an existing group' do
+      group_name = 'testgroup'
+      @db.models[:group].create(name: group_name)
+
+      name = 'testhost'
+      actual = runner do
+        @app.start(%W(host add #{name} --groups #{group_name}))
+      end
+
+      # Check output
+      desired = {}
+      desired[:STDOUT] =
+        "Add host '#{name}':\n"\
+        "  - Creating host '#{name}'...\n"\
+        "    - OK\n"\
+        "  - Adding association {host:#{name} <-> group:#{group_name}}...\n"\
+        "    - OK\n"\
+        "  - All OK\n"\
+        "Succeeded\n"
+
+      expected(actual, desired)
+
+      # Check db
+      host = @db.models[:host].find(name: name)
+      groups = host.groups_dataset
+      expect(groups).not_to be_nil
+      expect(groups.count).to eq(1)
+      expect(groups[name: group_name]).not_to be_nil # i.e. not 'ungrouped'
+    end
+
+    # --------------------
+    it 'HOST1 --groups GROUP ... should add the host and associate '\
+      ' it with a non-existent group, creating the group' do
+      group_name = 'testgroup'
+
+      # DON'T MANUALLY CREATE THE GROUP! That's the point of the test.
+
+      name = 'testhost'
+      actual = runner do
+        @app.start(%W(host add #{name} --groups #{group_name}))
+      end
+
+      # Check output
+      desired = {}
+      desired[:STDOUT] =
+        "Add host '#{name}':\n"\
+        "  - Creating host '#{name}'...\n"\
+        "    - OK\n"\
+        "  - Adding association {host:#{name} <-> group:#{group_name}}...\n"\
+        "    - OK\n"\
+        "  - All OK\n"\
+        "Succeeded\n"
+      desired[:STDERR] =
+        "WARNING: The group '#{group_name}' doesn't exist, but will be created.\n"
+        
+      expected(actual, desired)
+
+      # Check db
+      host = @db.models[:host].find(name: name)
+      groups = host.groups_dataset
+      expect(groups).not_to be_nil
+      expect(groups.count).to eq(1)
+      expect(groups[name: group_name]).not_to be_nil # i.e. not 'ungrouped'
+    end
+
+    # --------------------
+    it 'HOST1 --groups ungrouped ... should abort with an error' do
+      name = 'testhost'
+      actual = runner do
+        @app.start(%W(host add #{name} --groups ungrouped))
+      end
+
+      # Check output
+      desired = { aborted: true}
+      desired[:STDERR] =
+        "ERROR: Cannot manually manipulate the automatic group 'ungrouped'.\n"
+      expected(actual, desired)
+
+      # Check db
+      host = @db.models[:host].find(name: name)
+      expect(host).to be_nil
+    end
+
+    # --------------------
+    it 'HOST --groups GROUP1,GROUP2 ... should add the host and '\
+      'associate it with multiple groups' do
+      group_names = %w(group1 group2 group3)
+
+      name = 'testhost'
+      actual = runner do
+        @app.start(%W(host add #{name} --groups #{group_names.join(',')}))
+      end
+        
+      #@console.out(actual,'y')
+        
+      # Check output
+      desired = {STDOUT: '', STDERR: ''}
+      desired[:STDOUT] = 
+        "Add host '#{name}':\n"\
+        "  - Creating host '#{name}'...\n"\
+        "    - OK\n"\
+        
+      group_names.each do |group|
+        desired[:STDOUT] =  desired[:STDOUT] +
+          "  - Adding association {host:#{name} <-> group:#{group}}...\n"\
+          "    - OK\n"
+        desired[:STDERR] =  desired[:STDERR] +
+          "WARNING: The group '#{group}' doesn't exist, but will be created.\n"
+      end
+      desired[:STDOUT] = desired[:STDOUT] + 
+        "  - All OK\n"\
+        "Succeeded\n"
+
+      expected(actual, desired)
+
+      # Check db
+      host = @db.models[:host].find(name: name)
+      groups = host.groups_dataset
+      expect(groups.count).to eq(group_names.count)
+      expect(groups[name: 'ungrouped']).to be_nil # i.e. not 'ungrouped'
+      group_names.each do |group|
+        expect(groups[name: group]).not_to be_nil
+      end
+    end
+
+    it 'HOST --groups GROUP1,ungrouped ... should bail '\
+      'with an error' do
+      name = 'testhost'
+      group_names = %w(group1 ungrouped)
+
+      actual = runner do
+        @app.start(%W(host add #{name} --groups #{group_names.join(',')}))
+      end
+
+      # Check output
+      desired = { aborted: true}
+      desired[:STDERR] =
+        "ERROR: Cannot manually manipulate the automatic group 'ungrouped'.\n"
+
+      expected(actual, desired)
+    end
+
+    it 'HOST1 HOST2 --groups GROUP1,GROUP2 ... should add multiple '\
+      'hosts, associating each with multiple groups' do
+      #
+      group_names = %w(group1 group2 group3)
+      # Note, relies on auto-generation of groups
+
+      names = %w(host1 host2 host3)
+      actual = runner do
+        @app.start(%w(host add) + names + %W(--groups #{group_names.join(',')}))
+      end
+
+      # Check output
+      desired = { aborted: false, STDERR: '', STDOUT: '' }
+      names.each do |name|
+        desired[:STDOUT] = desired[:STDOUT] + 
+          "Add host '#{name}':\n"\
+          "  - Creating host '#{name}'...\n"\
+          "    - OK\n"
+        group_names.each do |group|
+          desired[:STDOUT] =  desired[:STDOUT] +
+            "  - Adding association {host:#{name} <-> group:#{group}}...\n"\
+            "    - OK\n"
+        end
+        desired[:STDOUT] = desired[:STDOUT] +
+          "  - All OK\n" 
+      end
+      
+      group_names.each do |group|
+        desired[:STDERR] =  desired[:STDERR] +
+          "WARNING: The group '#{group}' doesn't exist, but will be created.\n"
+      end
+      desired[:STDOUT] = desired[:STDOUT] + "Succeeded\n"
+
+      #@console.out(desired,'y')
+      
+      expected(actual, desired)
+
+      # Check db
+      names.each do |name|
+        host = @db.models[:host].find(name: name)
+        expect(host).not_to be_nil
+        groups = host.groups_dataset
+        expect(groups.count).to eq(group_names.count)
+        expect(groups[name: 'ungrouped']).to be_nil # i.e. not 'ungrouped'
+        group_names.each do |group|
+          expect(groups[name: group]).not_to be_nil
+        end
+      end
+    end
+  end
+end