beaker 0.0.0

data/lib/beaker/host/unix.rb

@@ -0,0 +1,54 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'host'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'command_factory'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'command'))
+
+module Unix
+  class Host < Beaker::Host
+    require File.expand_path(File.join(File.dirname(__FILE__), 'unix', 'user'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'unix', 'group'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'unix', 'exec'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'unix', 'pkg'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'unix', 'file'))
+
+    include Unix::User
+    include Unix::Group
+    include Unix::File
+    include Unix::Exec
+    include Unix::Pkg
+
+    def self.pe_defaults
+      {
+        'user'          => 'root',
+        'group'         => 'pe-puppet',
+        'puppetpath'    => '/etc/puppetlabs/puppet',
+        'puppetbin'     => '/opt/puppet/bin/puppet',
+        'puppetbindir'  => '/opt/puppet/bin',
+        'puppetvardir'  => '/var/opt/lib/pe-puppet',
+        'hieradatadir'  => '/var/lib/hiera',
+        'hieraconf'     => '/etc/puppetlabs/puppet/hiera.yaml',
+        'distmoduledir' => '/etc/puppetlabs/puppet/modules',
+        'sitemoduledir' => '/opt/puppet/share/puppet/modules',
+        'pathseparator' => ':',
+      }
+    end
+
+    def self.foss_defaults
+      {
+        'user'              => 'root',
+        'group'             => 'puppet',
+        'puppetpath'        => '/etc/puppet',
+        'puppetvardir'      => '/var/lib/puppet',
+        'puppetbin'         => '/usr/bin/puppet',
+        'puppetbindir'      => '/usr/bin',
+        'hieralibdir'       => '/opt/puppet-git-repos/hiera/lib',
+        'hierapuppetlibdir' => '/opt/puppet-git-repos/hiera-puppet/lib',
+        'hierabindir'       => '/opt/puppet-git-repos/hiera/bin',
+        'hieradatadir'      => '/etc/puppet/hieradata',
+        'hieraconf'         => '/etc/puppet/hiera.yaml',
+        'distmoduledir'     => '/etc/puppet/modules',
+        'sitemoduledir'     => '/usr/share/puppet/modules',
+        'pathseparator'     => ':',
+      }
+    end
+  end
+end

data/lib/beaker/host/unix/exec.rb

@@ -0,0 +1,15 @@
+module Unix::Exec
+  include Beaker::CommandFactory
+
+  def echo(msg, abs=true)
+    (abs ? '/bin/echo' : 'echo') + " #{msg}"
+  end
+
+  def touch(file, abs=true)
+    (abs ? '/bin/touch' : 'touch') + " #{file}"
+  end
+
+  def path
+    '/bin:/usr/bin'
+  end
+end

data/lib/beaker/host/unix/file.rb

@@ -0,0 +1,16 @@
+module Unix::File
+  include Beaker::CommandFactory
+
+  def tmpfile(name)
+    execute("mktemp -t #{name}.XXXXXX")
+  end
+
+  def tmpdir(name)
+    execute("mktemp -td #{name}.XXXXXX")
+  end
+
+  def path_split(paths)
+    paths.split(':')
+  end
+
+end

data/lib/beaker/host/unix/group.rb

@@ -0,0 +1,40 @@
+module Unix::Group
+  include Beaker::CommandFactory
+
+  def group_list(&block)
+    execute("getent group") do |result|
+      groups = []
+      result.stdout.each_line do |line|
+        groups << (line.match(/^([^:]+)/) or next)[1]
+      end
+
+      yield result if block_given?
+
+      groups
+    end
+  end
+
+  def group_get(name, &block)
+    execute("getent group #{name}") do |result|
+      fail_test "failed to get group #{name}" unless result.stdout =~ /^#{name}:.*:[0-9]+:/
+
+      yield result if block_given?
+    end
+  end
+
+  def group_gid(name)
+    execute("getent group #{name}") do |result|
+      # Format is:
+      # wheel:x:10:root
+      result.stdout.split(':')[2]
+    end
+  end
+
+  def group_present(name, &block)
+    execute("if ! getent group #{name}; then groupadd #{name}; fi", {}, &block)
+  end
+
+  def group_absent(name, &block)
+    execute("if getent group #{name}; then groupdel #{name}; fi", {}, &block)
+  end
+end

data/lib/beaker/host/unix/pkg.rb

@@ -0,0 +1,22 @@
+module Unix::Pkg
+  include Beaker::CommandFactory
+
+  def check_for_package name
+    result = exec(Beaker::Command.new("which #{name}"), :acceptable_exit_codes => (0...127))
+    result.exit_code == 0
+  end
+
+  def install_package name
+    if self['platform'] =~ /el-4/
+      @logger.debug("Package installation not supported on rhel4")
+    elsif self['platform'] =~ /(fedora)|(centos)|(el)/
+      execute("yum -y install #{name}")
+    elsif self['platform'] =~ /(ubuntu)|(debian)/
+      execute("apt-get update")
+      execute("apt-get install -y #{name}")
+    else
+      raise "Package #{name} cannot be installed on #{host}"
+    end
+  end
+
+end

data/lib/beaker/host/unix/user.rb

@@ -0,0 +1,32 @@
+module Unix::User
+  include Beaker::CommandFactory
+
+  def user_list(&block)
+    execute("getent passwd") do |result|
+      users = []
+      result.stdout.each_line do |line|
+        users << (line.match( /^([^:]+)/) or next)[1]
+      end
+
+      yield result if block_given?
+
+      users
+    end
+  end
+
+  def user_get(name, &block)
+    execute("getent passwd #{name}") do |result|
+      fail_test "failed to get user #{name}" unless result.stdout =~  /^#{name}:/
+
+      yield result if block_given?
+    end
+  end
+
+  def user_present(name, &block)
+    execute("if ! getent passwd #{name}; then useradd #{name}; fi", {}, &block)
+  end
+
+  def user_absent(name, &block)
+    execute("if getent passwd #{name}; then userdel #{name}; fi", {}, &block)
+  end
+end

data/lib/beaker/host/windows.rb

@@ -0,0 +1,44 @@
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'host'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'command_factory'))
+require File.expand_path(File.join(File.dirname(__FILE__), '..', 'command'))
+
+module Windows
+  class Host < Beaker::Host
+    require File.expand_path(File.join(File.dirname(__FILE__), 'windows', 'user'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'windows', 'group'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'windows', 'exec'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'windows', 'pkg'))
+    require File.expand_path(File.join(File.dirname(__FILE__), 'windows', 'file'))
+
+    include Windows::User
+    include Windows::Group
+    include Windows::File
+    include Windows::Exec
+    include Windows::Pkg
+
+    def self.pe_defaults
+      {
+        'user'          => 'Administrator',
+        'group'         => 'Administrators',
+        'puppetpath'    => '`cygpath -smF 35`/PuppetLabs/puppet/etc',
+        'puppetvardir'  => '`cygpath -smF 35`/PuppetLabs/puppet/var',
+        'puppetbindir'  => '`cygpath -F 38`/Puppet Labs/Puppet Enterprise/bin',
+        'pathseparator' => ';',
+      }
+    end
+
+    def self.foss_defaults
+      {
+        'user'              => 'Administrator',
+        'group'             => 'Administrators',
+        'puppetpath'        => '`cygpath -smF 35`/PuppetLabs/puppet/etc',
+        'puppetvardir'      => '`cygpath -smF 35`/PuppetLabs/puppet/var',
+        'hieralibdir'       => '`cygpath -w /opt/puppet-git-repos/hiera/lib`',
+        'hierapuppetlibdir' => '`cygpath -w /opt/puppet-git-repos/hiera-puppet/lib`',
+        # PATH related variables need to be Unix, which cygwin converts
+        'hierabindir'       => '/opt/puppet-git-repos/hiera/bin',
+        'pathseparator'     => ';',
+      }
+    end
+  end
+end

data/lib/beaker/host/windows/exec.rb

@@ -0,0 +1,18 @@
+module Windows::Exec
+  include Beaker::CommandFactory
+
+  ABS_CMD = 'c:\\\\windows\\\\system32\\\\cmd.exe'
+  CMD = 'cmd.exe'
+
+  def echo(msg, abs=true)
+    (abs ? ABS_CMD : CMD) + " /c echo #{msg}"
+  end
+
+  def touch(file, abs=true)
+    (abs ? ABS_CMD : CMD) + " /c echo. 2> #{file}"
+  end
+
+  def path
+    'c:/windows/system32;c:/windows'
+  end
+end

data/lib/beaker/host/windows/file.rb

@@ -0,0 +1,15 @@
+module Windows::File
+  include Beaker::CommandFactory
+
+  def tmpfile(name)
+    execute("cygpath -m $(mktemp -t #{name}.XXXXXX)")
+  end
+
+  def tmpdir(name)
+    execute("cygpath -m $(mktemp -td #{name}.XXXXXX)")
+  end
+
+  def path_split(paths)
+    paths.split(';')
+  end
+end

data/lib/beaker/host/windows/group.rb

@@ -0,0 +1,36 @@
+module Windows::Group
+  include Beaker::CommandFactory
+
+  def group_list(&block)
+    execute('cmd /c echo "" | wmic group where localaccount="true" get name /format:value') do |result|
+      groups = []
+      result.stdout.each_line do |line|
+        groups << (line.match(/^Name=([\w ]+)/) or next)[1]
+      end
+
+      yield result if block_given?
+
+      groups
+    end
+  end
+
+  def group_get(name, &block)
+    execute("net localgroup \"#{name}\"") do |result|
+      fail_test "failed to get group #{name}" unless result.stdout =~ /^Alias name\s+#{name}/
+
+      yield result if block_given?
+    end
+  end
+
+  def group_gid(name)
+    raise NotImplementedError, "Can't retrieve group gid on a Windows host"
+  end
+
+  def group_present(name, &block)
+    execute("net localgroup /add \"#{name}\"", {:acceptable_exit_codes => [0,2]}, &block)
+  end
+
+  def group_absent(name, &block)
+    execute("net localgroup /delete \"#{name}\"", {:acceptable_exit_codes => [0,2]}, &block)
+  end
+end

data/lib/beaker/host/windows/pkg.rb

@@ -0,0 +1,26 @@
+module Windows::Pkg
+  include Beaker::CommandFactory
+
+  def check_for_package name
+    result = exec(Beaker::Command.new("which #{name}"), :acceptable_exit_codes => (0...127))
+    result.exit_code == 0 
+  end
+
+  def install_package name
+    cygwin = ""
+    rootdir = ""
+    result = exec(Beaker::Command.new("wmic os get osarchitecture | grep 32"), :acceptable_exit_codes => (0...127))
+    if result.exit_code == 0 #32 bit version
+      rootdir = "c:\\\\cygwin"
+      cygwin = "setup-x86.exe"
+    else  #64 bit version
+      rootdir = "c:\\\\cygwin64"
+      cygwin = "setup-x86_64.exe"
+    end
+    if not check_for_package(cygwin)
+      execute("curl --retry 5 http://cygwin.com/#{cygwin} -o /cygdrive/c/Windows/System32/#{cygwin}")
+    end
+    execute("#{cygwin} -q -n -N -d -R #{rootdir} -s http://cygwin.osuosl.org -P #{name}") 
+  end
+
+end

data/lib/beaker/host/windows/user.rb

@@ -0,0 +1,32 @@
+module Windows::User
+  include Beaker::CommandFactory
+
+  def user_list(&block)
+    execute('cmd /c echo "" | wmic useraccount where localaccount="true" get name /format:value') do |result|
+      users = []
+      result.stdout.each_line do |line|
+        users << (line.match(/^Name=([\w ]+)/) or next)[1]
+      end
+
+      yield result if block_given?
+
+      users
+    end
+  end
+
+  def user_get(name, &block)
+    execute("net user \"#{name}\"") do |result|
+      fail_test "failed to get user #{name}" unless result.stdout =~ /^User name\s+#{name}/
+
+      yield result if block_given?
+    end
+  end
+
+  def user_present(name, &block)
+    execute("net user /add \"#{name}\"", {:acceptable_exit_codes => [0,2]}, &block)
+  end
+
+  def user_absent(name, &block)
+    execute("net user /delete \"#{name}\"", {:acceptable_exit_codes => [0,2]}, &block)
+  end
+end

data/lib/beaker/hypervisor.rb

@@ -0,0 +1,37 @@
+module Beaker
+  class Hypervisor
+
+    def configure(hosts)
+      @logger.debug "No post-provisioning configuration necessary for #{self.class.name} boxes"
+    end
+
+    def self.create type, hosts_to_provision, options, config
+      @logger = options[:logger]
+      @logger.notify("Beaker::Hypervisor, found some #{type} boxes to create") 
+      case type
+        when /aix/
+          Beaker::Aixer.new hosts_to_provision, options, config
+        when /solaris/
+          Beaker::Solaris.new hosts_to_provision, options, config
+        when /vsphere/
+          Beaker::Vsphere.new hosts_to_provision, options, config
+        when /fusion/
+          Beaker::Fusion.new hosts_to_provision, options, config
+        when /blimpy/
+          Beaker::Blimper.new hosts_to_provision, options, config
+        when /vcloud/
+          Beaker::Vcloud.new hosts_to_provision, options, config
+        when /vagrant/
+          Beaker::Vagrant.new hosts_to_provision, options, config
+        end
+    end
+  end
+end
+
+%w( vsphere_helper vagrant fusion blimper vsphere vcloud aixer solaris).each do |lib|
+  begin
+    require "hypervisor/#{lib}"
+  rescue LoadError
+    require File.expand_path(File.join(File.dirname(__FILE__), "hypervisor", lib))
+  end
+end

data/lib/beaker/hypervisor/aixer.rb

@@ -0,0 +1,52 @@
+module Beaker 
+  class Aixer < Beaker::Hypervisor
+
+    def initialize(aix_hosts, options, config)
+      @options = options
+      @config = config['CONFIG'].dup
+      @logger = options[:logger]
+      @aix_hosts = aix_hosts
+      #aix machines are reverted to known state, not a snapshot
+      fog_file = nil
+      if File.exists?( File.join(ENV['HOME'], '.fog') )
+        fog_file = YAML.load_file( File.join(ENV['HOME'], '.fog') )
+      end
+      raise "Cant load ~/.fog config" unless fog_file
+
+      # Running the rake task on rpm-builder
+      hypername = fog_file[:default][:aix_hypervisor_server]
+      hyperconf = {
+        'HOSTS'  => {
+          hypername => { 'platform' => 'el-6-x86_64' }
+        },
+        'CONFIG' => {
+          'user' => fog_file[:default][:aix_hypervisor_username] || ENV['USER'],
+          'ssh'  => {
+            :keys => fog_file[:default][:aix_hypervisor_keyfile] || "#{ENV['HOME']}/.ssh/id_rsa"
+          }
+        }
+      }
+      hyperconfig = Beaker::TestConfig.new( hyperconf, @options )
+
+      @logger.notify "Connecting to hypervisor at #{hypername}"
+      hypervisor = Beaker::Host.create( hypername, @options, hyperconfig )
+
+      @aix_hosts.each do |host|
+        vm_name = host['vmname'] || host.name
+
+        @logger.notify "Reverting #{vm_name} to aix clean state"
+        start = Time.now
+        # Restore AIX image, ID'd by the hostname
+        hypervisor.exec(Command.new("cd pe-aix && rake restore:#{host.name}"))
+        time = Time.now - start
+        @logger.notify "Spent %.2f seconds reverting" % time
+      end
+      hypervisor.close
+    end
+
+    def cleanup
+      @logger.notify "No cleanup for aix boxes"
+    end
+
+  end
+end