spec_helper_acceptance.rb

require 'beaker-rspec'
require 'beaker/puppet_install_helper'

run_puppet_install_helper

RSpec.configure do |c|
  # Project root
  proj_root = File.expand_path(File.join(File.dirname(__FILE__), '..'))

  # Readable test descriptions
  c.formatter = :documentation

  # Configure all nodes in nodeset
  c.before :suite do
    # Install module and dependencies
    puppet_module_install(:source => proj_root, :module_name => 'haproxy')
    hosts.each do |host|
      on host, puppet('module','install','puppetlabs-stdlib'), { :acceptable_exit_codes => [0,1] }
      on host, puppet('module','install','puppetlabs-concat'), { :acceptable_exit_codes => [0,1] }
      if fact('osfamily') == 'RedHat'
        on host, puppet('module','install','stahnma/epel'), { :acceptable_exit_codes => [0,1] }
      end
      if fact('operatingsystem') == 'Debian'
        on host, puppet('module','install','puppetlabs-apt'), { :acceptable_exit_codes => [0,1] }
        apply_manifest(%{
          include apt
          include apt::backports
        })
      end
      pp = <<-EOS
        package { 'socat': ensure => present, }
        package { 'screen': ensure => present, }
        if $::osfamily == 'RedHat' {
          class { 'epel': before => Package['socat'], }
          service { 'iptables': ensure => stopped, }
          exec { 'setenforce 0':
            path   => ['/bin','/usr/bin','/sbin','/usr/sbin'],
            onlyif => 'which getenforce && getenforce | grep Enforcing',
          }
          if $::operatingsystemmajrelease == '7' {
            # For `netstat` for serverspec
            package { 'net-tools': ensure => present, }
          }
        }
      EOS
      apply_manifest(pp, :catch_failures => true)

      ['5556','5557'].each do |port|
        content = "socat -v tcp-l:#{port},reuseaddr,fork system:\"printf \\'HTTP/1.1 200 OK\r\n\r\nResponse on #{port}\\'\",nofork"
        create_remote_file(host, "/root/script-#{port}.sh", content)
        shell(%{/usr/bin/screen -dmS script-#{port} sh /root/script-#{port}.sh})
        sleep 1
        shell(%{netstat -tnl|grep ':#{port}'})
        end
    end
  end

  # FM-5470, this was added to reset failed count and work around puppet 3.x
  if ( (fact('operatingsystem') == 'SLES' and fact('operatingsystemmajrelease') == '12') or (fact('osfamily') == 'RedHat' and fact('operatingsystemmajrelease') == '7') )
    c.after :each do
      # not all tests have a haproxy service, so the systemctl call can fail,
      # but we don't care as we only need to reset when it does.
      shell('systemctl reset-failed haproxy.service || true')
    end
  end
end