1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 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 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124
|
require 'helper'
module SSHKit
class TestConfiguration < UnitTest
def setup
super
SSHKit.config.command_map.clear
SSHKit.config.output = SSHKit::Formatter::Pretty.new($stdout)
end
def test_deprecation_output
output = ''
SSHKit.config.deprecation_output = output
SSHKit.config.deprecation_logger.log('Test')
assert_equal "[Deprecated] Test\n", output.lines.first
end
def test_default_deprecation_output
SSHKit.config.deprecation_logger.log('Test')
end
def test_nil_deprecation_output
SSHKit.config.deprecation_output = nil
SSHKit.config.deprecation_logger.log('Test')
end
def test_output
assert SSHKit.config.output.is_a? SSHKit::Formatter::Pretty
assert SSHKit.config.output = $stderr
end
def test_umask
assert SSHKit.config.umask.nil?
assert SSHKit.config.umask = '007'
assert_equal '007', SSHKit.config.umask
end
def test_output_verbosity
assert_equal Logger::INFO, SSHKit.config.output_verbosity
assert SSHKit.config.output_verbosity = :debug
assert_equal Logger::DEBUG, SSHKit.config.output_verbosity
assert SSHKit.config.output_verbosity = Logger::INFO
assert_equal Logger::INFO, SSHKit.config.output_verbosity
assert SSHKit.config.output_verbosity = 0
assert_equal Logger::DEBUG, SSHKit.config.output_verbosity
end
def test_default_env
assert SSHKit.config.default_env
end
def test_default_runner
assert_equal :parallel, SSHKit.config.default_runner
SSHKit.config.default_runner = :sequence
assert_equal :sequence, SSHKit.config.default_runner
end
def test_default_runner_config
config_hash = { wait: 5 }
config_hash_with_runner = { in: :groups, limit: 5 }
default_hash = { in: SSHKit.config.default_runner }
assert_equal default_hash, SSHKit.config.default_runner_config
SSHKit.config.default_runner_config = config_hash
assert_equal default_hash.merge(config_hash), SSHKit.config.default_runner_config
SSHKit.config.default_runner_config = config_hash_with_runner
assert_equal config_hash_with_runner, SSHKit.config.default_runner_config
assert_equal config_hash_with_runner[:in], SSHKit.config.default_runner
end
def test_backend
assert_equal SSHKit::Backend::Netssh, SSHKit.config.backend
assert SSHKit.config.backend = SSHKit::Backend::Printer
assert_equal SSHKit::Backend::Printer, SSHKit.config.backend
end
def test_command_map
assert_equal SSHKit.config.command_map.is_a?(SSHKit::CommandMap), true
cm = Hash.new { |h,k| h[k] = "/opt/sites/example/current/bin #{k}"}
assert SSHKit.config.command_map = cm
assert_equal SSHKit.config.command_map.is_a?(SSHKit::CommandMap), true
assert_equal "/opt/sites/example/current/bin ruby", SSHKit.config.command_map[:ruby]
end
def test_setting_formatter_types
{
dot: SSHKit::Formatter::Dot,
blackhole: SSHKit::Formatter::BlackHole,
simpletext: SSHKit::Formatter::SimpleText,
}.each do |format, expected_class|
SSHKit.config.format = format
assert SSHKit.config.output.is_a? expected_class
end
end
def test_prohibits_unknown_formatter_type_with_exception
assert_raises(NameError) do
SSHKit.config.format = :doesnotexist
end
end
def test_options_can_be_provided_to_formatter
SSHKit.config.use_format(TestFormatter, :color => false)
formatter = SSHKit.config.output
assert_instance_of(TestFormatter, formatter)
assert_equal($stdout, formatter.output)
assert_equal({ :color => false }, formatter.options)
end
class TestFormatter
attr_accessor :output, :options
def initialize(output, options={})
@output = output
@options = options
end
end
end
end
|