beaker 0.0.0

data/Rakefile

@@ -0,0 +1,121 @@
+
+task :default => [ 'test:spec' ]
+
+task :spec do
+  Rake::Task['test:spec'].invoke
+end
+
+namespace :test do
+  desc 'Run specs (with coverage on 1.9), alias `spec` & the default'
+  task :spec do
+    original_dir = Dir.pwd
+    Dir.chdir( File.expand_path(File.dirname(__FILE__)) )
+    sh 'export COVERAGE=true; bundle exec rspec'
+    Dir.chdir( original_dir )
+  end
+end
+
+
+###########################################################
+#
+#   Documentation Tasks
+#
+###########################################################
+
+
+DOCS_DAEMON = "yard server --reload --daemon --server thin"
+FOREGROUND_SERVER = 'bundle exec yard server --reload --verbose --server thin lib/beaker'
+
+def running?( cmdline )
+  ps = `ps -ef`
+  found = ps.lines.grep( /#{Regexp.quote( cmdline )}/ )
+  if found.length > 1
+    raise StandardError, "Found multiple YARD Servers. Don't know what to do."
+  end
+
+  yes = found.empty? ? false : true
+  return yes, found.first
+end
+
+def pid_from( output )
+  output.squeeze(' ').strip.split(' ')[1]
+end
+
+desc 'Start the documentation server in the foreground'
+task :docs => 'docs:clear' do
+  original_dir = Dir.pwd
+  Dir.chdir( File.expand_path(File.dirname(__FILE__)) )
+  sh FOREGROUND_SERVER
+  Dir.chdir( original_dir )
+end
+
+namespace :docs do
+
+  desc 'Clear the generated documentation cache'
+  task :clear do
+    original_dir = Dir.pwd
+    Dir.chdir( File.expand_path(File.dirname(__FILE__)) )
+    sh 'rm -rf docs'
+    Dir.chdir( original_dir )
+  end
+
+  desc 'Generate static documentation'
+  task :gen => 'docs:clear' do
+    original_dir = Dir.pwd
+    Dir.chdir( File.expand_path(File.dirname(__FILE__)) )
+    sh 'bundle exec yard doc'
+    Dir.chdir( original_dir )
+  end
+
+  desc 'Run the documentation server in the background, alias `bg`'
+  task :background => 'docs:clear' do
+    yes, output = running?( DOCS_DAEMON )
+    if yes
+      puts "Not starting a new YARD Server..."
+      puts "Found one running with pid #{pid_from( output )}."
+    else
+      original_dir = Dir.pwd
+      Dir.chdir( File.expand_path(File.dirname(__FILE__)) )
+      sh "bundle exec #{DOCS_DAEMON}"
+      Dir.chdir( original_dir )
+    end
+  end
+
+  task(:bg) { Rake::Task['docs:background'].invoke }
+
+  desc 'Check the status of the documentation server'
+  task :status do
+    yes, output = running?( DOCS_DAEMON )
+    if yes
+      pid = pid_from( output )
+      puts "Found a YARD Server running with pid #{pid}"
+    else
+      puts "Could not find a running YARD Server."
+    end
+  end
+
+  desc "Stop a running YARD Server"
+  task :stop do
+    yes, output = running?( DOCS_DAEMON )
+    if yes
+      pid = pid_from( output )
+      puts "Found a YARD Server running with pid #{pid}"
+      `kill #{pid}`
+      puts "Stopping..."
+      yes, output = running?( DOCS_DAEMON )
+      if yes
+        `kill -9 #{pid}`
+        yes, output = running?( DOCS_DAEMON )
+        if yes
+          puts "Could not Stop Server!"
+        else
+          puts "Server stopped."
+        end
+      else
+        puts "Server stopped."
+      end
+    else
+      puts "Could not find a running YARD Server"
+    end
+  end
+end

data/beaker.gemspec

@@ -0,0 +1,42 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require 'rbconfig'
+ruby_conf = defined?(RbConfig) ? RbConfig::CONFIG : Config::CONFIG
+less_than_one_nine = ruby_conf['MAJOR'].to_i == 1 && ruby_conf['MINOR'].to_i < 9
+
+Gem::Specification.new do |s|
+  s.name        = "beaker"
+  s.version     = '0.0.0'
+  s.authors     = ["Puppetlabs"]
+  s.email       = ["delivery@puppetlabs.com"]
+  s.homepage    = "https://github.com/puppetlabs/beaker"
+  s.summary     = %q{Let's test Puppet!}
+  s.description = %q{Puppetlabs accceptance testing harness}
+  s.license     = 'Apache2'
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  # Testing dependencies
+  s.add_development_dependency 'rspec', '2.11.0'
+  s.add_development_dependency 'fakefs', '0.4'
+  s.add_development_dependency 'rake'
+  s.add_development_dependency 'simplecov' unless less_than_one_nine
+
+  # Documentation dependencies
+  s.add_development_dependency 'yard'
+  s.add_development_dependency 'markdown' unless less_than_one_nine
+  s.add_development_dependency 'thin'
+
+  # Run time dependencies
+  s.add_runtime_dependency 'json'
+  s.add_runtime_dependency 'net-ssh'
+  s.add_runtime_dependency 'net-scp'
+  s.add_runtime_dependency 'rbvmomi'
+  s.add_runtime_dependency 'blimpy'
+  s.add_runtime_dependency 'nokogiri', '1.5.10'
+  s.add_runtime_dependency 'fission' if RUBY_PLATFORM =~ /darwin/i
+  s.add_runtime_dependency 'inifile'
+end

data/beaker.rb

@@ -0,0 +1,10 @@
+#!/usr/bin/env ruby
+
+if ENV['RUBYLIB']
+  ENV['RUBYLIB'] += ':lib'
+else
+  ENV['RUBYLIB'] = 'lib'
+end
+
+system('bin/beaker', *ARGV)
+exit $?.exitstatus  

data/bin/beaker

@@ -0,0 +1,9 @@
+#!/usr/bin/env ruby
+
+require 'rubygems' unless defined?(Gem)
+require 'beaker'
+
+Beaker::CLI.new.execute!
+
+puts "systest completed successfully, thanks."
+exit 0

data/lib/beaker.rb

@@ -0,0 +1,36 @@
+require 'rubygems' unless defined?(Gem)
+module Beaker
+
+  %w( utils test_suite test_config result command options_parsing network_manager cli ).each do |lib|
+    begin
+      require "beaker/#{lib}"
+    rescue LoadError
+      require File.expand_path(File.join(File.dirname(__FILE__), 'beaker', lib))
+    end
+  end
+  # These really are our sub-systems that live within the harness today
+  # Ideally I would like to see them split out into modules that can be
+  # included as such here
+  #
+  # The Testing DSL
+  require 'beaker/dsl'
+  #
+  # Our Host Abstraction Layer
+  require 'beaker/host'
+  #
+  # Our Hypervisor Abstraction Layer
+  require 'beaker/hypervisor'
+  #
+  # How we manage connecting to hosts and hypervisors
+  #require 'beaker/connectivity'
+  #
+  # Our test runner, suite, test cases and steps
+  #require 'beaker/runner'
+  #
+  # Common setup and testing steps
+  #require 'beaker/steps'
+  #
+  # Shared methods and helpers
+  require 'beaker/shared'
+
+end

data/lib/beaker/answers.rb

@@ -0,0 +1,29 @@
+[ 'version30', 'version28' ].each do |file|
+  begin
+    require "beaker/answers/#{file}"
+  rescue LoadError
+    require File.expand_path(File.join(File.dirname(__FILE__), 'answers', file))
+  end
+end
+
+module Beaker
+  module Answers
+
+    def self.answers(version, hosts, master_certname, options)
+
+      case version
+        when /\A3\.0/
+          Version30.answers(hosts, master_certname, options)
+        when /\A2\.8/
+          Version28.answers(hosts, master_certname, options)
+        else
+          raise NotImplementedError, "Don't know how to generate answers for #{version}"
+      end
+    end
+
+    def self.answer_string(host, answers)
+      answers[host.name].map { |k,v| "#{k}=#{v}" }.join("\n")
+    end
+
+  end
+end

data/lib/beaker/answers/version28.rb

@@ -0,0 +1,104 @@
+module Beaker
+  module Answers
+    module Version28
+
+      def self.host_answers(host, master_certname, master, dashboard, options)
+        return nil if host['platform'] =~ /windows/
+
+        agent_a = {
+          :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 => host,
+          :q_puppetagent_server => master,
+
+          # Disable console and master by default
+          # This will be overridden by other blocks being merged in
+          :q_puppetmaster_install => 'n',
+          :q_puppet_enterpriseconsole_install => 'n',
+        }
+
+        master_a = {
+          :q_puppetmaster_install => 'y',
+          :q_puppetmaster_certname => master_certname,
+          :q_puppetmaster_install => 'y',
+          :q_puppetmaster_dnsaltnames => master_certname+",puppet",
+          :q_puppetmaster_enterpriseconsole_hostname => dashboard,
+          :q_puppetmaster_enterpriseconsole_port => 443,
+          :q_puppetmaster_forward_facts => 'y',
+        }
+
+        if master['ip']
+          master_a[:q_puppetmaster_dnsaltnames]+=","+master['ip']
+        end
+
+        dashboard_user = "'#{ENV['q_puppet_enterpriseconsole_auth_user_email'] || 'admin@example.com'}'"
+        smtp_host = "'#{ENV['q_puppet_enterpriseconsole_smtp_host'] || dashboard}'"
+        dashboard_password = ENV['q_puppet_enterpriseconsole_auth_password'] || '~!@#$%^*-/ aZ'
+        smtp_port = "'#{ENV['q_puppet_enterpriseconsole_smtp_port'] || 25}'"
+        smtp_username = ENV['q_puppet_enterpriseconsole_smtp_username']
+        smtp_password = ENV['q_puppet_enterpriseconsole_smtp_password']
+        smtp_use_tls = "'#{ENV['q_puppet_enterpriseconsole_smtp_use_tls'] || 'n'}'"
+
+        console_a = {
+          :q_puppet_enterpriseconsole_install => '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 => "'#{dashboard_password}'",
+          :q_puppet_enterpriseconsole_database_name => 'console',
+          :q_puppet_enterpriseconsole_database_user => 'mYc0nS03u3r',
+          :q_puppet_enterpriseconsole_database_password => "'#{dashboard_password}'",
+          :q_puppet_enterpriseconsole_inventory_hostname => host,
+          :q_puppet_enterpriseconsole_inventory_certname => host,
+          :q_puppet_enterpriseconsole_inventory_dnsaltnames => master,
+          :q_puppet_enterpriseconsole_inventory_port => 8140,
+          :q_puppet_enterpriseconsole_master_hostname => master,
+
+          :q_puppet_enterpriseconsole_auth_user_email => dashboard_user,
+          :q_puppet_enterpriseconsole_auth_password => "'#{dashboard_password}'",
+
+          :q_puppet_enterpriseconsole_httpd_port => 443,
+
+          :q_puppet_enterpriseconsole_smtp_host => smtp_host,
+          :q_puppet_enterpriseconsole_smtp_use_tls => smtp_use_tls,
+          :q_puppet_enterpriseconsole_smtp_port => smtp_port,
+        }
+
+        console_a[:q_puppet_enterpriseconsole_auth_user] = console_a[:q_puppet_enterpriseconsole_auth_user_email]
+
+        if smtp_password and smtp_username
+          console_a.merge!({
+                             :q_puppet_enterpriseconsole_smtp_password => "'#{smtp_password}'",
+                             :q_puppet_enterpriseconsole_smtp_username => "'#{smtp_username}'",
+                             :q_puppet_enterpriseconsole_smtp_user_auth => 'y'
+                           })
+        end
+
+        answers = agent_a.dup
+        if host == master
+          answers.merge! master_a
+        end
+
+        if host == dashboard
+          answers.merge! console_a
+        end
+
+        return answers
+      end
+
+      def self.answers(hosts, master_certname, options)
+        the_answers = {}
+        dashboard = only_host_with_role(hosts, 'dashboard')
+        master = only_host_with_role(hosts, 'master')
+        hosts.each do |h|
+            the_answers[h.name] = host_answers(h, master_certname, master, dashboard, options)
+        end
+        return the_answers
+      end
+
+    end
+  end
+end

data/lib/beaker/answers/version30.rb

@@ -0,0 +1,194 @@
+module Beaker
+  module Answers
+    module Version30
+
+      def self.host_answers(host, master_certname, master, database, dashboard, options)
+        # Windows hosts don't have normal answers...
+        return nil if host['platform'] =~ /windows/
+
+        # Everything's an agent
+        agent_a = {
+          :q_puppetagent_install => 'y',
+          :q_puppet_cloud_install => 'y',
+          :q_verify_packages => ENV['q_verify_packages'] || 'y',
+          :q_puppet_symlinks_install => 'y',
+          :q_puppetagent_certname => host,
+          :q_puppetagent_server => master,
+
+          # Disable database, console, and master by default
+          # This will be overridden by other blocks being merged in.
+          :q_puppetmaster_install => 'n',
+          :q_all_in_one_install => 'n',
+          :q_puppet_enterpriseconsole_install => 'n',
+          :q_puppetdb_install => 'n',
+          :q_database_install => 'n',
+        }
+
+        # These base answers are needed by all
+        common_a = {
+          :q_install => 'y',
+          :q_vendor_packages_install => 'y',
+        }
+
+        # master/database answers
+        master_database_a = {
+          :q_puppetmaster_certname => master_certname
+        }
+
+        # Master/dashboard answers
+        master_console_a = {
+          :q_puppetdb_hostname => database,
+          :q_puppetdb_port => 8081
+        }
+
+        # Master only answers
+        master_a = {
+          :q_puppetmaster_install => 'y',
+          :q_puppetmaster_dnsaltnames => master_certname+",puppet",
+          :q_puppetmaster_enterpriseconsole_hostname => dashboard,
+          :q_puppetmaster_enterpriseconsole_port => 443,
+        }
+
+        if master['ip']
+          master_a[:q_puppetmaster_dnsaltnames]+=","+master['ip']
+        end
+
+        # Common answers for console and database
+        dashboard_password = "'#{ENV['q_puppet_enterpriseconsole_auth_password'] || '~!@#$%^*-/ aZ'}'"
+        puppetdb_password = "'#{ENV['q_puppetdb_password'] || '~!@#$%^*-/ aZ'}'"
+
+        console_database_a = {
+          :q_puppetdb_database_name => 'pe-puppetdb',
+          :q_puppetdb_database_user => 'mYpdBu3r',
+          :q_puppetdb_database_password => puppetdb_password,
+          :q_puppet_enterpriseconsole_auth_database_name => 'console_auth',
+          :q_puppet_enterpriseconsole_auth_database_user => 'mYu7hu3r',
+          :q_puppet_enterpriseconsole_auth_database_password => dashboard_password,
+          :q_puppet_enterpriseconsole_database_name => 'console',
+          :q_puppet_enterpriseconsole_database_user => 'mYc0nS03u3r',
+          :q_puppet_enterpriseconsole_database_password => dashboard_password,
+
+          :q_database_host => database,
+          :q_database_port => 5432
+        }
+
+        # Console only answers
+        dashboard_user = "'#{ENV['q_puppet_enterpriseconsole_auth_user_email'] || 'admin@example.com'}'"
+
+        smtp_host = "'#{ENV['q_puppet_enterpriseconsole_smtp_host'] || dashboard}'"
+        smtp_port = "'#{ENV['q_puppet_enterpriseconsole_smtp_port'] || 25}'"
+        smtp_username = ENV['q_puppet_enterpriseconsole_smtp_username']
+        smtp_password = ENV['q_puppet_enterpriseconsole_smtp_password']
+        smtp_use_tls = "'#{ENV['q_puppet_enterpriseconsole_smtp_use_tls'] || 'n'}'"
+
+        console_a = {
+          :q_puppet_enterpriseconsole_install => 'y',
+          :q_puppet_enterpriseconsole_inventory_hostname => host,
+          :q_puppet_enterpriseconsole_inventory_certname => host,
+          :q_puppet_enterpriseconsole_inventory_dnsaltnames => dashboard,
+          :q_puppet_enterpriseconsole_inventory_port => 8140,
+          :q_puppet_enterpriseconsole_master_hostname => master,
+
+          :q_puppet_enterpriseconsole_auth_user_email => dashboard_user,
+          :q_puppet_enterpriseconsole_auth_password => dashboard_password,
+
+          :q_puppet_enterpriseconsole_httpd_port => 443,
+
+          :q_puppet_enterpriseconsole_smtp_host => smtp_host,
+          :q_puppet_enterpriseconsole_smtp_use_tls => smtp_use_tls,
+          :q_puppet_enterpriseconsole_smtp_port => smtp_port,
+        }
+
+        if smtp_password and smtp_username
+          console_a.merge!({
+                             :q_puppet_enterpriseconsole_smtp_password => "'#{smtp_password}'",
+                             :q_puppet_enterpriseconsole_smtp_username => "'#{smtp_username}'",
+                             :q_puppet_enterpriseconsole_smtp_user_auth => 'y'
+                           })
+        end
+
+        # Database only answers
+        database_a = {
+          :q_puppetdb_install => 'y',
+          :q_database_install => 'y',
+          :q_database_root_password => "'=ZYdjiP3jCwV5eo9s1MBd'",
+          :q_database_root_user => 'pe-postgres',
+        }
+
+        # Special answers for special hosts
+        aix_a = {
+          :q_run_updtvpkg => 'y',
+        }
+
+        answers = common_a.dup
+
+        unless options[:type] == :upgrade
+          answers.merge! agent_a
+        end
+
+        if host == master
+          answers.merge! master_console_a
+          unless options[:type] == :upgrade
+            answers.merge! master_a
+            answers.merge! master_database_a
+          end
+        end
+
+        if host == dashboard
+          answers.merge! master_console_a
+          answers.merge! console_database_a
+          answers[:q_pe_database] = 'y'
+          unless options[:type] == :upgrade
+            answers.merge! console_a
+          else
+            answers[:q_database_export_dir] = '/tmp'
+          end
+        end
+
+        if host == database
+          if database != master
+            if options[:type] == :upgrade
+              # This is kinda annoying - if we're upgrading to 3.0 and are
+              # puppetdb, we're actually doing a clean install. We thus
+              # need the core agent answers.
+              answers.merge! agent_a
+            end
+            answers.merge! master_database_a
+          end
+          answers.merge! database_a
+          answers.merge! console_database_a
+        end
+
+        if host == master and host == database and host == dashboard
+          answers[:q_all_in_one_install] = 'y'
+        end
+
+        if host['platform'].include? 'aix'
+          answers.merge! aix_a
+        end
+
+        return answers
+      end
+
+      def self.answers(hosts, master_certname, options)
+        the_answers = {}
+        database = only_host_with_role(hosts, 'database')
+        dashboard = only_host_with_role(hosts, 'dashboard')
+        master = only_host_with_role(hosts, 'master')
+        hosts.each do |h|
+          if options[:type] == :upgrade and options[:from] =~ /\A3.0/
+            # 3.0.x to 3.0.x should require no answers
+            the_answers[h.name] = {
+              :q_install => 'y',
+              :q_install_vendor_packages => 'y',
+            }
+          else
+            the_answers[h.name] = host_answers(h, master_certname, master, database, dashboard, options)
+          end
+        end
+        return the_answers
+      end
+
+    end
+  end
+end