beaker-puppet 0.10.0 → 0.11.0

data/setup/git/010_TestSetup.rb

@@ -0,0 +1,103 @@
+test_name "Install packages and repositories on target machines..." do
+  extend Beaker::DSL::InstallUtils
+
+  SourcePath  = Beaker::DSL::InstallUtils::SourcePath
+  GitHubSig   = Beaker::DSL::InstallUtils::GitHubSig
+
+  repositories = options[:install].map do |url|
+    extract_repo_info_from(build_git_url(url))
+  end
+
+  hosts.each_with_index do |host, index|
+    on host, "echo #{GitHubSig} >> $HOME/.ssh/known_hosts"
+
+    repositories.each do |repository|
+      step "Install #{repository[:name]}"
+      if repository[:path] =~ /^file:\/\/(.+)$/
+        on host, "test -d #{SourcePath} || mkdir -p #{SourcePath}"
+        source_dir = $1
+        checkout_dir = "#{SourcePath}/#{repository[:name]}"
+        on host, "rm -f #{checkout_dir}" # just the symlink, do not rm -rf !
+        on host, "ln -s #{source_dir} #{checkout_dir}"
+        on host, "cd #{checkout_dir} && if [ -f install.rb ]; then ruby ./install.rb ; else true; fi"
+      else
+        puppet_dir = host.tmpdir('puppet')
+        on(host, "chmod 755 #{puppet_dir}")
+
+        sha = ENV['SHA'] || `git rev-parse HEAD`.chomp
+        gem_source = ENV["GEM_SOURCE"] || "https://rubygems.org"
+        gemfile_contents = <<END
+source '#{gem_source}'
+gem '#{repository[:name]}', :git => '#{repository[:path]}', :ref => '#{sha}'
+END
+        case host['platform']
+        when /windows/
+          create_remote_file(host, "#{puppet_dir}/Gemfile", gemfile_contents)
+          # bundle must be passed a Windows style path for a binstubs location
+          bindir = host['puppetbindir'].split(':').first
+          binstubs_dir = on(host, "cygpath -m \"#{bindir}\"").stdout.chomp
+          # note passing --shebang to bundle is not useful because Cygwin
+          # already finds the Ruby interpreter OK with the standard shebang of:
+          # !/usr/bin/env ruby
+          # the problem is a Cygwin style path is passed to the interpreter and this can't be modified:
+          # http://cygwin.1069669.n5.nabble.com/Pass-windows-style-paths-to-the-interpreter-from-the-shebang-line-td43870.html
+          on host, "cd #{puppet_dir} && cmd.exe /c \"bundle install --system --binstubs '#{binstubs_dir}'\""
+          # puppet.bat isn't written by Bundler, but facter.bat is - copy this generic file
+          on host, "cd #{host['puppetbindir']} && test -f ./puppet.bat || cp ./facter.bat ./puppet.bat"
+          # to access gem / facter / puppet / bundle / irb with Cygwin generally requires aliases
+          # so that commands in /usr/bin are overridden and the binstub wrappers won't run inside Cygwin
+          # but rather will execute as batch files through cmd.exe
+          # without being overridden, Cygwin reads the shebang and causes errors like:
+          # C:\cygwin64\bin\ruby.exe: No such file or directory -- /usr/bin/puppet (LoadError)
+          # NOTE /usr/bin/puppet is a Cygwin style path that our custom Ruby build
+          # does not understand - it expects a standard Windows path like c:\cygwin64\bin\puppet
+
+          # a workaround in interactive SSH is to add aliases to local session / .bashrc:
+          #   on host, "echo \"alias puppet='C:/\\cygwin64/\\bin/\\puppet.bat'\" >> ~/.bashrc"
+          # note that this WILL NOT impact Beaker runs though
+          puppet_bundler_install_dir = on(host, "cd #{puppet_dir} && cmd.exe /c bundle show puppet").stdout.chomp
+        when /el-7/
+          create_remote_file(host, "#{puppet_dir}/Gemfile", gemfile_contents + "gem 'json'\n")
+          on host, "cd #{puppet_dir} && bundle install --system --binstubs #{host['puppetbindir']}"
+          puppet_bundler_install_dir = on(host, "cd #{puppet_dir} && bundle show puppet").stdout.chomp
+        when /solaris/
+          create_remote_file(host, "#{puppet_dir}/Gemfile", gemfile_contents)
+          on host, "cd #{puppet_dir} && bundle install --system --binstubs #{host['puppetbindir']} --shebang #{host['puppetbindir']}/ruby"
+          puppet_bundler_install_dir = on(host, "cd #{puppet_dir} && bundle show puppet").stdout.chomp
+        else
+          create_remote_file(host, "#{puppet_dir}/Gemfile", gemfile_contents)
+          on host, "cd #{puppet_dir} && bundle install --system --binstubs #{host['puppetbindir']}"
+          puppet_bundler_install_dir = on(host, "cd #{puppet_dir} && bundle show puppet").stdout.chomp
+        end
+
+        # install.rb should also be called from the Puppet gem install dir
+        # this is required for the puppetres.dll event log dll on Windows
+        on host, "cd #{puppet_bundler_install_dir} && if [ -f install.rb ]; then ruby ./install.rb ; else true; fi"
+      end
+    end
+  end
+
+  step "Hosts: create basic puppet.conf" do
+    hosts.each do |host|
+      confdir = host.puppet['confdir']
+      on host, "mkdir -p #{confdir}"
+      puppetconf = File.join(confdir, 'puppet.conf')
+
+      if host['roles'].include?('agent')
+        on host, "echo '[agent]' > '#{puppetconf}' && " +
+                 "echo server=#{master} >> '#{puppetconf}'"
+      else
+        on host, "touch '#{puppetconf}'"
+      end
+    end
+  end
+
+  step "Hosts: create environments directory like AIO does" do
+    hosts.each do |host|
+      codedir = host.puppet['codedir']
+      on host, "mkdir -p #{codedir}/environments/production/manifests"
+      on host, "mkdir -p #{codedir}/environments/production/modules"
+      on host, "chmod -R 755 #{codedir}"
+    end
+  end
+end

data/setup/git/020_PuppetUserAndGroup.rb

@@ -0,0 +1,10 @@
+test_name 'Puppet User and Group' do
+  hosts.each do |host|
+
+    step "ensure puppet user and group added to all nodes because this is what the packages do" do
+      on host, puppet("resource user puppet ensure=present")
+      on host, puppet("resource group puppet ensure=present")
+    end
+
+  end
+end

data/setup/git/030_PuppetMasterSanity.rb

@@ -0,0 +1,18 @@
+test_name "Puppet Master sanity checks: PID file and SSL dir creation"
+
+hostname = on(master, 'facter hostname').stdout.strip
+fqdn = on(master, 'facter fqdn').stdout.strip
+
+with_puppet_running_on(master, :main => { :dns_alt_names => "puppet,#{hostname},#{fqdn}", :verbose => true, :noop => true }) do
+  # SSL dir created?
+  step "SSL dir created?"
+  on master,  "[ -d #{master.puppet('master')['ssldir']} ]"
+
+  # PID file exists?
+  step "PID file created?"
+  on master, "[ -f #{master.puppet('master')['pidfile']} ]"
+end
+
+step "Create module directories normally handled via packaging"
+on master, "mkdir -p #{master['distmoduledir']}"
+on master, "mkdir -p #{master['sitemoduledir']}"

data/setup/git/060_InstallModules.rb

@@ -0,0 +1,76 @@
+require 'pathname'
+
+# Given an array of modules specified by the --modules command line option,
+# Parse all of them into an array of usable hash structures.
+class PuppetModules
+  attr_reader :modules
+
+  def initialize(modules=[])
+    @modules = modules
+  end
+
+  def list
+    return [] unless modules
+    modules.collect do |uri|
+      git_url, git_ref = uri.split '#'
+      folder = Pathname.new(git_url).basename('.git')
+      name = folder.to_s.split('-', 2)[1] || folder.to_s
+      {
+        :name     => name,
+        :url      => git_url,
+        :folder   => folder.to_s,
+        :ref      => git_ref,
+        :protocol => git_url.split(':')[0].intern,
+      }
+    end
+  end
+end
+
+def install_git_module(mod, hosts)
+  # The idea here is that each test can symlink the modules they want from a
+  # temporary directory to this location.  This will preserve the global
+  # state of the system while allowing individual test cases to quickly run
+  # with a module "installed" in the module path.
+  moddir = "/opt/puppet-git-repos"
+  target = "#{moddir}/#{mod[:name]}"
+
+  step "Clone #{mod[:url]} if needed"
+  on hosts, "test -d #{moddir} || mkdir -p #{moddir}"
+  on hosts, "test -d #{target} || git clone #{mod[:url]} #{target}"
+  step "Update #{mod[:name]} and check out revision #{mod[:ref]}"
+
+  commands = ["cd #{target}",
+              "remote rm origin",
+              "remote add origin #{mod[:url]}",
+              "fetch origin",
+              "checkout -f #{mod[:ref]}",
+              "reset --hard refs/remotes/origin/#{mod[:ref]}",
+              "clean -fdx",
+  ]
+
+  on hosts, commands.join(" && git ")
+end
+
+def install_scp_module(mod, hosts)
+  moddir = "/opt/puppet-git-repos"
+  target = "#{moddir}/#{mod[:name]}"
+
+  step "Purge #{target} if needed"
+  on hosts, "test -d #{target} && rm -rf #{target} || true"
+
+  step "Copy #{mod[:name]} to hosts"
+  scp_to hosts, mod[:url].split(':', 2)[1], target
+end
+
+modules = PuppetModules.new(options[:modules]).list
+
+step "Masters: Install Puppet Modules"
+masters = hosts.select { |host| host['roles'].include? 'master' }
+
+modules.each do |mod|
+  if mod[:protocol] == :scp
+    install_scp_module(mod, masters)
+  else
+    install_git_module(mod, masters)
+  end
+end

data/setup/git/070_InstallCACerts.rb

@@ -0,0 +1,93 @@
+test_name "Install CA Certs"
+confine :to, :platform => 'windows'
+
+GEOTRUST_GLOBAL_CA = <<-EOM
+-----BEGIN CERTIFICATE-----
+MIIDVDCCAjygAwIBAgIDAjRWMA0GCSqGSIb3DQEBBQUAMEIxCzAJBgNVBAYTAlVT
+MRYwFAYDVQQKEw1HZW9UcnVzdCBJbmMuMRswGQYDVQQDExJHZW9UcnVzdCBHbG9i
+YWwgQ0EwHhcNMDIwNTIxMDQwMDAwWhcNMjIwNTIxMDQwMDAwWjBCMQswCQYDVQQG
+EwJVUzEWMBQGA1UEChMNR2VvVHJ1c3QgSW5jLjEbMBkGA1UEAxMSR2VvVHJ1c3Qg
+R2xvYmFsIENBMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA2swYYzD9
+9BcjGlZ+W988bDjkcbd4kdS8odhM+KhDtgPpTSEHCIjaWC9mOSm9BXiLnTjoBbdq
+fnGk5sRgprDvgOSJKA+eJdbtg/OtppHHmMlCGDUUna2YRpIuT8rxh0PBFpVXLVDv
+iS2Aelet8u5fa9IAjbkU+BQVNdnARqN7csiRv8lVK83Qlz6cJmTM386DGXHKTubU
+1XupGc1V3sjs0l44U+VcT4wt/lAjNvxm5suOpDkZALeVAjmRCw7+OC7RHQWa9k0+
+bw8HHa8sHo9gOeL6NlMTOdReJivbPagUvTLrGAMoUgRx5aszPeE4uwc2hGKceeoW
+MPRfwCvocWvk+QIDAQABo1MwUTAPBgNVHRMBAf8EBTADAQH/MB0GA1UdDgQWBBTA
+ephojYn7qwVkDBF9qn1luMrMTjAfBgNVHSMEGDAWgBTAephojYn7qwVkDBF9qn1l
+uMrMTjANBgkqhkiG9w0BAQUFAAOCAQEANeMpauUvXVSOKVCUn5kaFOSPeCpilKIn
+Z57QzxpeR+nBsqTP3UEaBU6bS+5Kb1VSsyShNwrrZHYqLizz/Tt1kL/6cdjHPTfS
+tQWVYrmm3ok9Nns4d0iXrKYgjy6myQzCsplFAMfOEVEiIuCl6rYVSAlk6l5PdPcF
+PseKUgzbFbS9bZvlxrFUaKnjaZC2mqUPuLk/IH2uSrW4nOQdtqvmlKXBx4Ot2/Un
+hw4EbNX/3aBd7YdStysVAq45pmp06drE57xNNB6pXE0zX5IJL4hmXXeXxx12E6nV
+5fEWCRE11azbJHFwLJhWC9kXtNHjUStedejV0NxPNO3CBWaAocvmMw==
+-----END CERTIFICATE-----
+EOM
+
+USERTRUST_NETWORK_CA = <<-EOM
+-----BEGIN CERTIFICATE-----
+MIIEdDCCA1ygAwIBAgIQRL4Mi1AAJLQR0zYq/mUK/TANBgkqhkiG9w0BAQUFADCB
+lzELMAkGA1UEBhMCVVMxCzAJBgNVBAgTAlVUMRcwFQYDVQQHEw5TYWx0IExha2Ug
+Q2l0eTEeMBwGA1UEChMVVGhlIFVTRVJUUlVTVCBOZXR3b3JrMSEwHwYDVQQLExho
+dHRwOi8vd3d3LnVzZXJ0cnVzdC5jb20xHzAdBgNVBAMTFlVUTi1VU0VSRmlyc3Qt
+SGFyZHdhcmUwHhcNOTkwNzA5MTgxMDQyWhcNMTkwNzA5MTgxOTIyWjCBlzELMAkG
+A1UEBhMCVVMxCzAJBgNVBAgTAlVUMRcwFQYDVQQHEw5TYWx0IExha2UgQ2l0eTEe
+MBwGA1UEChMVVGhlIFVTRVJUUlVTVCBOZXR3b3JrMSEwHwYDVQQLExhodHRwOi8v
+d3d3LnVzZXJ0cnVzdC5jb20xHzAdBgNVBAMTFlVUTi1VU0VSRmlyc3QtSGFyZHdh
+cmUwggEiMA0GCSqGSIb3DQEBAQUAA4IBDwAwggEKAoIBAQCx98M4P7Sof885glFn
+0G2f0v9Y8+efK+wNiVSZuTiZFvfgIXlIwrthdBKWHTxqctU8EGc6Oe0rE81m65UJ
+M6Rsl7HoxuzBdXmcRl6Nq9Bq/bkqVRcQVLMZ8Jr28bFdtqdt++BxF2uiiPsA3/4a
+MXcMmgF6sTLjKwEHOG7DpV4jvEWbe1DByTCP2+UretNb+zNAHqDVmBe8i4fDidNd
+oI6yqqr2jmmIBsX6iSHzCJ1pLgkzmykNRg+MzEk0sGlRvfkGzWitZky8PqxhvQqI
+DsjfPe58BEydCl5rkdbux+0ojatNh4lz0G6k0B4WixThdkQDf2Os5M1JnMWS9Ksy
+oUhbAgMBAAGjgbkwgbYwCwYDVR0PBAQDAgHGMA8GA1UdEwEB/wQFMAMBAf8wHQYD
+VR0OBBYEFKFyXyYbKJhDlV0HN9WFlp1L0sNFMEQGA1UdHwQ9MDswOaA3oDWGM2h0
+dHA6Ly9jcmwudXNlcnRydXN0LmNvbS9VVE4tVVNFUkZpcnN0LUhhcmR3YXJlLmNy
+bDAxBgNVHSUEKjAoBggrBgEFBQcDAQYIKwYBBQUHAwUGCCsGAQUFBwMGBggrBgEF
+BQcDBzANBgkqhkiG9w0BAQUFAAOCAQEARxkP3nTGmZev/K0oXnWO6y1n7k57K9cM
+//bey1WiCuFMVGWTYGufEpytXoMs61quwOQt9ABjHbjAbPLPSbtNk28Gpgoiskli
+CE7/yMgUsogWXecB5BKV5UU0s4tpvc+0hY91UZ59Ojg6FEgSxvunOxqNDYJAB+gE
+CJChicsZUN/KHAG8HQQZexB2lzvukJDKxA4fFm517zP4029bHpbj4HR3dHuKom4t
+3XbWOTCC8KucUvIqx69JXn7HaOWCgchqJ/kniCrVWFCVH/A7HFe7fRQ5YiuayZSS
+KqMiDP+JJn1fIytH1xUdqWqeUQ0qUZ6B+dQ7XnASfxAynB67nfhmqA==
+-----END CERTIFICATE-----
+EOM
+
+EQUIFAX_CA = <<-EOM
+-----BEGIN CERTIFICATE-----
+MIIDIDCCAomgAwIBAgIENd70zzANBgkqhkiG9w0BAQUFADBOMQswCQYDVQQGEwJV
+UzEQMA4GA1UEChMHRXF1aWZheDEtMCsGA1UECxMkRXF1aWZheCBTZWN1cmUgQ2Vy
+dGlmaWNhdGUgQXV0aG9yaXR5MB4XDTk4MDgyMjE2NDE1MVoXDTE4MDgyMjE2NDE1
+MVowTjELMAkGA1UEBhMCVVMxEDAOBgNVBAoTB0VxdWlmYXgxLTArBgNVBAsTJEVx
+dWlmYXggU2VjdXJlIENlcnRpZmljYXRlIEF1dGhvcml0eTCBnzANBgkqhkiG9w0B
+AQEFAAOBjQAwgYkCgYEAwV2xWGcIYu6gmi0fCG2RFGiYCh7+2gRvE4RiIcPRfM6f
+BeC4AfBONOziipUEZKzxa1NfBbPLZ4C/QgKO/t0BCezhABRP/PvwDN1Dulsr4R+A
+cJkVV5MW8Q+XarfCaCMczE1ZMKxRHjuvK9buY0V7xdlfUNLjUA86iOe/FP3gx7kC
+AwEAAaOCAQkwggEFMHAGA1UdHwRpMGcwZaBjoGGkXzBdMQswCQYDVQQGEwJVUzEQ
+MA4GA1UEChMHRXF1aWZheDEtMCsGA1UECxMkRXF1aWZheCBTZWN1cmUgQ2VydGlm
+aWNhdGUgQXV0aG9yaXR5MQ0wCwYDVQQDEwRDUkwxMBoGA1UdEAQTMBGBDzIwMTgw
+ODIyMTY0MTUxWjALBgNVHQ8EBAMCAQYwHwYDVR0jBBgwFoAUSOZo+SvSspXXR9gj
+IBBPM5iQn9QwHQYDVR0OBBYEFEjmaPkr0rKV10fYIyAQTzOYkJ/UMAwGA1UdEwQF
+MAMBAf8wGgYJKoZIhvZ9B0EABA0wCxsFVjMuMGMDAgbAMA0GCSqGSIb3DQEBBQUA
+A4GBAFjOKer89961zgK5F7WF0bnj4JXMJTENAKaSbn+2kmOeUJXRmm/kEd5jhW6Y
+7qj/WsjTVbJmcVfewCHrPSqnI0kBBIZCe/zuf6IWUrVnZ9NA2zsmWLIodz2uFHdh
+1voqZiegDfqnc1zqcPGUIWVEX/r87yloqaKHee9570+sB3c4
+-----END CERTIFICATE-----
+EOM
+
+hosts.each do |host|
+  step "Installing Geotrust CA cert"
+  create_remote_file(host, "geotrustglobal.pem", GEOTRUST_GLOBAL_CA)
+  on host, "chmod 644 geotrustglobal.pem"
+  on host, "cmd /c certutil -v -addstore Root `cygpath -w geotrustglobal.pem`"
+
+  step "Installing Usertrust Network CA cert"
+  create_remote_file(host, "usertrust-network.pem", USERTRUST_NETWORK_CA)
+  on host, "chmod 644 usertrust-network.pem"
+  on host, "cmd /c certutil -v -addstore Root `cygpath -w usertrust-network.pem`"
+
+  step "Installing Equifax CA cert"
+  create_remote_file(host, "equifax.pem", EQUIFAX_CA)
+  on host, "chmod 644 equifax.pem"
+  on host, "cmd /c certutil -v -addstore Root `cygpath -w equifax.pem`"
+end

data/tasks/ci.rake

@@ -0,0 +1,233 @@
+require 'rake/clean'
+require 'pp'
+require 'yaml'
+require 'securerandom'
+require 'fileutils'
+require 'beaker-hostgenerator'
+require 'beaker/dsl/install_utils'
+extend Beaker::DSL::InstallUtils
+
+REPO_CONFIGS_DIR = 'repo-configs'
+CLEAN.include('*.tar', REPO_CONFIGS_DIR, 'tmp', '.beaker')
+
+# Default test target if none specified
+DEFAULT_MASTER_TEST_TARGET = 'redhat7-64m'
+DEFAULT_TEST_TARGETS = "#{DEFAULT_MASTER_TEST_TARGET}a-windows2012r2-64a"
+
+USAGE = <<-EOS
+Usage: bundle exec rake <target> [arguments]
+
+where <target> is one of:
+
+  ci:test:git
+  ci:test:aio
+  ci:test:gem
+  ci:test:quick
+
+See `bundle exec rake -D <target>` for information about each target. Each rake
+task accepts the following environment variables:
+
+REQUIRED
+--------
+
+SHA:
+    The git SHA of either the puppet-agent package or component repo must be
+    specified depending on the target. The `git` and `gem` use the component
+    SHA, where as `aio` and `quick` use the puppet-agent package SHA.
+
+OPTIONAL
+--------
+
+HOSTS:
+    The hosts to run against. Must be specified as a path to a file or a
+    beaker-hostgenerator string. Defaults to #{DEFAULT_TEST_TARGETS}.
+
+    HOSTS=mynodes.yaml
+    HOSTS=redhat7-64ma
+
+TESTS:
+    A comma-delimited list of files/directories containing tests to run.
+    Defaults to running all tests, except for `ci:test:quick` which defaults
+    to running a predefined set of tests.
+
+    TESTS=tests/cycle_detection.rb
+    TESTS=tests/pluginsync,tests/language
+
+OPTIONS:
+    Additional options to pass to beaker. Defaults to ''.
+
+    OPTIONS='--dry-run --no-color'
+
+BEAKER_HOSTS:
+    Same as HOSTS.
+
+TEST_TARGET:
+    A beaker-hostgenerator string of agent-only hosts to run against.
+    The MASTER_TEST_TARGET host will be added this list. This option is
+    only intended to be used in CI.
+
+MASTER_TEST_TARGET:
+    Override the default master test target. Should only be used with
+    TEST_TARGET, and is only intended to be used in CI. Defaults to
+    #{DEFAULT_MASTER_TEST_TARGET}.
+EOS
+
+namespace :ci do
+  desc "Print usage information"
+  task :help do
+    puts USAGE
+    exit 1
+  end
+
+  task :check_env do
+    if ENV['SHA'].nil?
+      puts "Error: A SHA must be specified"
+      puts "\n"
+      puts USAGE
+      exit 1
+    end
+
+    if ENV['TESTS'].nil?
+      ENV['TESTS'] ||= ENV['TEST']
+      ENV['TESTS'] ||= 'tests'
+    end
+  end
+
+  task :gen_hosts do
+    hosts =
+      if ENV['HOSTS']
+        ENV['HOSTS']
+      elsif ENV['BEAKER_HOSTS']
+        ENV['BEAKER_HOSTS']
+      elsif env_config = ENV['CONFIG']
+        puts 'Warning: environment variable CONFIG deprecated. Please use HOSTS to match beaker options.'
+        env_config
+      else
+        # By default we assume TEST_TARGET is an agent-only string
+        if agent_target = ENV['TEST_TARGET']
+          master_target = ENV['MASTER_TEST_TARGET'] || DEFAULT_MASTER_TEST_TARGET
+          "#{master_target}-#{agent_target}"
+        else
+          DEFAULT_TEST_TARGETS
+        end
+      end
+
+    if File.exists?(hosts)
+      ENV['HOSTS'] = hosts
+    else
+      hosts_file = "tmp/#{hosts}-#{SecureRandom.uuid}.yaml"
+      cli = BeakerHostGenerator::CLI.new([hosts, '--disable-default-role', '--osinfo-version', '1'])
+      FileUtils.mkdir_p('tmp') # -p ignores when dir already exists
+      File.open(hosts_file, 'w') do |fh|
+        fh.print(cli.execute)
+      end
+      ENV['HOSTS'] = hosts_file
+    end
+  end
+
+  namespace :test do
+    desc <<-EOS
+Run the acceptance tests using puppet-agent (AIO) packages.
+
+  $ SHA=<full sha> bundle exec rake ci:test:aio
+
+SHA should be the full SHA for the puppet-agent package.
+EOS
+    task :aio => ['ci:check_env', 'ci:gen_hosts'] do
+      beaker_suite(:aio)
+    end
+
+    desc <<-EOS
+Run the acceptance tests against puppet gem on various platforms, performing a
+basic smoke test.
+
+  $ SHA=<full sha> bundle exec rake:ci:gem
+
+SHA should be the full SHA for the component.
+EOS
+    task :gem => ['ci:check_env'] do
+      beaker(:init, '--hosts', 'config/nodes/gem.yaml', '--options-file', 'config/gem/options.rb')
+      beaker(:provision)
+      beaker(:exec, 'pre-suite', '--pre-suite', pre_suites(:gem))
+      beaker(:exec, "#{File.dirname(__dir__)}/setup/gem/010_GemInstall.rb")
+      beaker(:destroy)
+    end
+
+    desc <<-EOS
+Run the acceptance tests against a git checkout.
+
+  $ SHA=<full sha> bundle exec rake ci:test:git
+
+SHA should be the full SHA for the component. Other options:
+
+FORK: to test against your fork, defaults to 'puppetlabs'
+
+SERVER: to git fetch from an alternate GIT server, defaults to 'github.com'
+EOS
+    task :git => ['ci:check_env', 'ci:gen_hosts'] do
+      beaker_suite(:git)
+    end
+  end
+
+  task :test_and_preserve_hosts => ['ci:check_env', 'ci:gen_hosts'] do
+    puts "WARNING, the test_and_preserve_hosts task is deprecated, use ci:test:aio instead."
+    Rake::Task['ci:test:aio'].execute
+  end
+end
+
+task :default do
+  sh('rake -T')
+end
+
+task :spec do
+  sh('rspec lib')
+end
+
+def beaker(command, *argv)
+  argv.concat(ENV['OPTIONS'].split(' ')) if ENV['OPTIONS']
+
+  sh('beaker', command.to_s, *argv)
+end
+
+def beaker_suite(type)
+  beaker(:init, '--hosts', ENV['HOSTS'], '--options-file', "config/#{String(type)}/options.rb")
+  beaker(:provision)
+  beaker(:exec, 'pre-suite', '--pre-suite', pre_suites(type))
+  beaker(:exec, ENV['TESTS'])
+  beaker(:destroy)
+end
+
+def pre_suites(type)
+  beaker_root = File.dirname(__dir__)
+  presuites = case type
+  when :aio
+    [
+      "#{beaker_root}/setup/common/000-delete-puppet-when-none.rb",
+      "#{beaker_root}/setup/aio/010_Install.rb",
+      "#{beaker_root}/setup/aio/020_InstallCumulusModules.rb",
+      "#{beaker_root}/setup/aio/021_InstallAristaModule.rb",
+      "#{beaker_root}/setup/common/025_StopFirewall.rb",
+      "#{beaker_root}/setup/common/030_StopSssd.rb",
+      "#{beaker_root}/setup/common/040_ValidateSignCert.rb",
+      "#{beaker_root}/setup/aio/045_EnsureMasterStarted.rb",
+    ]
+  when :gem
+    [
+      "#{beaker_root}/setup/common/000-delete-puppet-when-none.rb",
+      "#{beaker_root}/setup/git/000_EnvSetup.rb",
+    ]
+  when :git
+    [
+      "#{beaker_root}/setup/common/000-delete-puppet-when-none.rb",
+      "#{beaker_root}/setup/git/000_EnvSetup.rb",
+      "#{beaker_root}/setup/git/010_TestSetup.rb",
+      "#{beaker_root}/setup/git/020_PuppetUserAndGroup.rb",
+      "#{beaker_root}/setup/common/025_StopFirewall.rb",
+      "#{beaker_root}/setup/git/030_PuppetMasterSanity.rb",
+      "#{beaker_root}/setup/common/040_ValidateSignCert.rb",
+      "#{beaker_root}/setup/git/060_InstallModules.rb",
+      "#{beaker_root}/setup/git/070_InstallCACerts.rb",
+    ]
+  end
+  presuites.join(',')
+end