Module: Beaker::DSL::Helpers::HostHelpers

Included in:
Beaker::DSL::Helpers
Defined in:
lib/beaker/dsl/helpers/host_helpers.rb

Overview

Methods that help you interact and manage the state of your Beaker SUTs, these methods do not require puppet to be installed to execute correctly

Instance Method Summary collapse

Instance Method Details

#add_system32_hosts_entry(host, opts = {}) ⇒ Object

Configure a host entry on the give host @example: will add a host entry for forge.puppetlabs.com

add_system32_hosts_entry(host, { :ip => '23.251.154.122', :name => 'forge.puppetlabs.com' })

Returns:

  • nil



432
433
434
435
436
437
438
439
440
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 432

def add_system32_hosts_entry(host, opts = {})
  if host.is_powershell?
    hosts_file = 'C:\Windows\System32\Drivers\etc\hosts'
    host_entry = "#{opts['ip']}`t`t#{opts['name']}"
    on host, powershell("\$text = \\\"#{host_entry}\\\"; Add-Content -path '#{hosts_file}' -value \$text")
  else
    raise "nothing to do for #{host.name} on #{host['platform']}"
  end
end

#archive_file_from(host, from_path, opts = {}, archive_root = 'archive/sut-files', archive_name = 'sut-files.tgz') ⇒ Result

Copy a remote file to the local system and save it under a directory meant for storing SUT files to be viewed in the event of test failures.

Files are stored locally with the following structure:

./<archive_root>/<hostname>/<from_path>

This can be used during the post-suite phase to persist relevant log files from the SUTs so they can available with the test results (without having to preserve the SUT host and SSH in after the fact).

Example

Archive the Puppet Server log file from the master ('abc123'),
and archive the Puppet Agent log file from the agent ('xyz098'):

  archive_file_from(master, '/var/log/puppetlabs/puppetserver.log')
  archive_file_from(agent, '/var/log/puppetlabs/puppetagent.log')

Results in the following files on the test runner:

  archive/sut-files/abc123/var/log/puppetlabs/puppetserver.log
  archive/sut-files/xyz098/var/log/puppetlabs/puppetagent.log

Parameters:

  • host (Host) —

    A host object (or some object that can be passed to #scp_from)

  • from_path (String) —

    A remote absolute path on the host to copy.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

  • [String] (Hash) —

    a customizable set of options

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    Returns the result of the #scp_from operation.



262
263
264
265
266
267
268
269
270
271
272
273
274
275
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 262

def archive_file_from(host, from_path, opts = {}, archive_root = 'archive/sut-files', archive_name = 'sut-files.tgz')
  require 'minitar'
  filedir = File.dirname(from_path)
  targetdir = File.join(archive_root, host.hostname, filedir)
  # full path to check for existance later
  filename = "#{targetdir}/" + File.basename(from_path)
  FileUtils.mkdir_p(targetdir)
  scp_from(host, from_path, targetdir, opts)
  # scp_from does succeed on a non-existant file, checking if the file/folder actually exists
  if not File.exists?(filename)
    raise IOError, "No such file or directory - #{filename}"
  end
  create_tarball(archive_root, archive_name)
end

#backup_the_file(host, current_dir, new_dir, filename = 'puppet.conf') ⇒ String?

Back up the given file in the current_dir to the new_dir

Parameters:

  • host (Beaker::Host) —

    The target host

  • current_dir (String) —

    The directory containing the file to back up

  • new_dir (String) —

    The directory to copy the file to

  • filename (String) (defaults to: 'puppet.conf') —

    The file to back up. Defaults to ‘puppet.conf’

Returns:

  • (String, nil) —

    The path to the file if the file exists, nil if it doesn’t exist.



451
452
453
454
455
456
457
458
459
460
461
462
463
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 451

def backup_the_file host, current_dir, new_dir, filename = 'puppet.conf'

  old_location = current_dir + '/' + filename
  new_location = new_dir + '/' + filename + '.bak'

  if host.file_exist? old_location
    host.exec( Command.new( "cp #{old_location} #{new_location}" ) )
    return new_location
  else
    logger.warn "Could not backup file '#{old_location}': no such file"
    nil
  end
end

#check_for_package(host, package_name) ⇒ Boolean

Check to see if a package is installed on a remote host

Parameters:

  • host (Host) —

    A host object

  • package_name (String) —

    Name of the package to check for.

Returns:

  • (Boolean) —

    true/false if the package is found



413
414
415
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 413

def check_for_package host, package_name
  host.check_for_package package_name
end

#create_remote_file(hosts, file_path, file_content, opts = {}) ⇒ Result

Note:

This method uses Tempfile in Ruby’s STDLIB as well as #scp_to.

Create a remote file out of a string

Parameters:

  • hosts (Host, #do_scp_to) —

    One or more hosts (or some object that responds like Host#do_scp_from.

  • file_path (String) —

    A remote path to place file_content at.

  • file_content (String) —

    The contents of the file to be placed.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

  • :protocol (String) —

    Name of the underlying transfer method. Valid options are ‘scp’ or ‘rsync’.

Returns:

  • (Result) —

    Returns the result of the underlying SCP operation.



319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 319

def create_remote_file(hosts, file_path, file_content, opts = {})
  Tempfile.open 'beaker' do |tempfile|
    File.open(tempfile.path, 'w') {|file| file.puts file_content }

    opts[:protocol] ||= 'scp'
    case opts[:protocol]
      when 'scp'
        scp_to hosts, tempfile.path, file_path, opts
      when 'rsync'
        rsync_to hosts, tempfile.path, file_path, opts
      else
        logger.debug "Unsupported transfer protocol, returning nil"
        nil
    end
  end
end

#curl_on(host, cmd, opts = {}, &block) ⇒ Object

Run a curl command on the provided host(s)

Parameters:

  • host (Host, Array<Host>, String, Symbol) —

    One or more hosts to act upon, or a role (String or Symbol) that identifies one or more hosts.

  • cmd (String, Command) —

    The curl command to execute on host.

  • block (Proc) —

    Additional actions or assertions.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.



579
580
581
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 579

def curl_on(host, cmd, opts = {}, &block)
  on host, "curl --tlsv1 %s" % cmd, opts, &block
end

#curl_with_retries(desc, host, url, desired_exit_codes, max_retries = 60, retry_interval = 1) ⇒ Object 



583
584
585
586
587
588
589
590
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 583

def curl_with_retries(desc, host, url, desired_exit_codes, max_retries = 60, retry_interval = 1)
  opts = {
    :desired_exit_codes => desired_exit_codes,
    :max_retries => max_retries,
    :retry_interval => retry_interval
  }
  retry_on(host, "curl -m 1 #{url}", opts)
end

#directory_exists_on(host, dir_path) ⇒ Boolean

Check whether a directory exists on the host

Parameters:

  • host (Beaker::Host) —

    The target host

  • dir_path (String) —

    The absolute path of the directory

Returns:

  • (Boolean) —

    Whether the directory exists on the host (using ‘test -d`)



518
519
520
521
522
523
524
525
526
527
528
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 518

def directory_exists_on(host, dir_path)
  if host['platform'] =~ /windows/
    dir_path = "#{dir_path}\\" unless (dir_path[-1].chr == '\\')

    command = Command.new(%{IF exist "#{dir_path}" ( exit 0 ) ELSE ( exit 1 )}, [], { :cmdexe => true })
  else
    command = Command.new(%(test -d "#{dir_path}"), [])
  end

  return on(host, command, { :accept_all_exit_codes => true }).exit_code.zero?
end

#echo_on(hosts, val) ⇒ String, Array<String> The echo'ed value(s) returned by the host(s)

‘echo’ the provided value on the given host(s)

Parameters:

  • hosts (Host, Array<Host>, String, Symbol) —

    One or more hosts to act upon, or a role (String or Symbol) that identifies one or more hosts.

  • val (String) —

    The string to ‘echo’ on the host(s)

Returns:

  • (String, Array<String> The echo'ed value(s) returned by the host(s)) —

    String, Array<String> The echo’ed value(s) returned by the host(s)



682
683
684
685
686
687
688
689
690
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 682

def echo_on hosts, val
  block_on hosts do |host|
    if host.is_powershell?
      host.exec(Command.new("echo #{val}")).stdout.chomp
    else
      host.exec(Command.new("echo \"#{val}\"")).stdout.chomp
    end
  end
end

#execute_powershell_script_on(hosts, powershell_script, opts = {}) ⇒ Result

Note:

This method uses Tempfile in Ruby’s STDLIB as well as #create_remote_file.

Execute a powershell script from file, remote file created from provided string

Parameters:

  • hosts (Host) —

    One or more hosts (or some object that responds like Host#do_scp_from.

  • powershell_script (String) —

    A string describing a set of powershell actions

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    Returns the result of the powershell command execution



347
348
349
350
351
352
353
354
355
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 347

def execute_powershell_script_on(hosts, powershell_script, opts = {})
  block_on hosts, opts do |host|
    script_path = "beaker_powershell_script_#{Time.now.to_i}.ps1"
    create_remote_file(host, script_path, powershell_script, opts)
    native_path = script_path.gsub(/\//, "\\")
    on host, powershell("", {"File" => native_path }), opts
  end

end

#exit_code ⇒ Object

Deprecated.

An proxy for the last Result#exit_code returned by a method that makes remote calls. Use the Result object returned by the method directly instead. For Usage see Result.



153
154
155
156
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 153

def exit_code
  return nil if @result.nil?
  @result.exit_code
end

#file_contents_on(host, file_path) ⇒ String

Get the contents of a file on the host

Parameters:

  • host (Beaker::Host) —

    The target host

  • file_path (String) —

    The absoltue path to the file

Returns:

  • (String) —

    The contents of the file



549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 549

def file_contents_on(host, file_path)
  file_contents = nil

  split_path = win_ads_path(file_path)
  if file_exists_on(host, split_path[:path])
    if host['platform'] =~ /windows/
      file_path.gsub!('/', '\\')

      command = %{Get-Content -Raw -Path #{file_path}}
      command += %{ -Stream #{split_path[:ads]}} if split_path[:ads]

      file_contents = on(host, powershell(command))&.stdout&.strip
    else
      file_contents = on(host, %(cat "#{file_path}"))&.stdout&.strip
    end
  else
    logger.warn("File '#{file_path}' on '#{host} does not exist")
  end

  return file_contents
end

#file_exists_on(host, file_path) ⇒ Boolean

Check whether a file exists on the host

Parameters:

  • host (Beaker::Host) —

    The target host

  • file_path (String) —

    The absolute path of the file

Returns:

  • (Boolean) —

    Whether the file exists on the host (using ‘test -f`)



493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 493

def file_exists_on(host, file_path)
  if host['platform'] =~ /windows/
    command = %(Test-Path #{file_path})

    if file_path.include?(':')
      split_path = win_ads_path(file_path)

      command = %(Test-Path #{split_path[:path]})
      command += %( -AND Get-Item -path #{split_path[:path]} -stream #{split_path[:ads]}) if split_path[:ads]
    end

    command = powershell(command)
  else
    command = %(test -f "#{file_path}")
  end

  return on(host, command, { :accept_all_exit_codes => true}).exit_code.zero?
end

#install_package(host, package_name, package_version = nil) ⇒ Result

Install a package on a host

Parameters:

  • host (Host) —

    A host object

  • package_name (String) —

    Name of the package to install

Returns:

  • (Result) —

    An object representing the outcome of *install command*.



393
394
395
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 393

def install_package host, package_name, package_version = nil
  host.install_package package_name, '', package_version
end

#link_exists_on(host, link_path) ⇒ Boolean

Check whether a symlink exists on the host

Parameters:

  • host (Beaker::Host) —

    The target host

  • link_path (String) —

    The absolute path of the symlink

Returns:

  • (Boolean) —

    Whether the symlink exists on the host (using ‘test -L`)



536
537
538
539
540
541
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 536

def link_exists_on(host, link_path)
  # Links are weird on windows, fall back to seeing if the file exists
  return file_exists_on(host, link_path) if host['platform'] =~ /windows/

  return on(host, Command.new(%(test -L "#{link_path}"), accept_all_exit_codes: true)).exit_code.zero?
end

#on(host, command, opts = {}, &block) ⇒ Result

The primary method for executing commands on some set of hosts.

Examples:

Most basic usage

on hosts, 'ls /tmp'

Allowing additional exit codes to pass

on agents, 'puppet agent -t', :acceptable_exit_codes => [0,2]

Using the returned result for any kind of checking

if on(host, 'ls -la ~').stdout =~ /\.bin/
  ...do some action...
end

Using TestCase helpers from within a test.

agents.each do |agent|
  on agent, 'cat /etc/puppet/puppet.conf' do
    assert_match stdout, /server = #{master}/, 'WTF Mate'
  end
end

Using a role (defined in a String) to identify the host

on "master", "echo hello"

Using a role (defined in a Symbol) to identify the host

on :dashboard, "echo hello"

Parameters:

  • host (Host, Array<Host>, String, Symbol) —

    One or more hosts to act upon, or a role (String or Symbol) that identifies one or more hosts.

  • command (String, Command) —

    The command to execute on host.

  • block (Proc) —

    Additional actions or assertions.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    An object representing the outcome of command.

Raises:

  • (FailTest) —

    Raises an exception if command obviously fails.



62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 62

def on(host, command, opts = {}, &block)
  block_on host, opts do | host |
    if command.is_a? String
      cmd_opts = {}
      #add any additional environment variables to the command
      if opts[:environment]
        cmd_opts['ENV'] = opts[:environment]
      end
      command_object = Command.new(command.to_s, [], cmd_opts)
    elsif command.is_a? Command
      if opts[:environment]
        command_object = command.clone
        command_object.environment = opts[:environment]
      else
        command_object = command
      end
    else
      msg = "DSL method `on` can only be called with a String or Beaker::Command"
      msg << " object as the command parameter, not #{command.class}."
      raise ArgumentError, msg
    end
    @result = host.exec(command_object, opts)

    # Also, let additional checking be performed by the caller.
    if block_given?
      case block.arity
        #block with arity of 0, just hand back yourself
        when 0
          yield self
        #block with arity of 1 or greater, hand back the result object
        else
          yield @result
      end
    end
    @result
  end
end

#retry_on(host, command, opts = {}, &block) ⇒ Result

This command will execute repeatedly until success or it runs out with an error

Parameters:

  • host (Host, Array<Host>, String, Symbol) —

    One or more hosts to act upon, or a role (String or Symbol) that identifies one or more hosts.

  • command (String, Command) —

    The command to execute on host.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

  • block (Proc) —

    Additional actions or assertions.

Options Hash (opts):

  • :desired_exit_codes (Array<Fixnum>, Fixnum) — default: 0 —

    An array or integer exit code(s) that should be considered acceptable. An error will be thrown if the exit code never matches one of the values in this list.

  • :max_retries (Fixnum) — default: 60 —

    number of times the command will be tried before failing

  • :retry_interval (Float) — default: 1 —

    number of seconds that we’ll wait between tries

  • :verbose (Boolean) — default: false

Returns:

  • (Result) —

    Result object of the last command execution



611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 611

def retry_on(host, command, opts = {}, &block)
  option_exit_codes     = opts[:desired_exit_codes]
  option_max_retries    = opts[:max_retries].to_i
  option_retry_interval = opts[:retry_interval].to_f
  desired_exit_codes    = option_exit_codes ? [option_exit_codes].flatten : [0]
  desired_exit_codes    = [0] if desired_exit_codes.empty?
  max_retries           = option_max_retries == 0 ? 60 : option_max_retries  # nil & "" both return 0
  retry_interval        = option_retry_interval == 0 ? 1 : option_retry_interval
  verbose               = true.to_s == opts[:verbose]

  log_prefix = host.log_prefix
  logger.debug "\n#{log_prefix} #{Time.new.strftime('%H:%M:%S')}$ #{command}"
  logger.debug "  Trying command #{max_retries} times."
  logger.debug ".", add_newline=false

  result = on host, command, {:accept_all_exit_codes => true, :silent => !verbose}, &block
  num_retries = 0
  until desired_exit_codes.include?(result.exit_code)
    sleep retry_interval
    result = on host, command, {:accept_all_exit_codes => true, :silent => !verbose}, &block
    num_retries += 1
    logger.debug ".", add_newline=false
    if (num_retries > max_retries)
      logger.debug "  Command \`#{command}\` failed."
      fail("Command \`#{command}\` failed.")
    end
  end
  logger.debug "\n#{log_prefix} #{Time.new.strftime('%H:%M:%S')}$ #{command} ostensibly successful."
  result
end

#rsync_to(host, from_path, to_path, opts = {}) ⇒ Result

Note:

rsync is required on the local host.

Move a local file or directory to a remote host using rsync

Parameters:

  • host (Host, #do_scp_to) —

    A host object that responds like Host.

  • from_path (String) —

    A local path to a file or directory.

  • to_path (String) —

    A remote path to copy from_path to.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    Returns the result of the rsync operation



216
217
218
219
220
221
222
223
224
225
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 216

def rsync_to host, from_path, to_path, opts = {}
  block_on host do | host |
    if host['platform'] =~ /windows/ && to_path.match('`cygpath')
      result = host.echo "#{to_path}"
      to_path = result.raw_output.chomp
    end
    @result = host.do_rsync_to(from_path, to_path, opts)
    @result
  end
end

#run_script(script, opts = {}, &block) ⇒ Object

Move a local script to default host and execute it

See Also:



383
384
385
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 383

def run_script(script, opts = {}, &block)
  run_script_on(default, script, opts, &block)
end

#run_script_on(host, script, opts = {}, &block) ⇒ Result

Note:

this relies on #on and #scp_to

Move a local script to a remote host and execute it

Parameters:

  • host (Host, #do_scp_to) —

    One or more hosts (or some object that responds like Host#do_scp_from.

  • script (String) —

    A local path to find an executable script at.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

  • block (Proc) —

    Additional tests to run after script has executed

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    Returns the result of the underlying SCP operation.



368
369
370
371
372
373
374
375
376
377
378
379
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 368

def run_script_on(host, script, opts = {}, &block)
  # this is unsafe as it uses the File::SEPARATOR will be set to that
  # of the coordinator node.  This works for us because we use cygwin
  # which will properly convert the paths.  Otherwise this would not
  # work for running tests on a windows machine when the coordinator
  # that the harness is running on is *nix. We should use
  # {Beaker::Host#temp_path} instead. TODO
  remote_path = File.join("", "tmp", File.basename(script))

  scp_to host, script, remote_path
  on host, remote_path, opts, &block
end

#scp_from(host, from_path, to_path, opts = {}) ⇒ Result

Note:

If using Host for the hosts scp is not required on the system as it uses Ruby’s net/scp library. The net-scp gem however is required (and specified in the gemspec).

Move a file from a remote to a local path

Parameters:

  • host (Host, #do_scp_from) —

    One or more hosts (or some object that responds like Host#do_scp_from.

  • from_path (String) —

    A remote path to a file.

  • to_path (String) —

    A local path to copy from_path to.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    Returns the result of the SCP operation



171
172
173
174
175
176
177
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 171

def scp_from host, from_path, to_path, opts = {}
  block_on host do | host |
    @result = host.do_scp_from(from_path, to_path, opts)
    @result.log logger
    @result
  end
end

#scp_to(host, from_path, to_path, opts = {}) ⇒ Result

Note:

If using Host for the hosts scp is not required on the system as it uses Ruby’s net/scp library. The net-scp gem however is required (and specified in the gemspec. When using SCP with Windows it will now auto expand path when using ‘cygpath instead of failing or requiring full path

Move a local file to a remote host using scp

Parameters:

  • host (Host, #do_scp_to) —

    One or more hosts (or some object that responds like Host#do_scp_to.

  • from_path (String) —

    A local path to a file.

  • to_path (String) —

    A remote path to copy from_path to.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    Returns the result of the SCP operation



194
195
196
197
198
199
200
201
202
203
204
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 194

def scp_to host, from_path, to_path, opts = {}
  block_on host do | host |
    if host['platform'] =~ /windows/ && to_path.match('`cygpath')
      result = on host, "echo #{to_path}"
      to_path = result.raw_output.chomp
    end
    @result = host.do_scp_to(from_path, to_path, opts)
    @result.log logger
    @result
  end
end

#shell(command, opts = {}, &block) ⇒ Result

The method for executing commands on the default host

Examples:

Most basic usage

shell 'ls /tmp'

Allowing additional exit codes to pass

shell 'puppet agent -t', :acceptable_exit_codes => [0,2]

Using the returned result for any kind of checking

if shell('ls -la ~').stdout =~ /\.bin/
  ...do some action...
end

Using TestCase helpers from within a test.

shell('cat /etc/puppet/puppet.conf') do |result|
  assert_match result.stdout, /server = #{master}/, 'WTF Mate'
end

Parameters:

  • command (String, Command) —

    The command to execute on host.

  • block (Proc) —

    Additional actions or assertions.

  • opts (Hash{Symbol=>String}) (defaults to: {}) —

    Options to alter execution.

Options Hash (opts):

  • :silent (Boolean) — default: false —

    Do not produce log output

  • :acceptable_exit_codes (Array<Fixnum>) — default: [0] —

    An array (or range) of integer exit codes that should be considered acceptable. An error will be thrown if the exit code does not match one of the values in this list.

  • :accept_all_exit_codes (Boolean) — default: false —

    Consider all exit codes as passing.

  • :dry_run (Boolean) — default: false —

    Do not actually execute any commands on the SUT

  • :stdin (String) — default: nil —

    Input to be provided during command execution on the SUT.

  • :pty (Boolean) — default: false —

    Execute this command in a pseudoterminal.

  • :expect_connection_failure (Boolean) — default: false —

    Expect this command to result in a connection failure, reconnect and continue execution.

  • :environment (Hash{String=>String}) — default: {} —

    These will be treated as extra environment variables that should be set before running the command.

  • :run_in_parallel (Boolean) —

    Whether to run on each host in parallel.

Returns:

  • (Result) —

    An object representing the outcome of command.

Raises:

  • (FailTest) —

    Raises an exception if command obviously fails.



124
125
126
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 124

def shell(command, opts = {}, &block)
  on(default, command, opts, &block)
end

#stderr ⇒ Object

Deprecated.

An proxy for the last Result#stderr returned by a method that makes remote calls. Use the Result object returned by the method directly instead. For Usage see Result.



143
144
145
146
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 143

def stderr
  return nil if @result.nil?
  @result.stderr
end

#stdout ⇒ Object

Deprecated.

An proxy for the last Result#stdout returned by a method that makes remote calls. Use the Result object returned by the method directly instead. For Usage see Result.



133
134
135
136
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 133

def stdout
  return nil if @result.nil?
  @result.stdout
end

#uninstall_package(host, package_name) ⇒ Result

Uninstall a package on a host

Parameters:

  • host (Host) —

    A host object

  • package_name (String) —

    Name of the package to uninstall

Returns:

  • (Result) —

    An object representing the outcome of *uninstall command*.



403
404
405
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 403

def uninstall_package host, package_name
  host.uninstall_package package_name
end

#upgrade_package(host, package_name) ⇒ Result

Upgrade a package on a host. The package must already be installed

Parameters:

  • host (Host) —

    A host object

  • package_name (String) —

    Name of the package to install

Returns:

  • (Result) —

    An object representing the outcome of *upgrade command*.



423
424
425
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 423

def upgrade_package host, package_name
  host.upgrade_package package_name
end

#win_ads_path(file_path) ⇒ Hash

Return a Hash with the Alternate Data Stream (ADS) name as well as the base file path

Parameters:

  • file_path (String) —

    The full path with the ADS attached

Returns:

  • (Hash) —

    The base file path and ADS name



471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
 
# File 'lib/beaker/dsl/helpers/host_helpers.rb', line 471

def win_ads_path(file_path)
  ads_path = {
    path: file_path,
    ads: nil
  }

  split_path = file_path.split(':')

  if split_path.size > 2
    ads_path[:ads] = split_path.pop
    ads_path[:path] = split_path.join(':')
  end

  return ads_path
end