beaker 0.0.0 → 1.0.0

data/spec/beaker/options/parser_spec.rb

@@ -0,0 +1,172 @@
+require "spec_helper"
+
+module Beaker
+  module Options
+
+    describe Parser do
+      let(:parser)    { Parser.new }
+      let(:opts_path) { File.join(File.expand_path(File.dirname(__FILE__)), "data", "opts.txt") }
+      let(:hosts_path)  { File.join(File.expand_path(File.dirname(__FILE__)), "data", "hosts.cfg") }
+      let(:badyaml_path)  { File.join(File.expand_path(File.dirname(__FILE__)), "data", "badyaml.cfg") }
+      let(:home) {ENV['HOME']}
+
+      it "supports usage function" do
+        expect{parser.usage}.to_not raise_error
+      end
+
+      repo = 'git://github.com/puppetlabs'
+
+      it "has repo set to #{repo}" do
+        expect(parser.repo).to be === "#{repo}"
+      end
+
+      #test parse_install_options
+      it "can transform --install PUPPET/3.1 into #{repo}/puppet.git#3.1" do
+        opts = ["PUPPET/3.1"]
+        expect(parser.parse_git_repos(opts)).to be === ["#{repo}/puppet.git#3.1"] 
+      end
+      it "can transform --install FACTER/v.1.0 into #{repo}/facter.git#v.1.0" do
+        opts = ["FACTER/v.1.0"]
+        expect(parser.parse_git_repos(opts)).to be === ["#{repo}/facter.git#v.1.0"] 
+      end
+      it "can transform --install HIERA/xyz into #{repo}/hiera.git#xyz" do
+        opts = ["HIERA/xyz"]
+        expect(parser.parse_git_repos(opts)).to be === ["#{repo}/hiera.git#xyz"] 
+      end
+      it "can transform --install HIERA-PUPPET/path/to/repo into #{repo}/hiera-puppet.git#path/to/repo" do
+        opts = ["HIERA-PUPPET/path/to/repo"]
+        expect(parser.parse_git_repos(opts)).to be === ["#{repo}/hiera-puppet.git#path/to/repo"] 
+      end
+      it "can transform --install PUPPET/3.1,FACTER/v.1.0 into #{repo}/puppet.git#3.1,#{repo}/facter.git#v.1.0" do
+        opts = ["PUPPET/3.1", "FACTER/v.1.0"]
+        expect(parser.parse_git_repos(opts)).to be === ["#{repo}/puppet.git#3.1", "#{repo}/facter.git#v.1.0"] 
+      end
+      it "can leave --install git://github.com/puppetlabs/puppet.git#my/full/path alone" do
+        opts = ["git://github.com/puppetlabs/puppet.git#my/full/path"]
+        expect(parser.parse_git_repos(opts)).to be === ["git://github.com/puppetlabs/puppet.git#my/full/path"]
+      end
+
+      #split_arg testing
+      it "can split comma separated list into an array" do
+        arg = "file1,file2,file3"
+        expect(parser.split_arg(arg)).to be === ["file1", "file2", "file3"]
+      end
+      it "can use an existing Array as an acceptable argument" do
+        arg = ["file1", "file2", "file3"]
+        expect(parser.split_arg(arg)).to be === ["file1", "file2", "file3"]
+      end
+      it "can generate an array from a single value" do
+        arg = "i'mjustastring"
+        expect(parser.split_arg(arg)).to be === ["i'mjustastring"]
+      end
+
+      context 'testing path traversing', :use_fakefs => true do
+        let(:test_dir) { 'tmp/tests' }
+
+        let(:paths)  { 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')   }
+        let(:rb_other)  { File.expand_path(test_dir + '/other/my_other_ruby_file.rb')   }
+
+        it 'only collects ruby files as test files' do
+          @files = [ rb_test, pl_test, sh_test, rb_other ]
+          paths
+          expect(parser.file_list([File.expand_path(test_dir)])).to be === [rb_test, rb_other]
+        end
+        it 'raises an error when no ruby files are found' do
+          @files = [ pl_test, sh_test ]
+          paths
+          expect{parser.file_list([File.expand_path(test_dir)])}.to raise_error(ArgumentError) 
+        end
+        it 'raises an error when no paths are specified for searching' do
+          @files = ''
+          expect{parser.file_list('')}.to raise_error(ArgumentError)
+        end
+
+      end
+
+      #test yaml file checking
+      it "raises error on improperly formatted yaml file" do
+        FakeFS.deactivate!
+        expect{parser.check_yaml_file(badyaml_path)}.to raise_error(ArgumentError)
+      end
+      it "raises an error when a yaml file is missing" do
+        FakeFS.deactivate!
+        expect{parser.check_yaml_file("not a path")}.to raise_error(ArgumentError)
+      end
+
+      it "can correctly combine arguments from different sources" do
+        FakeFS.deactivate!
+        args = ["-h", hosts_path, "--debug", "--type", "git", "--install", "PUPPET/1.0,HIERA/hello"]
+        expect(parser.parse_args(args)).to be === {:hosts_file=>hosts_path, :options_file=>nil, :type=>"git", :provision=>true, :preserve_hosts=>false, :root_keys=>false, :quiet=>false, :xml=>false, :color=>true, :debug=>true, :dry_run=>false, :timeout=>300, :fail_mode=>nil, :timesync=>false, :repo_proxy=>false, :add_el_extras=>false, :consoleport=>443, :pe_dir=>"/opt/enterprise/dists", :pe_version_file=>"LATEST", :pe_version_file_win=>"LATEST-win", :dot_fog=>"#{home}/.fog", :help=>false, :ec2_yaml=>"config/image_templates/ec2.yaml", :ssh=>{:config=>false, :paranoid=>false, :timeout=>300, :auth_methods=>["publickey"], :port=>22, :forward_agent=>true, :keys=>["#{home}/.ssh/id_rsa"], :user_known_hosts_file=>"#{home}/.ssh/known_hosts"}, :install=>["git://github.com/puppetlabs/puppet.git#1.0", "git://github.com/puppetlabs/hiera.git#hello"], :HOSTS=>{:"pe-ubuntu-lucid"=>{:roles=>["agent", "dashboard", "database", "master"], :vmname=>"pe-ubuntu-lucid", :platform=>"ubuntu-10.04-i386", :snapshot=>"clean-w-keys", :hypervisor=>"fusion"}, :"pe-centos6"=>{:roles=>["agent"], :vmname=>"pe-centos6", :platform=>"el-6-i386", :hypervisor=>"fusion", :snapshot=>"clean-w-keys"}}, :nfs_server=>"none", :helper=>[], :load_path=>[], :tests=>[], :pre_suite=>[], :post_suite=>[], :modules=>[]}
+      end
+
+      it "ensures that file-mode is one of fast/stop" do
+        FakeFS.deactivate!
+        args = ["-h", hosts_path, "--debug", "--fail-mode", "slow"] 
+        expect{parser.parse_args(args)}.to raise_error(ArgumentError)
+      end
+
+      it "ensures that type is one of pe/git" do
+        FakeFS.deactivate!
+        args = ["-h", hosts_path, "--debug", "--type", "unkowns"]
+        expect{parser.parse_args(args)}.to raise_error(ArgumentError)
+      end
+
+      describe "normalize_args" do
+        let(:hosts) do
+          {
+            'HOSTS' => {
+              :master => {
+                :roles => ["master","agent"],
+              },
+              :agent => {
+                :roles => ["agent"],
+              },
+            }
+          }
+        end
+
+        def fake_hosts_file_for_platform(hosts, platform)
+          hosts['HOSTS'].values.each { |h| h[:platform] = platform }
+          filename = "hosts_file_#{platform}"
+          File.open(filename, "w") do |file|
+            YAML.dump(hosts, file)
+          end
+          filename
+        end
+
+        shared_examples_for(:a_platform_supporting_only_agents) do |platform,type|
+          let(:args) { ["--type", type] }
+
+          it "restricts #{platform} hosts to agent for #{type}" do
+            hosts_file = fake_hosts_file_for_platform(hosts, platform)
+            args << "--hosts" << hosts_file
+            expect { parser.parse_args(args) }.to raise_error(ArgumentError, /#{platform}.*can only have role 'agent'/)
+          end
+        end
+
+        context "for pe" do
+          it_should_behave_like(:a_platform_supporting_only_agents, 'solaris', 'pe')
+          it_should_behave_like(:a_platform_supporting_only_agents, 'windows', 'pe')
+          it_should_behave_like(:a_platform_supporting_only_agents, 'el-4', 'pe')
+        end
+
+        context "for foss" do
+          it_should_behave_like(:a_platform_supporting_only_agents, 'windows', 'git')
+          it_should_behave_like(:a_platform_supporting_only_agents, 'el-4', 'git')
+
+          it "allows master role for solaris" do
+            hosts_file = fake_hosts_file_for_platform(hosts, 'solaris')
+            args = ["--type", "git", "--hosts", hosts_file]
+            options_hash = parser.parse_args(args)
+            expect(options_hash[:HOSTS][:master][:platform]).to match(/solaris/)
+            expect(options_hash[:HOSTS][:master][:roles]).to include('master')
+            expect(options_hash[:type]).to eq('git')
+          end
+        end
+      end
+    end
+  end
+end

data/spec/beaker/options/pe_version_scaper_spec.rb

@@ -0,0 +1,15 @@
+require "spec_helper"
+
+module Beaker
+  module Options
+    describe PEVersionScraper do
+      it "can pull version from local LATEST file" do
+        expect(PEVersionScraper.load_pe_version(File.expand_path(File.dirname(__FILE__)), "data/LATEST")) === '3.0.0'
+      end
+      it "raises error when file doesn't exist" do
+        expect{PEVersionScraper.load_pe_version("not a valid path", "not a valid filename")}.to raise_error(ArgumentError)
+      end
+
+    end
+  end
+end

data/spec/beaker/options/presets_spec.rb

@@ -0,0 +1,24 @@
+require "spec_helper"
+
+module Beaker
+  module Options
+
+    describe Presets do
+      let(:presets)    { Presets }
+
+      it "returns an env_vars OptionsHash" do
+        expect(presets.env_vars).to be_instance_of(Beaker::Options::OptionsHash)
+      end
+
+      it "removes all empty/nil entries in env_vars" do
+        expect(presets.env_vars.has_value?(nil)).to be === false and expect(presets.env_vars.has_value?({})).to be === false
+      end
+
+      it "returns a presets OptionsHash" do
+        expect(presets.presets).to be_instance_of(Beaker::Options::OptionsHash)
+      end
+
+
+    end
+  end
+end

data/spec/beaker/puppet_command_spec.rb

@@ -5,13 +5,12 @@ module Beaker
   describe PuppetCommand do
     let(:command) { @command || '/bin/ls' }
     let(:args)    { @args    || Array.new }
-    let(:options) { @options || Hash.new  }
+    let(:options) { @options || {} }
+    subject(:cmd) { PuppetCommand.new( command, args, options ) }
+    let(:host)    { make_host( 'name', { :platform => @platform } ) }
 
-    subject(:cmd) { PuppetCommand.new( command, *args, options ) }
-    let :config do
-      MockConfig.new({}, {'name' => {'platform' => @platform}}, @pe)
-    end
-    let(:host)    { Host.create 'name', {}, config }
+    let( :nix_path ) { %q[PATH="/usr/bin:/opt/puppet-git-repos/hiera/bin:${PATH}"] }
+    let( :nix_lib  ) { %q[RUBYLIB="/opt/puppet-git-repos/hiera/lib:/opt/puppet-git-repos/hiera-puppet/lib:${RUBYLIB}"] }
 
     it 'creates a Windows env for a Windows host' do
       @platform = 'windows'
@@ -20,13 +19,21 @@ module Beaker
       @options = { :foo => 'bar' }
 
       expect( host ).to be_a_kind_of Windows::Host
-      expect( cmd.options ).to be == @options
-      expect( cmd.args    ).to be == @args
+      expect( cmd.options ).to be == options
+      expect( cmd.args    ).to be == [@args]
       expect( cmd.command ).to be == "puppet " + @command
 
       expect( cmd.args_string    ).to be == 'to the baz'
       expect( cmd.options_string ).to be == '--foo=bar'
-      expect( cmd.environment_string_for(host, cmd.environment) ).to be == "env PATH=\"/opt/puppet-git-repos/hiera/bin:${PATH}\" RUBYLIB=\"`cygpath -w /opt/puppet-git-repos/hiera/lib`;`cygpath -w /opt/puppet-git-repos/hiera-puppet/lib`;${RUBYLIB}\" cmd.exe /c"
+
+      win_path  = %q[PATH="/opt/puppet-git-repos/hiera/bin:${PATH}"]
+      win_lib   = %q[RUBYLIB="`cygpath -w /opt/puppet-git-repos/hiera/lib`;`cygpath -w /opt/puppet-git-repos/hiera-puppet/lib`;${RUBYLIB}"]
+      cmd_exe   = %q[cmd.exe /c]
+
+      command_line = cmd.environment_string_for( host, cmd.environment )
+      expect( command_line ).to include( win_path )
+      expect( command_line ).to include( win_lib )
+      expect( command_line ).to include( cmd_exe )
     end
 
     it 'creates a Unix env for a Unix host' do
@@ -37,12 +44,15 @@ module Beaker
 
       expect( host ).to be_a_kind_of Unix::Host
       expect( cmd.options ).to be == @options
-      expect( cmd.args    ).to be == @args
+      expect( cmd.args    ).to be == [@args]
       expect( cmd.command ).to be == "puppet " + @command
 
       expect( cmd.args_string    ).to be == 'to the baz'
       expect( cmd.options_string ).to be == '--foo=bar'
-      expect( cmd.environment_string_for(host, cmd.environment) ).to be == "env PATH=\"/usr/bin:/opt/puppet-git-repos/hiera/bin:${PATH}\" RUBYLIB=\"/opt/puppet-git-repos/hiera/lib:/opt/puppet-git-repos/hiera-puppet/lib:${RUBYLIB}\""
+
+      command_line = cmd.environment_string_for( host, cmd.environment )
+      expect( command_line ).to include( nix_path )
+      expect( command_line ).to include( nix_lib )
     end
 
     it 'creates an AIX env for an AIX host' do
@@ -53,12 +63,15 @@ module Beaker
 
       expect( host ).to be_a_kind_of Aix::Host
       expect( cmd.options ).to be == @options
-      expect( cmd.args    ).to be == @args
+      expect( cmd.args    ).to be == [@args]
       expect( cmd.command ).to be == "puppet " + @command
 
       expect( cmd.args_string    ).to be == 'to the baz'
       expect( cmd.options_string ).to be == '--foo=bar'
-      expect( cmd.environment_string_for(host, cmd.environment) ).to be == "env PATH=\"/usr/bin:/opt/puppet-git-repos/hiera/bin:${PATH}\" RUBYLIB=\"/opt/puppet-git-repos/hiera/lib:/opt/puppet-git-repos/hiera-puppet/lib:${RUBYLIB}\""
+
+      command_line = cmd.environment_string_for( host, cmd.environment )
+      expect( command_line ).to include( nix_path )
+      expect( command_line ).to include( nix_lib )
     end
 
     it 'correctly adds additional ENV to default ENV' do
@@ -69,12 +82,17 @@ module Beaker
 
       expect( host ).to be_a_kind_of Unix::Host
       expect( cmd.options ).to be == @options
-      expect( cmd.args    ).to be == @args
+      expect( cmd.args    ).to be == [@args]
       expect( cmd.command ).to be == "puppet " + @command
 
       expect( cmd.args_string    ).to be == 'to the baz'
       expect( cmd.options_string ).to be == '--foo=bar'
-      expect( cmd.environment_string_for(host, cmd.environment) ).to be == "env PATH=\"/STRING_ENV/bin\" PATH=\"/usr/bin:/opt/puppet-git-repos/hiera/bin:${PATH}\" RUBYLIB=\"/opt/puppet-git-repos/hiera/lib:/opt/puppet-git-repos/hiera-puppet/lib:${RUBYLIB}\""
+      custom = %q[PATH="/STRING_ENV/bin"]
+
+      command_line = cmd.environment_string_for( host, cmd.environment )
+      expect( command_line ).to include( nix_path )
+      expect( command_line ).to include( nix_lib )
+      expect( command_line ).to include( custom )
     end
 
     it 'correctly adds additional :ENV to default ENV' do
@@ -85,12 +103,17 @@ module Beaker
 
       expect( host ).to be_a_kind_of Unix::Host
       expect( cmd.options ).to be == @options
-      expect( cmd.args    ).to be == @args
+      expect( cmd.args    ).to be == [@args]
       expect( cmd.command ).to be == "puppet " + @command
 
       expect( cmd.args_string    ).to be == 'to the baz'
       expect( cmd.options_string ).to be == '--foo=bar'
-      expect( cmd.environment_string_for(host, cmd.environment) ).to be == "env PATH=\"/SYMBOL_ENV/bin\" PATH=\"/usr/bin:/opt/puppet-git-repos/hiera/bin:${PATH}\" RUBYLIB=\"/opt/puppet-git-repos/hiera/lib:/opt/puppet-git-repos/hiera-puppet/lib:${RUBYLIB}\""
+
+      custom = %q[PATH="/SYMBOL_ENV/bin"]
+      command_line = cmd.environment_string_for( host, cmd.environment )
+      expect( command_line ).to include( nix_path )
+      expect( command_line ).to include( nix_lib )
+      expect( command_line ).to include( custom )
     end
 
     it 'correctly adds additional :environment to default ENV' do
@@ -101,12 +124,17 @@ module Beaker
 
       expect( host ).to be_a_kind_of Unix::Host
       expect( cmd.options ).to be == @options
-      expect( cmd.args    ).to be == @args
+      expect( cmd.args    ).to be == [@args]
       expect( cmd.command ).to be == "puppet " + @command
 
       expect( cmd.args_string    ).to be == 'to the baz'
       expect( cmd.options_string ).to be == '--foo=bar'
-      expect( cmd.environment_string_for(host, cmd.environment) ).to be == "env PATH=\"/SYMBOL_ENVIRONMENT/bin\" PATH=\"/usr/bin:/opt/puppet-git-repos/hiera/bin:${PATH}\" RUBYLIB=\"/opt/puppet-git-repos/hiera/lib:/opt/puppet-git-repos/hiera-puppet/lib:${RUBYLIB}\""
+
+      custom = %q[PATH="/SYMBOL_ENVIRONMENT/bin"]
+      command_line = cmd.environment_string_for( host, cmd.environment )
+      expect( command_line ).to include( nix_path )
+      expect( command_line ).to include( nix_lib )
+      expect( command_line ).to include( custom )
     end
 
     it 'correctly adds additional environment to default ENV' do
@@ -117,12 +145,17 @@ module Beaker
 
       expect( host ).to be_a_kind_of Unix::Host
       expect( cmd.options ).to be == @options
-      expect( cmd.args    ).to be == @args
+      expect( cmd.args    ).to be == [@args]
       expect( cmd.command ).to be == "puppet " + @command
 
       expect( cmd.args_string    ).to be == 'to the baz'
       expect( cmd.options_string ).to be == '--foo=bar'
-      expect( cmd.environment_string_for(host, cmd.environment) ).to be == "env PATH=\"/STRING_ENVIRONMENT/bin\" PATH=\"/usr/bin:/opt/puppet-git-repos/hiera/bin:${PATH}\" RUBYLIB=\"/opt/puppet-git-repos/hiera/lib:/opt/puppet-git-repos/hiera-puppet/lib:${RUBYLIB}\""
+
+      custom = %q[PATH="/STRING_ENVIRONMENT/bin"]
+      command_line = cmd.environment_string_for( host, cmd.environment )
+      expect( command_line ).to include( nix_path )
+      expect( command_line ).to include( nix_lib )
+      expect( command_line ).to include( custom )
     end
   end
 end

data/spec/beaker/shared/error_handler_spec.rb

@@ -0,0 +1,40 @@
+require 'spec_helper'
+
+module Beaker
+  module Shared
+    describe ErrorHandler do
+
+      let( :backtrace ) { "I'm the backtrace\nYes I am!\nI have important information" }
+      let( :logger )    { double( 'logger' ) }
+
+      before :each do
+        logger.stub( :error ).and_return( true )
+        logger.stub( :pretty_backtrace ).and_return( backtrace )
+
+      end
+
+      context 'report_and_raise' do
+
+        it "records the backtrace of the exception to the logger" do
+          ex = Exception.new("ArgumentError")
+          ex.stub( :backtrace ).and_return(backtrace)
+          mesg = "I'm the extra message"
+         
+          backtrace.each_line do |line|
+            logger.should_receive( :error ).with(line)
+          end
+
+          subject.should_receive( :raise ).once
+
+          subject.report_and_raise(logger, ex, mesg) 
+
+        end
+
+
+
+      end
+
+    end
+
+  end
+end

data/spec/beaker/shared/host_handler_spec.rb

@@ -0,0 +1,104 @@
+require 'spec_helper'
+
+module Beaker
+  module Shared
+    describe HostHandler do
+      let( :host_handler )   { Beaker::Shared::HostHandler }
+      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 'get_domain_name' do
+
+        it "can find the domain for a host" do
+          host = make_host('name', { :stdout => "domain labs.lan d.labs.net dc1.labs.net labs.com\nnameserver 10.16.22.10\nnameserver 10.16.22.11" } )
+
+          Command.should_receive( :new ).with( "cat /etc/resolv.conf" ).once
+
+          expect( host_handler.get_domain_name( host ) ).to be === "labs.lan"
+
+        end
+
+        it "can find the search for a host" do
+          host = make_host('name', { :stdout => "search labs.lan d.labs.net dc1.labs.net labs.com\nnameserver 10.16.22.10\nnameserver 10.16.22.11" } )
+
+          Command.should_receive( :new ).with( "cat /etc/resolv.conf" ).once
+
+          expect( host_handler.get_domain_name( host ) ).to be === "labs.lan"
+
+        end
+      end
+
+      context "get_ip" do
+        it "can exec the get_ip command" do
+          host = make_host('name', { :stdout => "192.168.2.130\n" } )
+
+          Command.should_receive( :new ).with( "ip a|awk '/g/{print$2}' | cut -d/ -f1 | head -1" ).once
+
+          expect( host_handler.get_ip( host ) ).to be === "192.168.2.130"
+
+        end
+
+      end
+
+      context "set_etc_hosts" do
+        it "can set the /etc/hosts string on a host" do
+          host = make_host('name', {})
+          etc_hosts = "127.0.0.1  localhost\n192.168.2.130 pe-ubuntu-lucid\n192.168.2.128 pe-centos6\n192.168.2.131 pe-debian6"
+
+          Command.should_receive( :new ).with( "echo '#{etc_hosts}' > /etc/hosts" ).once
+          host.should_receive( :exec ).once
+
+          host_handler.set_etc_hosts(host, etc_hosts)
+        end
+
+      end
+
+      context "hosts_with_role" do
+        it "can find the master in a set of hosts" do
+
+          expect( host_handler.hosts_with_role( hosts, 'master' ) ).to be === [hosts[1]]
+
+        end
+
+        it "can find all agents in a set of hosts" do
+
+          expect( host_handler.hosts_with_role( hosts, 'agent' ) ).to be === hosts
+
+        end
+
+        it "returns [] when no match is found in a set of hosts" do
+
+          expect( host_handler.hosts_with_role( hosts, 'surprise' ) ).to be === []
+
+        end
+
+      end
+
+      context "only_host_with_role" do
+        it "can find the single master in a set of hosts" do
+
+          expect( host_handler.only_host_with_role( hosts, 'master' ) ).to be === hosts[1]
+
+        end
+
+        it "throws an error when more than one host with matching role is found" do
+
+          expect{ host_handler.only_host_with_role( hosts, 'agent' ) }.to raise_error(ArgumentError)
+
+        end
+
+        it "throws an error when no host is found matching the role" do
+
+          expect{ host_handler.only_host_with_role( hosts, 'surprise' ) }.to raise_error(ArgumentError)
+
+        end
+      end
+
+    end
+
+  end
+end