beaker 0.0.0 → 1.0.0

data/spec/beaker/shared/repetition_spec.rb

@@ -0,0 +1,72 @@
+require 'spec_helper'
+
+module Beaker
+  module Shared
+    describe Repetition do
+
+      context 'repeat_for' do
+        it "repeats a block for 5 seconds" do
+          Time.stub( :now ).and_return( 0, 1, 2, 3, 4, 5, 6 )
+
+          block = double( 'block' )
+          block.should_receive( :exec ).exactly( 5 ).times.and_return( false )
+          
+          subject.repeat_for( 5 ) do
+            block.exec
+          end
+        end
+
+        it "should short circuit if the block is complete" do
+          Time.stub( :now ).and_return( 0, 1, 2, 3, 4, 5 )
+
+          block = double( 'block' )
+          block.should_receive( :exec ).once.and_return( true )
+          
+          subject.repeat_for( 5 ) do
+            block.exec
+          end
+
+        end
+
+      end
+
+      context 'repeat_fibonacci_style_for' do
+        it "sleeps in fibonacci increasing intervals" do
+
+          block = double( 'block' )
+          block.should_receive( :exec ).exactly( 5 ).times.and_return( false )
+          subject.stub( 'sleep' ).and_return( true )
+          subject.should_receive( :sleep ).with( 1 ).exactly( 2 ).times
+          subject.should_receive( :sleep ).with( 2 ).once
+          subject.should_receive( :sleep ).with( 3 ).once
+          subject.should_receive( :sleep ).with( 5 ).once
+          subject.should_receive( :sleep ).with( 8 ).never
+
+          subject.repeat_fibonacci_style_for( 5 ) do
+            block.exec
+          end
+
+        end
+
+        it "should short circuit if the block is complete" do
+
+          block = double( 'block' )
+          block.should_receive( :exec ).once.and_return( true )
+          subject.stub( 'sleep' ).and_return( true )
+          subject.should_receive( :sleep ).with( 1 ).once
+          subject.should_receive( :sleep ).with( 2 ).never
+          subject.should_receive( :sleep ).with( 3 ).never
+          subject.should_receive( :sleep ).with( 5 ).never
+          subject.should_receive( :sleep ).with( 8 ).never
+
+          subject.repeat_fibonacci_style_for( 5 ) do
+            block.exec
+          end
+
+        end
+      end
+
+    end
+
+  end
+end

data/spec/beaker/test_suite_spec.rb

@@ -7,38 +7,25 @@ module Beaker
     context 'new', :use_fakefs => true do
       let(:test_dir) { 'tmp/tests' }
 
-      let(:options)  { {:tests => create_files(@files)} }
+      let(:options)  { {'name' => create_files(@files)} }
       let(:rb_test)  { File.expand_path(test_dir + '/my_ruby_file.rb')    }
       let(:pl_test)  { File.expand_path(test_dir + '/my_perl_file.pl')    }
       let(:sh_test)  { File.expand_path(test_dir + '/my_shell_file.sh')   }
 
       it 'fails without test files' do
         expect { Beaker::TestSuite.new 'name', 'hosts',
-                  Hash.new, 'config', :stop_on_error }.to raise_error
+                  Hash.new, :stop_on_error }.to raise_error
       end
 
       it 'includes specific files as test file when explicitly passed' do
         @files = [ rb_test ]
         ts = Beaker::TestSuite.new 'name', 'hosts', options,
-                                             'config', :stop_on_error
+                                             :stop_on_error
 
         expect { ts.instance_variable_get(:@test_files).
                   include? rb_test }.to be_true
       end
 
-      it 'includes only .rb files as test files when dir is passed' do
-        create_files [ rb_test, pl_test, sh_test ]
-        @files = [ test_dir ]
-
-        ts = Beaker::TestSuite.new 'name', 'hosts',
-               options, 'config', :stop_on_error
-
-        processed_files = ts.instance_variable_get :@test_files
-
-        expect(processed_files).to include(rb_test)
-        expect(processed_files).to_not include(sh_test)
-        expect(processed_files).to_not include(pl_test)
-      end
     end
   end
 end

data/spec/beaker/utils/ntp_control_spec.rb

@@ -0,0 +1,42 @@
+require 'spec_helper'
+
+module Beaker
+  module Utils
+    describe NTPControl do
+      let( :ntpserver )   { Beaker::Utils::NTPControl::NTPSERVER }
+      let( :ntp_control ) { Beaker::Utils::NTPControl.new( make_opts, @hosts) }
+
+      it "can sync time on unix hosts" do
+        @hosts = make_hosts( { :platform => 'unix' } )
+
+        Command.should_receive( :new ).with("ntpdate -t 20 #{ntpserver}").exactly( 3 ).times
+
+        ntp_control.timesync
+
+      end
+
+      it "can sync time on solaris-10 hosts" do
+        @hosts = make_hosts( { :platform => 'solaris-10' } )
+
+        Command.should_receive( :new ).with("sleep 10 && ntpdate -w #{ntpserver}").exactly( 3 ).times
+
+        ntp_control.timesync
+
+      end
+
+      it "can sync time on windows hosts" do
+        @hosts = make_hosts( { :platform => 'windows' } )
+
+        Command.should_receive( :new ).with("w32tm /register").exactly( 3 ).times
+        Command.should_receive( :new ).with("net start w32time").exactly( 3 ).times
+        Command.should_receive( :new ).with("w32tm /config /manualpeerlist:#{ntpserver} /syncfromflags:manual /update").exactly( 3 ).times
+        Command.should_receive( :new ).with("w32tm /resync").exactly( 3 ).times
+
+        ntp_control.timesync
+
+      end
+
+    end
+
+  end
+end

data/spec/beaker/utils/repo_control_spec.rb

@@ -0,0 +1,168 @@
+require 'spec_helper'
+
+module Beaker
+  module Utils
+    describe RepoControl do
+      let( :apt_cfg )      { Beaker::Utils::RepoControl::APT_CFG }
+      let( :ips_pkg_repo ) { Beaker::Utils::RepoControl::IPS_PKG_REPO }
+      let( :repo_control ) { Beaker::Utils::RepoControl.new( make_opts, @hosts) }
+
+      context "epel_info_for!" do
+        
+        it "can return the correct url for an el-6 host" do
+          host = make_host( 'testhost', { :platform => 'el-6-platform' } )
+
+          expect( repo_control.epel_info_for!( host )).to be === "http://mirror.itc.virginia.edu/fedora-epel/6/i386/epel-release-6-8.noarch.rpm"
+        end
+
+        it "can return the correct url for an el-5 host" do
+          host = make_host( 'testhost', { :platform => 'el-5-platform' } )
+
+          expect( repo_control.epel_info_for!( host )).to be === "http://archive.linux.duke.edu/pub/epel/5/i386/epel-release-5-4.noarch.rpm"
+
+        end
+
+        it "raises an error on non el-5/6 host" do
+          host = make_host( 'testhost', { :platform => 'el-4-platform' } )
+
+          expect{ repo_control.epel_info_for!( host )}.to raise_error
+
+        end
+
+      end
+
+      context "apt_get_update" do
+
+        it "can perform apt-get on ubuntu hosts" do
+          host = make_host( 'testhost', { :platform => 'ubuntu' } )
+
+          Command.should_receive( :new ).with("apt-get -y -f -m update").once
+
+          repo_control.apt_get_update( host )
+
+        end
+
+        it "can perform apt-get on debian hosts" do
+          host = make_host( 'testhost', { :platform => 'debian' } )
+
+          Command.should_receive( :new ).with("apt-get -y -f -m update").once
+
+          repo_control.apt_get_update( host )
+
+        end
+
+        it "does nothing on non debian/ubuntu hosts" do
+          host = make_host( 'testhost', { :platform => 'windows' } )
+
+          Command.should_receive( :new ).never
+
+          repo_control.apt_get_update( host )
+
+        end
+
+      end
+
+      context "copy_file_to_remote" do
+
+        it "can copy a file to a remote host" do
+          content = "this is the content"
+          tempfilepath = "/path/to/tempfile"
+          filepath = "/path/to/file"
+          host = make_host( 'testhost', { :platform => 'windows' })
+          tempfile = double( 'tempfile' )
+          tempfile.stub( :path ).and_return( tempfilepath )
+          Tempfile.stub( :open ).and_yield( tempfile )
+          file = double( 'file' )
+          File.stub( :open ).and_yield( file )
+
+          file.should_receive( :puts ).with( content ).once
+          host.should_receive( :do_scp_to ).with( tempfilepath, filepath, repo_control.instance_variable_get( :@options ) ).once
+
+          repo_control.copy_file_to_remote(host, filepath, content)
+
+        end
+
+
+      end
+
+      context "proxy_config" do
+        
+        it "correctly configures ubuntu hosts" do
+          @hosts = make_hosts( { :platform => 'ubuntu', :exit_code => 1 } )
+
+          Command.should_receive( :new ).with( "if test -f /etc/apt/apt.conf; then mv /etc/apt/apt.conf /etc/apt/apt.conf.bk; fi" ).exactly( 3 )
+          @hosts.each do |host|
+            repo_control.should_receive( :copy_file_to_remote ).with( host, '/etc/apt/apt.conf', apt_cfg ).once
+            repo_control.should_receive( :apt_get_update ).with( host ).once
+          end
+
+          repo_control.proxy_config
+
+        end
+
+        it "correctly configures debian hosts" do
+          @hosts = make_hosts( { :platform => 'debian' } )
+
+          Command.should_receive( :new ).with( "if test -f /etc/apt/apt.conf; then mv /etc/apt/apt.conf /etc/apt/apt.conf.bk; fi" ).exactly( 3 ).times
+          @hosts.each do |host|
+            repo_control.should_receive( :copy_file_to_remote ).with( host, '/etc/apt/apt.conf', apt_cfg ).once
+            repo_control.should_receive( :apt_get_update ).with( host ).once
+          end
+
+          repo_control.proxy_config
+
+        end
+
+        it "correctly configures solaris-11 hosts" do
+          @hosts = make_hosts( { :platform => 'solaris-11' } )
+
+          Command.should_receive( :new ).with( "/usr/bin/pkg unset-publisher solaris || :" ).exactly( 3 ).times
+          @hosts.each do |host|
+            Command.should_receive( :new ).with( "/usr/bin/pkg set-publisher -g %s solaris" % ips_pkg_repo ).once
+          end
+
+          repo_control.proxy_config
+
+        end
+
+        it "does nothing for non ubuntu/debian/solaris-11 hosts" do
+          @hosts = make_hosts( { :platform => 'windows' } )
+          
+          Command.should_receive( :new ).never
+
+          repo_control.proxy_config
+
+        end
+      end
+
+      context "add_el_extras" do
+
+        it "add extras for el-5/6 hosts" do
+          @hosts = make_hosts( { :platform => 'el-5', :exit_code => 1 } )
+          @hosts[0][:platform] = 'el-6' 
+          url = "http://el_extras_url"
+
+          repo_control.stub( :epel_info_for! ).and_return( url )
+
+          Command.should_receive( :new ).with("rpm -qa | grep epel-release").exactly( 3 ).times
+          Command.should_receive( :new ).with("rpm -i #{url}").exactly( 3 ).times
+          Command.should_receive( :new ).with("yum clean all && yum makecache").exactly( 3 ).times
+
+          repo_control.add_el_extras
+
+        end
+
+        it "should do nothing for non el-5/6 hosts" do
+          @hosts = make_hosts( { :platform => 'windows' } )
+
+          Command.should_receive( :new ).never
+
+          repo_control.add_el_extras
+
+        end
+      end
+
+    end
+
+  end
+end

data/spec/beaker/utils/setup_helper_spec.rb

@@ -0,0 +1,82 @@
+require 'spec_helper'
+
+module Beaker
+  module Utils
+    describe SetupHelper do
+      let( :setup_helper ) { Beaker::Utils::SetupHelper.new( make_opts, hosts) }
+      let( :sync_cmd )     { Beaker::Utils::SetupHelper::ROOT_KEYS_SYNC_CMD }
+      let( :platform )     { @platform || 'unix' }
+      let( :ip )           { "ip.address.0.0" }
+      let( :stdout)        { @stdout || ip }
+      let( :hosts )        { hosts = make_hosts( { :stdout => stdout, :platform => platform } )
+                             hosts[0][:roles] = ['agent']
+                             hosts[1][:roles] = ['master', 'dashboard', 'agent', 'database']
+                             hosts[2][:roles] = ['agent']
+                             hosts }
+
+      context "add_master_entry" do
+        
+        it "can configure /etc/hosts on a unix master" do
+          path = Beaker::Utils::SetupHelper::ETC_HOSTS_PATH
+          master = setup_helper.only_host_with_role(hosts, :master)
+
+          Command.should_receive( :new ).with( "ip a|awk '/g/{print$2}' | cut -d/ -f1 | head -1" ).once
+          Command.should_receive( :new ).with( "cp %s %s.old" % [path, path] ).once
+          Command.should_receive( :new ).with( "cp %s %s.new" % [path, path] ).once
+          Command.should_receive( :new ).with( "grep -v '#{ip} #{master}' %s > %s.new" % [path, path] ).once
+          Command.should_receive( :new ).with( "echo '#{ip} #{master}' >> %s.new" % path ).once
+          Command.should_receive( :new ).with( "mv %s.new %s" % [path, path] ).once
+
+          setup_helper.add_master_entry
+        end
+
+        it "can configure /etc/hosts on a solaris master" do
+          @platform = 'solaris'
+          path = Beaker::Utils::SetupHelper::ETC_HOSTS_PATH_SOLARIS
+          master = setup_helper.only_host_with_role(hosts, :master)
+
+          Command.should_receive( :new ).with( "ifconfig -a inet| awk '/broadcast/ {print $2}' | cut -d/ -f1 | head -1" ).once
+          Command.should_receive( :new ).with( "cp %s %s.old" % [path, path] ).once
+          Command.should_receive( :new ).with( "cp %s %s.new" % [path, path] ).once
+          Command.should_receive( :new ).with( "grep -v '#{ip} #{master}' %s > %s.new" % [path, path] ).once
+          Command.should_receive( :new ).with( "echo '#{ip} #{master}' >> %s.new" % path ).once
+          Command.should_receive( :new ).with( "mv %s.new %s" % [path, path] ).once
+
+          setup_helper.add_master_entry
+        end
+
+        it "does nothing on a vagrant master" do
+          master = setup_helper.only_host_with_role(hosts, :master)
+          master[:hypervisor] = 'vagrant'
+
+          Command.should_receive( :new ).never
+          
+          setup_helper.add_master_entry
+
+        end
+      end
+
+      context "sync_root_keys" do
+
+        it "can sync keys on a solaris host" do
+          @platform = 'solaris'
+
+          Command.should_receive( :new ).with( sync_cmd % "bash" ).exactly( 3 ).times
+
+          setup_helper.sync_root_keys
+
+        end
+
+        it "can sync keys on a non-solaris host" do
+
+          Command.should_receive( :new ).with( sync_cmd % "env PATH=/usr/gnu/bin:$PATH bash" ).exactly( 3 ).times
+
+          setup_helper.sync_root_keys
+
+        end
+
+      end
+
+    end
+  end
+end

data/spec/beaker/utils/validator_spec.rb

@@ -0,0 +1,58 @@
+require 'spec_helper'
+
+module Beaker
+  module Utils
+    describe Validator do
+      let( :validator )      { Beaker::Utils::Validator }
+      let( :pkgs )           { Beaker::Utils::Validator::PACKAGES }
+      let( :unix_only_pkgs ) { Beaker::Utils::Validator::UNIX_PACKAGES }
+      let( :platform )       { @platform || 'unix' }
+      let( :hosts )          { hosts = make_hosts( { :platform => platform } )
+                               hosts[0][:roles] = ['agent']
+                               hosts[1][:roles] = ['master', 'dashboard', 'agent', 'database']
+                               hosts[2][:roles] = ['agent']
+                               hosts }
+
+      context "can validate the SUTs" do
+
+        it "can validate unix hosts" do
+
+          hosts.each do |host|
+            pkgs.each do |pkg|
+              host.should_receive( :check_for_package ).with( pkg ).once.and_return( false )
+              host.should_receive( :install_package ).with( pkg ).once
+            end
+            unix_only_pkgs.each do |pkg|
+              host.should_receive( :check_for_package ).with( pkg ).once.and_return( false )
+              host.should_receive( :install_package ).with( pkg ).once
+            end
+            
+          end
+
+          validator.validate(hosts, logger)
+
+        end
+
+        it "can validate windows hosts" do
+          @platform = 'windows'
+
+          hosts.each do |host|
+            pkgs.each do |pkg|
+              host.should_receive( :check_for_package ).with( pkg ).once.and_return( false )
+              host.should_receive( :install_package ).with( pkg ).once
+            end
+            unix_only_pkgs.each do |pkg|
+              host.should_receive( :check_for_package).with( pkg ).never
+              host.should_receive( :install_package ).with( pkg ).never
+            end
+          end
+
+          validator.validate(hosts, logger)
+
+        end
+      end
+
+
+    end
+  end
+end