moose-inventory 0.1.0

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

@@ -0,0 +1,206 @@
+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
+
+    @config = Moose::Inventory::Config
+    @config.init(@mockargs)
+
+    @db = Moose::Inventory::DB
+    @db.init if @db.db.nil?
+
+    @console = Moose::Inventory::Cli::Formatter
+    @host = Moose::Inventory::Cli::Host
+    @app = Moose::Inventory::Cli::Application
+  end
+
+  before(:each) do
+    @db.reset
+  end
+
+  describe 'rmvar' do
+    it 'should be responsive' do
+      result = @host.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(host rmvar)) # <- no group given
+       end
+ 
+       # Check output
+       desired = { aborted: true}
+       desired[:STDERR] = "ERROR: Wrong number of arguments, 0 for 2 or more.\n"
+       expected(actual, desired)
+     end
+ 
+     #------------------------
+     it 'HOST key=value ... should abort if the host does not exist' do
+       host_name = "not-a-host"
+       var_name = "foo=bar"
+       actual = runner do
+         @app.start(%W(host rmvar #{host_name} #{var_name}))
+       end
+ 
+       # Check output
+       desired = { aborted: true}
+       desired[:STDOUT] =
+         "Remove variable(s) '#{var_name}' from host '#{host_name}':\n"\
+         "  - retrieve host '#{host_name}'...\n"
+       desired[:STDERR] = 
+         "An error occurred during a transaction, any changes have been rolled back.\n"\
+         "ERROR: The host '#{host_name}' does not exist.\n"
+       expected(actual, desired)
+     end
+ 
+     #------------------------
+     it 'HOST <malformed> ... should abort with an error' do
+        # 1. Should add the var to the db
+        # 2. Should associate the host with the var
+        
+        host_name = 'test1'
+        @db.models[:host].create(name: host_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(host rmvar #{host_name} #{args} ))
+          end
+          #@console.out(actual,'p')
+    
+          desired = { aborted: true}
+          desired[:STDOUT] = 
+            "Remove variable(s) '#{args}' from host '#{host_name}':\n"\
+            "  - retrieve host '#{host_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 'host rmvar HOST <valid args> ... should remove the host variable' do
+       # 1. Should add the var to the db
+       # 2. Should associate the host with the var
+       
+       host_name = 'test1'
+
+       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 host and hostvar
+         @db.models[:host].create(name: host_name)
+         runner do
+           @app.start(%W(host addvar #{host_name} #{var[:name]}=#{var[:value]} ))
+         end
+         
+         # Try to remove the hostvar using the case example valid args 
+         actual = runner do
+           @app.start(%W(host rmvar #{host_name} #{example}))
+         end
+         #@console.out(actual,'p')
+         
+         # Check the output
+         desired = { aborted: false}
+         desired[:STDOUT] =
+           "Remove variable(s) '#{example}' from host '#{host_name}':\n"\
+           "  - retrieve host '#{host_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
+         host = @db.models[:host].find(name: host_name)
+         hostvars = host.hostvars_dataset
+         expect(hostvars.count).to eq(0)
+         
+         hostvars = @db.models[:hostvar].all
+         expect(hostvars.count).to eq(0)
+       end
+    end
+    
+    #------------------------
+    it 'HOST key1=value1 key2=value2 ... should remove multiple key/value pairs' do
+      host_name = 'test1'
+      varsarray  = [
+        {name: 'var1', value: "val1"},
+        {name: 'var2', value: "val2"}
+      ]
+          
+      vars = []
+      varsarray.each do |var|
+        vars << "#{var[:name]}=#{var[:value]}"
+      end
+        
+      @db.models[:host].create(name: host_name)
+      actual = runner do
+        @app.start(%W(host addvar #{host_name}) + vars )
+      end
+      
+      actual = runner do
+        @app.start(%W(host rmvar #{host_name}) + vars )
+      end
+      #@console.out(actual,'p')
+      
+      desired = { aborted: false}
+      desired[:STDOUT] = 
+        "Remove variable(s) '#{vars.join(',')}' from host '#{host_name}':\n"\
+        "  - retrieve host '#{host_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
+      host = @db.models[:host].find(name: host_name)
+      hostvars = host.hostvars_dataset
+      expect(hostvars.count).to eq(0)
+    end        
+  end
+end

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

@@ -0,0 +1,12 @@
+require 'spec_helper'
+
+require_relative 'host_add_spec'
+require_relative 'host_get_spec'
+require_relative 'host_list_spec'
+require_relative 'host_rm_spec'
+
+require_relative 'host_addgroup_spec'
+require_relative 'host_rmgroup_spec'
+
+require_relative 'host_addvar_spec'
+require_relative 'host_rmvar_spec'

data/spec/lib/moose/inventory/config/config_spec.rb

@@ -0,0 +1,80 @@
+require 'spec_helper'
+
+RSpec.describe 'Moose::Inventory::Config' 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
+  end
+
+  # .init()
+  describe '.init()' do
+    it 'should be responsive' do
+      result = @config.respond_to?(:init)
+      expect(result).to eq(true)
+    end
+  end
+
+  # ._configopts
+  describe '._configopts' do
+    it 'should be responsive' do
+      result = @config.respond_to?(:_confopts)
+      expect(result).to eq(true)
+    end
+
+    it 'should not be nil' do
+      @config.init(@mockargs)
+      expect(@config._confopts).not_to be_nil
+    end
+
+    it 'should default "--format" to json' do
+      @config.init([])
+      expect(@config._confopts[:format]).to eq('json')
+    end
+
+    it 'should pick up "--format" from its argument list' do
+      @config.init(@mockargs)
+      expect(@config._confopts[:format]).to eq('yaml')
+    end
+
+    it 'should default "--env" to ""' do
+      @config.init([])
+      expect(@config._confopts[:env]).to eq('')
+    end
+
+    it 'should pick up "--env" from its argument list' do
+      tmpargs = ['--env', 'rspectest', '--config', @mockarg_parts[:config]]
+      @config.init(tmpargs)
+      expect(@config._confopts[:env]).to eq('rspectest')
+    end
+  end
+
+  # ._settings
+  describe '._settings' do
+    it 'should be responsive' do
+      result = @config.respond_to?(:_settings)
+      expect(result).to eq(true)
+    end
+
+    it 'should not be nil' do
+      @config.init(@mockargs)
+      expect(@config._confopts).not_to be_nil
+    end
+
+    it 'should pick up "config.db" from the configuration file' do
+      @config.init(@mockargs)
+      expect(@config._settings[:config][:db]).not_to be_nil
+    end
+  end
+end

data/spec/lib/moose/inventory/db/db_spec.rb

@@ -0,0 +1,184 @@
+require 'spec_helper'
+
+RSpec.describe 'Moose::Inventory::DB' do
+  #=============================
+  # Initialization
+  #
+
+  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
+  end
+
+  #=============================
+  # Tests
+  #
+
+  describe '.init()' do
+    it 'should be responsive' do
+      expect(@db.respond_to?(:init)).to eq(true)
+    end
+
+    it 'shouldn\'t throw an error' do
+      failed = false
+      # rubocop:disable Lint/RescueException
+      begin
+        @db.init
+      rescue Exception => e
+        p e
+        failed = true
+      end
+      # rubocop:enable Lint/RescueException
+
+      expect(failed).to eq(false)
+    end
+  end
+
+  describe '.db' do
+    it 'should be responsive' do
+      expect(@db.respond_to?(:db)).to eq(true)
+    end
+
+    it 'should not be nil' do
+      expect(@db.db).not_to be_nil
+    end
+  end
+
+  describe '.models' do
+    it 'should be responsive' do
+      expect(@db.respond_to?(:models)).to eq(true)
+    end
+
+    it 'should not be nil' do
+      expect(@db.models).not_to be_nil
+    end
+  end
+
+  describe '.exceptions' do
+    it 'should be responsive' do
+      expect(@db.respond_to?(:exceptions)).to eq(true)
+    end
+
+    it 'should not be nil' do
+      expect(@db.exceptions).not_to be_nil
+    end
+  end
+
+  describe 'tables' do
+    it 'should have a hosts table' do
+      schema = @db.db.schema(:hosts)
+      expect(schema).not_to be_nil
+    end
+
+    it 'should have a hostsvars table' do
+      schema = @db.db.schema(:hostvars)
+      expect(schema).not_to be_nil
+    end
+    it 'should have a groups table' do
+      schema = @db.db.schema(:groups)
+      expect(schema).not_to be_nil
+    end
+
+    it 'should have a groupvars table' do
+      schema = @db.db.schema(:groupvars)
+      expect(schema).not_to be_nil
+    end
+
+    it 'should have a groups_hosts table' do
+      schema = @db.db.schema(:groups_hosts)
+      expect(schema).not_to be_nil
+    end
+  end
+
+  describe '.reset()' do
+    it 'should be responsive' do
+      result = @db.respond_to?(:reset)
+      expect(result).to eq(true)
+    end
+
+    it 'should purge the database of contents' do
+      # Put at least one host and one group into the database
+      @db.models[:host].create(name: 'reset-host')
+      @db.models[:group].create(name: 'reset-group')
+
+      # Reset the DB
+      @db.reset
+
+      #
+      hosts = @db.models[:host].all
+      expect(hosts.count).to eq(0)
+
+      group = @db.models[:group].all
+      expect(group.count).to eq(0)
+    end
+  end
+
+  describe '.transaction()' do
+    before(:each) do
+      @db.reset
+    end
+
+    it 'should be responsive' do
+      result = @db.respond_to?(:transaction)
+      expect(result).to eq(true)
+    end
+
+    it 'should perform transactions' do
+      hosts = @db.models[:host].all
+      count = { initial: hosts.count, items: 0 }
+
+      @db.transaction  do
+        # @db.transaction('should perform transactions') do
+        (1..3).each do |_n|
+          count[:items] = count[:items] + 1
+          @db.models[:host].create(name: "transaction-#{count[:items]}")
+        end
+      end
+
+      hosts = @db.models[:host].all
+      count[:final] = hosts.count
+
+      # puts "\n#{ JSON.pretty_generate(@db.debug) }"
+      # puts "#{ JSON.pretty_generate(count) }"
+
+      expect(count[:final]).to eq(count[:initial] + count[:items])
+    end
+
+    it 'should roll back failed transactions' do
+      hosts = @db.models[:host].all
+      count = { initial: hosts.count, items: 0 }
+
+      @db.transaction  do
+        # @db.transaction('should roll back failed transactions') do
+        (1..3).each do |_n|
+          count[:items] = count[:items] + 1
+          @db.models[:host].create(name: "rollback-#{count[:items]}")
+        end
+        fail Sequel::Rollback, 'Test error' #
+      end
+
+      hosts = @db.models[:host].all
+      count[:final] = hosts.count
+
+      # puts "\n#{ JSON.pretty_generate(@db.debug) }"
+      # puts "#{ JSON.pretty_generate(count) }"
+
+      expect(count[:final]).to eq(count[:initial])
+    end
+  end
+end