101 lines
3.1 KiB
Ruby
101 lines
3.1 KiB
Ruby
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
|