module Aruba::Api
Constants
- DEFAULT_IO_WAIT_SECONDS
- DEFAULT_TIMEOUT_SECONDS
Public Instance Methods
# File lib/aruba/api.rb, line 91 def _create_file(file_name, file_content, check_presence) in_current_dir do file_name = File.expand_path(file_name) raise "expected #{file_name} to be present" if check_presence && !File.file?(file_name) _mkdir(File.dirname(file_name)) File.open(file_name, 'w') { |f| f << file_content } end end
# File lib/aruba/api.rb, line 163 def _create_fixed_size_file(file_name, file_size, check_presence) in_current_dir do file_name = File.expand_path(file_name) raise "expected #{file_name} to be present" if check_presence && !File.file?(file_name) _mkdir(File.dirname(file_name)) File.open(file_name, "wb"){ |f| f.seek(file_size - 1); f.write("\0") } end end
# File lib/aruba/api.rb, line 628 def _ensure_newline(str) str.chomp << "\n" end
# File lib/aruba/api.rb, line 369 def _mkdir(dir_name) dir_name = File.expand_path(dir_name) FileUtils.mkdir_p(dir_name) unless File.directory?(dir_name) end
# File lib/aruba/api.rb, line 624 def _read_interactive @interactive.read_stdout end
# File lib/aruba/api.rb, line 375 def _rm_rf(dir_name) dir_name = File.expand_path(dir_name) FileUtils.rm_rf(dir_name) end
# File lib/aruba/api.rb, line 619 def _write_interactive(input) @interactive.stdin.write(input) @interactive.stdin.flush end
# File lib/aruba/api.rb, line 435 def all_output all_stdout << all_stderr end
# File lib/aruba/api.rb, line 430 def all_stderr stop_processes! only_processes.inject("") { |out, ps| out << ps.stderr } end
# File lib/aruba/api.rb, line 425 def all_stdout stop_processes! only_processes.inject("") { |out, ps| out << ps.stdout } end
# File lib/aruba/api.rb, line 632 def announce_or_puts(msg) if(@puts) Kernel.puts(msg) else puts(msg) end end
# File lib/aruba/api.rb, line 509 def append_output_to(message) "#{message} Output:\n\n#{all_output}\n" end
Append data to file
@param [String] file_name
The name of the file to be used
@param [String] file_content
The content which should be appended to file
# File lib/aruba/api.rb, line 192 def append_to_file(file_name, file_content) in_current_dir do file_name = File.expand_path(file_name) _mkdir(File.dirname(file_name)) File.open(file_name, 'a') { |f| f << file_content } end end
# File lib/aruba/api.rb, line 439 def assert_exact_output(expected, actual) actual.force_encoding(expected.encoding) if RUBY_VERSION >= "1.9" expect(unescape(actual)).to eq unescape(expected) end
# File lib/aruba/api.rb, line 499 def assert_exit_status(status) expect(last_exit_status).to eq(status), append_output_to("Exit status was #{last_exit_status} but expected it to be #{status}.") end
TODO: Remove this. Call more methods elsewhere instead. Reveals more intent.
# File lib/aruba/api.rb, line 486 def assert_exit_status_and_output(expect_to_pass, expected_output, expect_exact_output) assert_success(expect_to_pass) if expect_exact_output assert_exact_output(expected_output, all_output) else assert_partial_output(expected_output, all_output) end end
# File lib/aruba/api.rb, line 480 def assert_exit_status_and_partial_output(expect_to_pass, expected) assert_success(expect_to_pass) assert_partial_output(expected, all_output) end
# File lib/aruba/api.rb, line 476 def assert_failing_with(expected) assert_exit_status_and_partial_output(false, expected) end
# File lib/aruba/api.rb, line 449 def assert_matching_output(expected, actual) actual.force_encoding(expected.encoding) if RUBY_VERSION >= "1.9" expect(unescape(actual)).to match /#{unescape(expected)}/m end
# File lib/aruba/api.rb, line 459 def assert_no_partial_output(unexpected, actual) actual.force_encoding(unexpected.encoding) if RUBY_VERSION >= "1.9" if Regexp === unexpected expect(unescape(actual)).not_to match unexpected else expect(unescape(actual)).not_to include(unexpected) end end
# File lib/aruba/api.rb, line 504 def assert_not_exit_status(status) expect(last_exit_status).not_to eq(status), append_output_to("Exit status was #{last_exit_status} which was not expected.") end
# File lib/aruba/api.rb, line 454 def assert_not_matching_output(expected, actual) actual.force_encoding(expected.encoding) if RUBY_VERSION >= "1.9" expect(unescape(actual)).not_to match /#{unescape(expected)}/m end
# File lib/aruba/api.rb, line 444 def assert_partial_output(expected, actual) actual.force_encoding(expected.encoding) if RUBY_VERSION >= "1.9" expect(unescape(actual)).to include(unescape(expected)) end
# File lib/aruba/api.rb, line 468 def assert_partial_output_interactive(expected) unescape(_read_interactive).include?(unescape(expected)) ? true : false end
# File lib/aruba/api.rb, line 472 def assert_passing_with(expected) assert_exit_status_and_partial_output(true, expected) end
# File lib/aruba/api.rb, line 495 def assert_success(success) success ? assert_exit_status(0) : assert_not_exit_status(0) end
Switch to directory
@param [String] dir
The directory
# File lib/aruba/api.rb, line 40 def cd(dir) dirs << dir raise "#{current_dir} is not a directory." unless File.directory?(current_dir) end
Check presence of a directory
@param [Array] paths
The paths to be checked
@param [true, false] expect_presence
Should the directory be there or should the directory not be there
# File lib/aruba/api.rb, line 350 def check_directory_presence(paths, expect_presence) prep_for_fs_check do paths.each do |path| path = File.expand_path(path) if expect_presence expect(File).to be_directory(path) else expect(File).not_to be_directory(path) end end end end
Check if the exact content can be found in file
@param [String] file
The file to be checked
@param [String] exact_content
The content of the file
# File lib/aruba/api.rb, line 339 def check_exact_file_content(file, exact_content) prep_for_fs_check { expect(IO.read(file)).to eq exact_content } end
Check the content of file
It supports partial content as well. And it is up to you to decided if the content must be there or not.
@param [String] file
The file to be checked
@param [String] partial_content
The content which must/must not be in the file
@param [true, false] expect_match
Must the content be in the file or not
# File lib/aruba/api.rb, line 318 def check_file_content(file, partial_content, expect_match) regexp = regexp(partial_content) prep_for_fs_check do file = File.expand_path(file) content = IO.read(file) if expect_match expect(content).to match regexp else expect(content).not_to match regexp end end end
Check if paths are present
@param [#each] paths
The paths which should be checked
@param [true,false] expect_presence
Should the given paths be present (true) or absent (false)
# File lib/aruba/api.rb, line 230 def check_file_presence(paths, expect_presence) prep_for_fs_check do paths.each do |path| if path.kind_of? Regexp if expect_presence expect(Dir.glob('**/*')).to include_regexp(path) else expect(Dir.glob('**/*')).not_to include_regexp(path) end else path = File.expand_path(path) if expect_presence expect(File).to be_file(path) else expect(File).not_to be_file(path) end end end end end
Check the file size of paths
@params [Hash] paths_and_sizes
A hash containing the path (key) and the expected size (value)
@example
paths_and_sizes = { 'file' => 10 } check_file_size(paths_and_sizes)
# File lib/aruba/api.rb, line 279 def check_file_size(paths_and_sizes) prep_for_fs_check do paths_and_sizes.each do |path, size| path = File.expand_path(path) expect(File.size(path)).to eq size end end end
Check file system permissions of file
@param [Octal] mode
Expected file system permissions, e.g. 0755
@param [String] file_name
Expected file system permissions, e.g. 0755
@param [Boolean] expect_permissions
Are the permissions expected to be mode or are they expected not to be mode?
# File lib/aruba/api.rb, line 136 def check_filesystem_permissions(mode, file_name, expect_permissions) in_current_dir do file_name = File.expand_path(file_name) raise "expected #{file_name} to be present" unless FileTest.exists?(file_name) if mode.kind_of? Integer expected_mode = mode.to_s(8) else expected_mode = mode.gsub(/^0*/, '') end file_mode = sprintf( "%o", File::Stat.new(file_name).mode )[-4,4].gsub(/^0*/, '') if expect_permissions expect(file_mode).to eq expected_mode else expect(file_mode).not_to eq expected_mode end end end
# File lib/aruba/api.rb, line 122 def chmod(*args, &block) warn('The use of "chmod" is deprecated. Use "filesystem_permissions" instead') filesystem_permissions(*args, &block) end
Clean the current directory
# File lib/aruba/api.rb, line 23 def clean_current_dir _rm_rf(current_dir) _mkdir(current_dir) end
Close stdin
# File lib/aruba/api.rb, line 610 def close_input @interactive.stdin.close end
Create a directory in current directory
@param [String] directory_name
The name of the directory which should be created
# File lib/aruba/api.rb, line 205 def create_dir(directory_name) in_current_dir do _mkdir(directory_name) end end
Get access to current dir
@return
Current directory
# File lib/aruba/api.rb, line 32 def current_dir File.join(*dirs) end
# File lib/aruba/api.rb, line 648 def current_ruby File.join(RbConfig::CONFIG['bindir'], RbConfig::CONFIG['ruby_install_name']) end
# File lib/aruba/api.rb, line 640 def detect_ruby(cmd) if cmd =~ /^ruby\s/ cmd.gsub(/^ruby\s/, "#{current_ruby} ") else cmd end end
The path to the directory which should contain all your test data You might want to overwrite this method to place your data else where.
@return [Array]
The directory path: Each subdirectory is a member of an array
# File lib/aruba/api.rb, line 50 def dirs @dirs ||= ['tmp', 'aruba'] end
# File lib/aruba/api.rb, line 614 def eot warn(%Q{\e[35m The \"#eot\"-method is deprecated. It will be deleted with the next major version. Please use \"#close_input\"-method instead.\e[0m}) close_input end
# File lib/aruba/api.rb, line 574 def exit_timeout @aruba_timeout_seconds || DEFAULT_TIMEOUT_SECONDS end
Change file system permissions of file
@param [Octal] mode
File system mode, eg. 0755
@param [String] file_name
Name of file to be modified. This file needs to be present to succeed
# File lib/aruba/api.rb, line 109 def filesystem_permissions(mode, file_name) in_current_dir do file_name = File.expand_path(file_name) raise "expected #{file_name} to be present" unless FileTest.exists?(file_name) if mode.kind_of? String FileUtils.chmod(mode.to_i(8),file_name) else FileUtils.chmod(mode, file_name) end end end
# File lib/aruba/api.rb, line 534 def get_process(wanted) matching_processes = processes.reverse.find{ |name, _| name == wanted } raise ArgumentError.new("No process named '#{wanted}' has been started") unless matching_processes matching_processes.last end
Execute block in current directory
@yield
The block which should be run in current directory
# File lib/aruba/api.rb, line 17 def in_current_dir(&block) _mkdir(current_dir) Dir.chdir(current_dir, &block) end
# File lib/aruba/api.rb, line 580 def io_wait @aruba_io_wait_seconds || DEFAULT_IO_WAIT_SECONDS end
# File lib/aruba/api.rb, line 157 def mod?(*args, &block) warn('The use of "mod?" is deprecated. Use "check_filesystem_permissions" instead') check_filesystem_permissions(*args, &block) end
# File lib/aruba/api.rb, line 540 def only_processes processes.collect{ |_, process| process } end
# File lib/aruba/api.rb, line 700 def original_env @original_env ||= {} end
Fetch output (stdout, stderr) from command
@param [String] cmd
The comand
# File lib/aruba/api.rb, line 402 def output_from(cmd) cmd = detect_ruby(cmd) get_process(cmd).output end
Create a file with given content
The method does check if file already exists and fails if the file is missing. If the file name is a path the method will create all neccessary directories.
# File lib/aruba/api.rb, line 87 def overwrite_file(file_name, file_content) _create_file(file_name, file_content, true) end
Pipe data in file
@param [String] file_name
The file which should be used to pipe in data
# File lib/aruba/api.rb, line 256 def pipe_in_file(file_name) in_current_dir do file_name = File.expand_path(file_name) File.open(file_name, 'r').each_line do |line| _write_interactive(line) end end end
# File lib/aruba/api.rb, line 364 def prep_for_fs_check(&block) stop_processes! in_current_dir{ block.call } end
# File lib/aruba/api.rb, line 513 def processes @processes ||= [] end
# File lib/aruba/api.rb, line 394 def regexp(string_or_regexp) Regexp === string_or_regexp ? string_or_regexp : Regexp.compile(Regexp.escape(string_or_regexp)) end
# File lib/aruba/api.rb, line 530 def register_process(name, process) processes << [name, process] end
Remove directory
@param [String] directory_name
The name of the directory which should be removed
# File lib/aruba/api.rb, line 215 def remove_dir(directory_name) in_current_dir do directory_name = File.expand_path(directory_name) FileUtils.rmdir(directory_name) end end
Remove file
@param [String] file_name
The file which should be deleted in current directory
# File lib/aruba/api.rb, line 177 def remove_file(file_name) in_current_dir do file_name = File.expand_path(file_name) FileUtils.rm(file_name) end end
Restore original process environment
# File lib/aruba/api.rb, line 690 def restore_env original_env.each do |key, value| if value ENV[key] = value else ENV.delete key end end end
Run given command and stop it if timeout is reached
@param [String] cmd
The command which should be executed
@param [Integer] timeout
If the timeout is reached the command will be killed
# File lib/aruba/api.rb, line 551 def run(cmd, timeout = nil) timeout ||= exit_timeout @commands ||= [] @commands << cmd cmd = detect_ruby(cmd) in_current_dir do Aruba.config.hooks.execute(:before_cmd, self, cmd) announcer.dir(Dir.pwd) announcer.cmd(cmd) process = Aruba.process.new(cmd, timeout, io_wait) register_process(cmd, process) process.run! block_given? ? yield(process) : process end end
Run a command interactively
@param [String] cmd
The command to by run
# File lib/aruba/api.rb, line 596 def run_interactive(cmd) @interactive = run(cmd) end
# File lib/aruba/api.rb, line 584 def run_simple(cmd, fail_on_error=true, timeout = nil) run(cmd, timeout) do |process| stop_process(process) end @timed_out = last_exit_status.nil? assert_exit_status(0) if fail_on_error end
Set environment variable
@param [String] key
The name of the environment variable as string, e.g. 'HOME'
@param [String] value
The value of the environment variable. Needs to be a string.
# File lib/aruba/api.rb, line 683 def set_env(key, value) announcer.env(key, value) original_env[key] = ENV.delete(key) unless original_env.key? key ENV[key] = value end
Fetch stderr from command
@param [String] cmd
The comand
# File lib/aruba/api.rb, line 420 def stderr_from(cmd) cmd = detect_ruby(cmd) get_process(cmd).stderr end
Fetch stdout from command
@param [String] cmd
The comand
# File lib/aruba/api.rb, line 411 def stdout_from(cmd) cmd = detect_ruby(cmd) get_process(cmd).stdout end
# File lib/aruba/api.rb, line 517 def stop_processes! processes.each do |_, process| stop_process(process) end end
# File lib/aruba/api.rb, line 523 def terminate_processes! processes.each do |_, process| terminate_process(process) stop_process(process) end end
Provide data to command via stdin
@param [String] input
The input for the command
# File lib/aruba/api.rb, line 604 def type(input) return close_input if "" == input _write_interactive(_ensure_newline(input)) end
Unescape string
@param [String] string
The string which should be unescaped, e.g. the output of a command
@return
The string stripped from escape sequences
# File lib/aruba/api.rb, line 388 def unescape(string) string = string.gsub('\n', "\n").gsub('\"', '"').gsub('\e', "\e") string = string.gsub(/\e\[\d+(?>(;\d+)*)m/, '') unless @aruba_keep_ansi string end
# File lib/aruba/api.rb, line 670 def unset_bundler_env_vars %w[RUBYOPT BUNDLE_PATH BUNDLE_BIN_PATH BUNDLE_GEMFILE].each do |key| set_env(key, nil) end end
# File lib/aruba/api.rb, line 652 def use_clean_gemset(gemset) run_simple(%Q{rvm gemset create "#{gemset}"}, true) if all_stdout =~ /'#{gemset}' gemset created \((.*)\)\./ gem_home = $1 set_env('GEM_HOME', gem_home) set_env('GEM_PATH', gem_home) set_env('BUNDLE_PATH', gem_home) paths = (ENV['PATH'] || "").split(File::PATH_SEPARATOR) paths.unshift(File.join(gem_home, 'bin')) set_env('PATH', paths.uniq.join(File::PATH_SEPARATOR)) run_simple("gem install bundler", true) else raise "I didn't understand rvm's output: #{all_stdout}" end end
# File lib/aruba/api.rb, line 704 def with_env(env = {}, &block) env.each do |k,v| set_env k, v block.call restore_env end end
Read content of file and yield the content to block
@param [String) file
The name of file which should be read from
@yield
Pass the content of the given file to this block
# File lib/aruba/api.rb, line 296 def with_file_content(file, &block) prep_for_fs_check do file = File.expand_path(file) content = IO.read(file) yield(content) end end
Create a file with given content
The method does not check if file already exists. If the file name is a path the method will create all neccessary directories.
@param [String] file_name
The name of the file
@param [String] file_content
The content which should be written to the file
# File lib/aruba/api.rb, line 64 def write_file(file_name, file_content) _create_file(file_name, file_content, false) end
Create a file with the given size
The method does not check if file already exists. If the file name is a path the method will create all neccessary directories.
@param [String] file_name
The name of the file
@param [Integer] file_size
The size of the file
# File lib/aruba/api.rb, line 78 def write_fixed_size_file(file_name, file_size) _create_fixed_size_file(file_name, file_size, false) end
Private Instance Methods
# File lib/aruba/api.rb, line 729 def announcer Announcer.new(self, :stdout => @announce_stdout, :stderr => @announce_stderr, :dir => @announce_dir, :cmd => @announce_cmd, :env => @announce_env) end
TODO: move some more methods under here!
# File lib/aruba/api.rb, line 715 def last_exit_status return @last_exit_status if @last_exit_status stop_processes! @last_exit_status end
# File lib/aruba/api.rb, line 721 def stop_process(process) @last_exit_status = process.stop(announcer) end
# File lib/aruba/api.rb, line 725 def terminate_process(process) process.terminate end