beaker 0.0.0 → 1.0.0

data/lib/beaker/options/pe_version_scraper.rb

@@ -0,0 +1,35 @@
+require 'open-uri'
+module Beaker
+  module Options
+    #A set of functions to determine the PE version to use during testing
+    module PEVersionScraper
+      # Scrape the PE version (such as 3.0) from the file at dist_dir/version_file
+      #
+      # Version file is of the format
+      #
+      #  3.0.1-3-g57b669e
+      #   
+      # @param [String] dist_dir The directory containing the version_file
+      # @param [String] version_file The file to scrape
+      #
+      # @return [String, nil] The PE version in the version_file or nil if not found 
+      # @raise [ArgumentError] Raises if version_file does not exist or cannot be opened
+      def self.load_pe_version dist_dir, version_file
+        version = nil
+        begin
+          open("#{dist_dir}/#{version_file}") do |file|
+            while line = file.gets
+              if /(\w.*)/ =~ line then
+                version = $1.strip
+              end
+            end
+          end
+        rescue Errno::ENOENT, OpenURI::HTTPError => e
+          raise ArgumentError, "Failure to examine #{dist_dir}/#{version_file}\n\t\t#{e.to_s}"
+        end
+        return version
+      end
+
+    end
+  end
+end

data/lib/beaker/options/presets.rb

@@ -0,0 +1,70 @@
+module Beaker
+  module Options
+    #A set of functions representing the environment variables and preset argument values to be incorporated
+    #into the Beaker options Object.
+    module Presets
+
+      # Generates an OptionsHash of the environment variables of interest to Beaker
+      # 
+      # Currently supports:
+      #
+      #   consoleport, IS_PE, pe_dist_dir, pe_version_file, pe_version_file_win
+      #
+      # @return [OptionsHash] The supported environment variables in an OptionsHash,
+      #                       empty or nil environment variables are removed from the OptionsHash
+      def self.env_vars
+        h = Beaker::Options::OptionsHash.new
+        h.merge({
+          :consoleport => ENV['consoleport'] ? ENV['consoleport'].to_i : nil,
+          :type => ENV['IS_PE'] ? 'pe' : nil,
+          :pe_dir => ENV['pe_dist_dir'],
+          :pe_version_file => ENV['pe_version_file'],
+          :pe_version_file_win => ENV['pe_version_file'],
+        }.delete_if {|key, value| value.nil? or value.empty? })
+      end
+
+      # Generates an OptionsHash of preset values for arguments supported by Beaker
+      # 
+      # @return [OptionsHash] The supported arguments in an OptionsHash
+      def self.presets
+        h = Beaker::Options::OptionsHash.new
+        h.merge({
+          :hosts_file => 'sample.cfg',
+          :options_file => nil,
+          :type => 'pe',
+          :provision => true,
+          :preserve_hosts => false,
+          :root_keys => false,
+          :quiet => false,
+          :xml => false,
+          :color => true,
+          :debug => false,
+          :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 => File.join(ENV['HOME'], '.fog'),
+          :ec2_yaml => 'config/image_templates/ec2.yaml',
+          :help => false,
+          :ssh => {
+            :config                => false,
+            :paranoid              => false,
+            :timeout               => 300,
+            :auth_methods          => ["publickey"],
+            :port                  => 22,
+            :forward_agent         => true,
+            :keys                  => ["#{ENV['HOME']}/.ssh/id_rsa"],
+            :user_known_hosts_file => "#{ENV['HOME']}/.ssh/known_hosts",
+          }
+        })
+      end
+
+    end
+  end
+end

data/lib/beaker/shared.rb

@@ -1,4 +1,4 @@
-[ 'repetition', 'error_handler', 'host_handler' ].each do |file|
+[ 'repetition', 'error_handler', 'host_handler', 'timed' ].each do |file|
   begin
     require "beaker/shared/#{file}"
   rescue LoadError
@@ -10,6 +10,7 @@ module Beaker
     include Beaker::Shared::ErrorHandler
     include Beaker::Shared::HostHandler
     include Beaker::Shared::Repetition
+    include Beaker::Shared::Timed
   end
 end
 include Beaker::Shared

data/lib/beaker/shared/host_handler.rb

@@ -37,8 +37,13 @@ module Beaker
 
       def only_host_with_role(hosts, role)
         a_host = hosts_with_role(hosts, role)
-        raise "There can be only one #{role}, but I found:" +
-          "#{a_host.map {|h| h.to_s } }" unless a_host.length == 1
+        case 
+          when a_host.length == 0
+            raise ArgumentError, "There should be one host with #{role} defined!"
+          when a_host.length > 1
+            host_string = ( a_host.map { |host| host.name } ).join( ', ')
+            raise ArgumentError, "There should be only one host with #{role} defined, but I found #{a_host.length} (#{host_string})" 
+        end
         a_host.first 
       end
     end

data/lib/beaker/shared/repetition.rb

@@ -17,6 +17,7 @@ module Beaker
         last_wait, wait = 0, 1
         while not done and attempt <= attempts do
           done = block.call
+          attempt += 1
           sleep wait
           last_wait, wait = wait, last_wait + wait
         end

data/lib/beaker/shared/timed.rb

@@ -0,0 +1,14 @@
+module Beaker
+  module Shared
+    module Timed
+
+      def run_and_report_duration &block
+        start = Time.now
+        block.call
+        Time.now - start
+      end
+
+    end
+  end
+end
+

data/lib/beaker/test_case.rb

@@ -51,9 +51,6 @@ module Beaker
     # 'pe' directories within 'ROOT/setup' for examples.
     attr_reader :version
 
-    # A Hash of values taken from host config file.
-    attr_reader :config
-
     # Parsed command line options.
     attr_reader :options
 
@@ -94,11 +91,9 @@ module Beaker
     #                                        against/on.
     # @param [Logger] logger A logger that implements
     #                        {Beaker::Logger}'s interface.
-    # @param [Hash{String=>String}] config Clusterfck of various config opts.
     # @param [Hash{Symbol=>String}] options Parsed command line options.
     # @param [String] path The local path to a test file to be executed.
-    def initialize(these_hosts, logger, config, options={}, path=nil)
-      @config  = config['CONFIG']
+    def initialize(these_hosts, logger, options={}, path=nil)
       @hosts   = these_hosts
       @logger = logger
       @options = options
@@ -171,43 +166,12 @@ module Beaker
     def to_hash
       hash = {}
       hash['HOSTS'] = {}
-      hash['CONFIG'] = @config
       @hosts.each do |host|
         hash['HOSTS'][host.name] = host.overrides
       end
       hash
     end
 
-    # @deprecated
-    # An proxy for the last {Beaker::Result#stdout} returned by
-    # a method that makes remote calls.  Use the {Beaker::Result}
-    # object returned by the method directly instead. For Usage see
-    # {Beaker::Result}.
-    def stdout
-      return nil if @result.nil?
-      @result.stdout
-    end
-
-    # @deprecated
-    # An proxy for the last {Beaker::Result#stderr} returned by
-    # a method that makes remote calls.  Use the {Beaker::Result}
-    # object returned by the method directly instead. For Usage see
-    # {Beaker::Result}.
-    def stderr
-      return nil if @result.nil?
-      @result.stderr
-    end
-
-    # @deprecated
-    # An proxy for the last {Beaker::Result#exit_code} returned by
-    # a method that makes remote calls.  Use the {Beaker::Result}
-    # object returned by the method directly instead. For Usage see
-    # {Beaker::Result}.
-    def exit_code
-      return nil if @result.nil?
-      @result.exit_code
-    end
-
     # This method retrieves the forge hostname from either:
     # * The environment variable 'forge_host'
     # * The parameter 'forge_host' from the CONFIG hash in a node definition
@@ -218,7 +182,7 @@ module Beaker
     # @return [String] hostname of test forge
     def forge
       ENV['forge_host'] ||
-        @config['forge_host'] ||
+        @options['forge_host'] ||
         'vulcan-acceptance.delivery.puppetlabs.net'
     end
   end

data/lib/beaker/test_suite.rb

@@ -16,29 +16,18 @@ module Beaker
   #   * File Creation Relative to CWD  -- Should be a config option
   #   * Better Method Documentation
   class TestSuite
-    attr_reader :name, :options, :config, :fail_mode
-
-    def initialize(name, hosts, options, config, fail_mode = nil)
-      @name      = name.gsub(/\s+/, '-')
-      @hosts     = hosts
-      @run       = false
-      @options   = options
-      @config    = config
-      @fail_mode = @options[:fail_mode] || fail_mode
-      @logger    = options[:logger]
+    attr_reader :name, :options, :fail_mode
 
+    def initialize(name, hosts, options, fail_mode = nil)
+      @logger     = options[:logger]
       @test_cases = []
-      @test_files = []
-
-      options[:tests].each do |root|
-        if File.file? root then
-          @test_files << root
-        else
-          @test_files += Dir.glob(
-            File.join(root, "**/*.rb")
-          ).select { |f| File.file?(f) }
-        end
-      end
+      @test_files = options[name]
+      @name       = name.to_s.gsub(/\s+/, '-')
+      @hosts      = hosts
+      @run        = false
+      @options    = options
+      @fail_mode  = options[:fail_mode] || fail_mode
+
       report_and_raise(@logger, RuntimeError.new("#{@name}: no test files found..."), "TestSuite: initialize") if @test_files.empty?
 
       @test_files = @test_files.sort
@@ -56,7 +45,7 @@ module Beaker
         @logger.notify
         @logger.notify "Begin #{test_file}"
         start = Time.now
-        test_case = TestCase.new(@hosts, @logger, config, options, test_file).run_test
+        test_case = TestCase.new(@hosts, @logger, options, test_file).run_test
         duration = Time.now - start
         @test_cases << test_case
 
@@ -206,8 +195,6 @@ module Beaker
     - Host Configuration Summary -
       HEREDOC
 
-      config.dump
-
       elapsed_time = @test_cases.inject(0.0) {|r, t| r + t.runtime.to_f }
       average_test_time = elapsed_time / test_count
 
@@ -264,7 +251,6 @@ module Beaker
       @@log_dir ||= File.join("log", @start_time.strftime("%F_%T"))
       unless File.directory?(@@log_dir) then
         FileUtils.mkdir_p(@@log_dir)
-        FileUtils.cp(options[:config],(File.join(@@log_dir,"config.yml")))
 
         latest = File.join("log", "latest")
         if !File.exist?(latest) or File.symlink?(latest) then

data/lib/beaker/utils/repo_control.rb

@@ -26,11 +26,9 @@ module Beaker
         return url
       end
 
-      def apt_get_update
-        @hosts.each do |host|
-          if host['platform'] =~ /ubuntu|debian/
-            host.exec(Command.new("apt-get -y -f -m update"))
-          end
+      def apt_get_update host
+        if host[:platform] =~ /(ubuntu)|(debian)/ 
+          host.exec(Command.new("apt-get -y -f -m update"))
         end
       end
 
@@ -51,14 +49,14 @@ module Beaker
           when host['platform'] =~ /ubuntu/
             host.exec(Command.new("if test -f /etc/apt/apt.conf; then mv /etc/apt/apt.conf /etc/apt/apt.conf.bk; fi"))
             copy_file_to_remote(host, '/etc/apt/apt.conf', APT_CFG)
-            apt_get_update
+            apt_get_update(host)
           when host['platform'] =~ /debian/
             host.exec(Command.new("if test -f /etc/apt/apt.conf; then mv /etc/apt/apt.conf /etc/apt/apt.conf.bk; fi"))
             copy_file_to_remote(host, '/etc/apt/apt.conf', APT_CFG)
-            apt_get_update
+            apt_get_update(host)
           when host['platform'] =~ /solaris-11/
             host.exec(Command.new("/usr/bin/pkg unset-publisher solaris || :"))
-            host.exec(Command.new(host,"/usr/bin/pkg set-publisher -g %s solaris" % IPS_PKG_REPO))
+            host.exec(Command.new("/usr/bin/pkg set-publisher -g %s solaris" % IPS_PKG_REPO))
           else
             @logger.debug "#{host}: repo proxy configuration not modified"
           end

data/lib/beaker/utils/setup_helper.rb

@@ -1,6 +1,10 @@
 module Beaker
   module Utils
     class SetupHelper
+      ETC_HOSTS_PATH = "/etc/hosts"
+      ETC_HOSTS_PATH_SOLARIS = "/etc/inet/hosts"
+      ROOT_KEYS_SCRIPT = "https://raw.github.com/puppetlabs/puppetlabs-sshkeys/master/templates/scripts/manage_root_authorized_keys"
+      ROOT_KEYS_SYNC_CMD = "curl -k -o - #{ROOT_KEYS_SCRIPT} | %s"
 
       def initialize(options, hosts)
         @options = options.dup
@@ -8,21 +12,9 @@ module Beaker
         @logger = options[:logger]
       end
 
-      def find_masters(hosts)
-        hosts.select do |host|
-          host['roles'].include?("master")
-        end
-      end
-
-      def find_only_master(hosts)
-        m = find_masters(hosts)
-        raise "too many masters, expected one but found #{m.map {|h| h.to_s }}" unless m.length == 1
-        m.first
-      end
-
       def add_master_entry
         @logger.notify "Add Master entry to /etc/hosts"
-        master = find_only_master(@hosts)
+        master = only_host_with_role(@hosts, :master)
         if master["hypervisor"] and master["hypervisor"] =~ /vagrant/
           @logger.debug "Don't update master entry on vagrant masters"
           return
@@ -35,9 +27,9 @@ module Beaker
         end
         ip=stdout.chomp
 
-        path = "/etc/hosts"
+        path = ETC_HOSTS_PATH
         if master['platform'].include? 'solaris'
-          path = "/etc/inet/hosts"
+          path = ETC_HOSTS_PATH_SOLARIS
         end
 
         @logger.debug "Update %s on #{master}" % path
@@ -57,15 +49,12 @@ module Beaker
         # but we're deliberately taking the approach of "assume it will work, fix it
         # when reality dictates otherwise"
         @logger.notify "Sync root authorized_keys from github"
-        script = "https://raw.github.com/puppetlabs/puppetlabs-sshkeys/master/templates/scripts/manage_root_authorized_keys"
-        setup_root_authorized_keys = "curl -k -o - #{script} | %s"
-        @logger.notify "Sync root authorized_keys from github"
         @hosts.each do |host|
           # Allow all exit code, as this operation is unlikely to cause problems if it fails.
           if host['platform'].include? 'solaris'
-            host.exec(Command.new(setup_root_authorized_keys % "bash"), :acceptable_exit_codes => (0..255))
+            host.exec(Command.new(ROOT_KEYS_SYNC_CMD % "bash"), :acceptable_exit_codes => (0..255))
           else
-            host.exec(Command.new(setup_root_authorized_keys % "env PATH=/usr/gnu/bin:$PATH bash"), :acceptable_exit_codes => (0..255))
+            host.exec(Command.new(ROOT_KEYS_SYNC_CMD % "env PATH=/usr/gnu/bin:$PATH bash"), :acceptable_exit_codes => (0..255))
           end
         end
       rescue => e

data/spec/beaker/answers_spec.rb

@@ -0,0 +1,109 @@
+require 'spec_helper'
+
+module Beaker
+  describe Answers do
+    let( :basic_hosts ) { make_hosts( { 'pe_ver' => @ver } ) }
+    let( :hosts )       { basic_hosts[0]['roles'] = ['master', 'database', 'dashboard']
+                          basic_hosts[1]['platform'] = 'windows'
+                          basic_hosts }
+    let( :master_certname ) { 'master_certname' }
+
+    it 'generates 3.0 answers for 3.1 hosts' do
+      @ver = '3.1'
+      Beaker::Answers::Version30.should_receive( :answers ).with( hosts, master_certname, {} ).once
+      subject.answers( @ver, hosts, master_certname, {} )
+    end
+
+    it 'generates 3.0 answers for 3.0 hosts' do
+      @ver = '3.0'
+      Beaker::Answers::Version30.should_receive( :answers ).with( hosts, master_certname, {} ).once
+      subject.answers( @ver, hosts, master_certname, {} )
+    end
+
+    it 'generates 2.8 answers for 2.8 hosts' do
+      @ver = '2.8'
+      Beaker::Answers::Version28.should_receive( :answers ).with( hosts, master_certname, {} ).once
+      subject.answers( @ver, hosts, master_certname, {} )
+    end
+
+    it 'generates 2.0 answers for 2.0 hosts' do
+      @ver = '2.0'
+      Beaker::Answers::Version20.should_receive( :answers ).with( hosts, master_certname, {} ).once
+      subject.answers( @ver, hosts, master_certname, {} )
+    end
+
+    it 'raises an error for an unknown version' do
+      @ver = 'x.x'
+      expect{ subject.answers( @ver, hosts, master_certname, {} ) }.to raise_error( NotImplementedError )
+    end
+  end
+
+  module Answers
+    describe Version30 do
+      let( :basic_hosts ) { make_hosts( { 'pe_ver' => @ver } ) }
+      let( :hosts )       { basic_hosts[0]['roles'] = ['master', 'database', 'dashboard']
+                            basic_hosts[1]['platform'] = 'windows'
+                            basic_hosts }
+      let( :master_certname ) { 'master_certname' }
+
+      it 'uses simple answers for upgrade from 3.0.x to 3.0.x' do
+        @ver = '3.0'
+        expect( subject.answers( hosts, master_certname, { :type => :upgrade } )).to be === { "vm2"=>{ :q_install=>"y", :q_install_vendor_packages=>"y" }, "vm1"=>{ :q_install=>"y", :q_install_vendor_packages=>"y" }, "vm3"=>{ :q_install=>"y", :q_install_vendor_packages=>"y" } }
+      end
+
+      it 'sets correct answers for an agent' do
+        expect( subject.answers( hosts, master_certname,{} )['vm3'] ).to be === { :q_install=>"y", :q_vendor_packages_install=>"y", :q_puppetagent_install=>"y", :q_puppet_cloud_install=>"y", :q_verify_packages=>"y", :q_puppet_symlinks_install=>"y", :q_puppetagent_certname=>hosts[2], :q_puppetagent_server=>master_certname, :q_puppetmaster_install=>"n", :q_all_in_one_install=>"n", :q_puppet_enterpriseconsole_install=>"n", :q_puppetdb_install=>"n", :q_database_install=>"n" }
+      end
+
+      it 'sets correct answers for a master' do
+        expect( subject.answers( hosts, master_certname, {} )['vm1'] ).to be === { :q_install=>"y", :q_vendor_packages_install=>"y", :q_puppetagent_install=>"y", :q_puppet_cloud_install=>"y", :q_verify_packages=>"y", :q_puppet_symlinks_install=>"y", :q_puppetagent_certname=>hosts[0], :q_puppetagent_server=>master_certname, :q_puppetmaster_install=>"y", :q_all_in_one_install=>"y", :q_puppet_enterpriseconsole_install=>"y", :q_puppetdb_install=>"y", :q_database_install=>"y", :q_puppetdb_hostname=>hosts[0], :q_puppetdb_port=>8081, :q_puppetmaster_dnsaltnames=>"master_certname,puppet,#{hosts[0][:ip]}", :q_puppetmaster_enterpriseconsole_hostname=>hosts[0], :q_puppetmaster_enterpriseconsole_port=>443, :q_puppetmaster_certname=>"master_certname", :q_puppetdb_database_name=>"pe-puppetdb", :q_puppetdb_database_user=>"mYpdBu3r", :q_puppetdb_database_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_auth_database_name=>"console_auth", :q_puppet_enterpriseconsole_auth_database_user=>"mYu7hu3r", :q_puppet_enterpriseconsole_auth_database_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_database_name=>"console", :q_puppet_enterpriseconsole_database_user=>"mYc0nS03u3r", :q_puppet_enterpriseconsole_database_password=>"'~!@\#$%^*-/ aZ'", :q_database_host=>hosts[0], :q_database_port=>5432, :q_pe_database=>"y", :q_puppet_enterpriseconsole_inventory_hostname=>hosts[0], :q_puppet_enterpriseconsole_inventory_certname=>hosts[0], :q_puppet_enterpriseconsole_inventory_dnsaltnames=>hosts[0], :q_puppet_enterpriseconsole_inventory_port=>8140, :q_puppet_enterpriseconsole_master_hostname=>hosts[0], :q_puppet_enterpriseconsole_auth_user_email=>"'admin@example.com'", :q_puppet_enterpriseconsole_auth_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_httpd_port=>443, :q_puppet_enterpriseconsole_smtp_host=>"'vm1'", :q_puppet_enterpriseconsole_smtp_use_tls=>"'n'", :q_puppet_enterpriseconsole_smtp_port=>"'25'", :q_database_root_password=>"'=ZYdjiP3jCwV5eo9s1MBd'", :q_database_root_user=>"pe-postgres" }
+      end
+      it 'generates nil answers for a windows host' do
+        expect( subject.answers( hosts, master_certname, {} )['vm2'] ).to be === nil
+      end
+    end
+
+    describe Version28 do
+      let( :basic_hosts ) { make_hosts( { 'pe_ver' => @ver } ) }
+      let( :hosts )       { basic_hosts[0]['roles'] = ['master', 'database', 'dashboard']
+                            basic_hosts[1]['platform'] = 'windows'
+                            basic_hosts }
+      let( :master_certname ) { 'master_certname' }
+
+      it 'sets correct answers for an agent' do
+        expect( subject.answers( hosts, master_certname,{} )['vm3'] ).to be === { :q_install=>"y", :q_puppetagent_install=>"y", :q_puppet_cloud_install=>"y", :q_puppet_symlinks_install=>"y", :q_vendor_packages_install=>"y", :q_puppetagent_certname=>hosts[2], :q_puppetagent_server=>hosts[0], :q_puppetmaster_install=>"n", :q_puppet_enterpriseconsole_install=>"n" }
+      end
+
+      it 'sets correct answers for a master' do
+        expect( subject.answers( hosts, master_certname, {} )['vm1'] ).to be === { :q_install=>"y", :q_puppetagent_install=>"y", :q_puppet_cloud_install=>"y", :q_puppet_symlinks_install=>"y", :q_vendor_packages_install=>"y", :q_puppetagent_certname=>hosts[0], :q_puppetagent_server=>hosts[0], :q_puppetmaster_install=>"y", :q_puppet_enterpriseconsole_install=>"y", :q_puppetmaster_certname=>"master_certname", :q_puppetmaster_dnsaltnames=>"master_certname,puppet,#{hosts[0][:ip]}", :q_puppetmaster_enterpriseconsole_hostname=>hosts[0], :q_puppetmaster_enterpriseconsole_port=>443, :q_puppetmaster_forward_facts=>"y", :q_puppet_enterpriseconsole_database_install=>"y", :q_puppet_enterpriseconsole_auth_database_name=>"console_auth", :q_puppet_enterpriseconsole_auth_database_user=>"mYu7hu3r", :q_puppet_enterpriseconsole_auth_database_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_database_name=>"console", :q_puppet_enterpriseconsole_database_user=>"mYc0nS03u3r", :q_puppet_enterpriseconsole_database_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_inventory_hostname=>hosts[0], :q_puppet_enterpriseconsole_inventory_certname=>hosts[0], :q_puppet_enterpriseconsole_inventory_dnsaltnames=>hosts[0], :q_puppet_enterpriseconsole_inventory_port=>8140, :q_puppet_enterpriseconsole_master_hostname=>hosts[0], :q_puppet_enterpriseconsole_auth_user_email=>"'admin@example.com'", :q_puppet_enterpriseconsole_auth_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_httpd_port=>443, :q_puppet_enterpriseconsole_smtp_host=>"'vm1'", :q_puppet_enterpriseconsole_smtp_use_tls=>"'n'", :q_puppet_enterpriseconsole_smtp_port=>"'25'", :q_puppet_enterpriseconsole_auth_user=>"'admin@example.com'" }
+      end
+
+      it 'generates nil answers for a windows host' do
+        expect( subject.answers( hosts, master_certname, {} )['vm2'] ).to be === nil
+      end
+
+
+    end
+    describe Version20 do
+      let( :basic_hosts ) { make_hosts( { 'pe_ver' => @ver } ) }
+      let( :hosts )       { basic_hosts[0]['roles'] = ['master', 'database', 'dashboard']
+                            basic_hosts[1]['platform'] = 'windows'
+                            basic_hosts }
+      let( :master_certname ) { 'master_certname' }
+
+      it 'sets correct answers for an agent' do
+        expect( subject.answers( hosts, master_certname,{} )['vm3'] ).to be === { :q_install=>"y", :q_puppetagent_install=>"y", :q_puppet_cloud_install=>"y", :q_puppet_symlinks_install=>"y", :q_vendor_packages_install=>"y", :q_puppetagent_certname=>hosts[2], :q_puppetagent_server=>hosts[0], :q_puppetmaster_install=>"n", :q_puppet_enterpriseconsole_install=>"n" }
+      end
+
+      it 'sets correct answers for a master' do
+        expect( subject.answers( hosts, master_certname, {} )['vm1'] ).to be === { :q_install=>"y", :q_puppetagent_install=>"y", :q_puppet_cloud_install=>"y", :q_puppet_symlinks_install=>"y", :q_vendor_packages_install=>"y", :q_puppetagent_certname=>hosts[0], :q_puppetagent_server=>hosts[0], :q_puppetmaster_install=>"y", :q_puppet_enterpriseconsole_install=>"y", :q_puppetmaster_certname=>"master_certname", :q_puppetmaster_dnsaltnames=>"master_certname,puppet,#{hosts[0][:ip]}", :q_puppetmaster_enterpriseconsole_hostname=>hosts[0], :q_puppetmaster_enterpriseconsole_port=>443, :q_puppetmaster_forward_facts=>"y", :q_puppet_enterpriseconsole_database_install=>"y", :q_puppet_enterpriseconsole_auth_database_name=>"console_auth", :q_puppet_enterpriseconsole_auth_database_user=>"mYu7hu3r", :q_puppet_enterpriseconsole_auth_database_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_database_name=>"console", :q_puppet_enterpriseconsole_database_user=>"mYc0nS03u3r", :q_puppet_enterpriseconsole_database_root_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_database_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_inventory_hostname=>hosts[0], :q_puppet_enterpriseconsole_inventory_certname=>hosts[0], :q_puppet_enterpriseconsole_inventory_dnsaltnames=>hosts[0], :q_puppet_enterpriseconsole_inventory_port=>8140, :q_puppet_enterpriseconsole_master_hostname=>hosts[0], :q_puppet_enterpriseconsole_auth_user_email=>"'admin@example.com'", :q_puppet_enterpriseconsole_auth_password=>"'~!@\#$%^*-/ aZ'", :q_puppet_enterpriseconsole_httpd_port=>443, :q_puppet_enterpriseconsole_smtp_host=>"'vm1'", :q_puppet_enterpriseconsole_smtp_use_tls=>"'n'", :q_puppet_enterpriseconsole_smtp_port=>"'25'", :q_puppet_enterpriseconsole_auth_user=>"'admin@example.com'" }
+      end
+
+      it 'generates nil answers for a windows host' do
+        expect( subject.answers( hosts, master_certname, {} )['vm2'] ).to be === nil
+      end
+
+    end
+
+  end
+end