123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101 |
- require 'beaker-rspec/spec_helper'
- require 'beaker-rspec/helpers/serverspec'
- require 'beaker/puppet_install_helper'
- run_puppet_install_helper
- UNSUPPORTED_PLATFORMS = ['AIX','windows','Solaris','Suse']
- class String
- # Provide ability to remove indentation from strings, for the purpose of
- # left justifying heredoc blocks.
- def unindent
- gsub(/^#{scan(/^\s*/).min_by{|l|l.length}}/, "")
- end
- end
- def shellescape(str)
- str = str.to_s
- # An empty argument will be skipped, so return empty quotes.
- return "''" if str.empty?
- str = str.dup
- # Treat multibyte characters as is. It is caller's responsibility
- # to encode the string in the right encoding for the shell
- # environment.
- str.gsub!(/([^A-Za-z0-9_\-.,:\/@\n])/, "\\\\\\1")
- # A LF cannot be escaped with a backslash because a backslash + LF
- # combo is regarded as line continuation and simply ignored.
- str.gsub!(/\n/, "'\n'")
- return str
- end
- def psql(psql_cmd, user = 'postgres', exit_codes = [0,1], &block)
- psql = "psql #{psql_cmd}"
- shell("su #{shellescape(user)} -c #{shellescape(psql)}", :acceptable_exit_codes => exit_codes, &block)
- end
- 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 => 'postgresql')
- # Set up selinux if appropriate.
- if fact('osfamily') == 'RedHat' && fact('selinux') == 'true'
- pp = <<-EOS
- if $::osfamily == 'RedHat' and $::selinux == 'true' {
- $semanage_package = $::operatingsystemmajrelease ? {
- '5' => 'policycoreutils',
- default => 'policycoreutils-python',
- }
- package { $semanage_package: ensure => installed }
- exec { 'set_postgres':
- command => 'semanage port -a -t postgresql_port_t -p tcp 5433',
- path => '/bin:/usr/bin/:/sbin:/usr/sbin',
- subscribe => Package[$semanage_package],
- }
- }
- EOS
- apply_manifest_on(agents, pp, :catch_failures => false)
- end
- # net-tools required for netstat utility being used by be_listening
- if fact('osfamily') == 'RedHat' && fact('operatingsystemmajrelease') == '7'
- pp = <<-EOS
- package { 'net-tools': ensure => installed }
- EOS
- apply_manifest_on(agents, pp, :catch_failures => false)
- end
- hosts.each do |host|
- on host, "/bin/touch #{default['puppetpath']}/hiera.yaml"
- on host, 'chmod 755 /root'
- if fact_on(host, 'osfamily') == 'Debian'
- on host, "echo \"en_US ISO-8859-1\nen_NG.UTF-8 UTF-8\nen_US.UTF-8 UTF-8\n\" > /etc/locale.gen"
- on host, '/usr/sbin/locale-gen'
- on host, '/usr/sbin/update-locale'
- end
- on host, puppet('module','install','puppetlabs-stdlib'), { :acceptable_exit_codes => [0,1] }
- on host, puppet('module','install','puppetlabs-apt'), { :acceptable_exit_codes => [0,1] }
- on host, puppet('module','install','--force','puppetlabs-concat'), { :acceptable_exit_codes => [0,1] }
- end
- end
- end
|