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
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
require 'helper'
require 'securerandom'
require 'benchmark'
module SSHKit
module Backend
class TestNetssh < FunctionalTest
def setup
super
@output = String.new
SSHKit.config.output_verbosity = :debug
SSHKit.config.output = SSHKit::Formatter::SimpleText.new(@output)
end
def a_host
VagrantWrapper.hosts['one']
end
def test_simple_netssh
Netssh.new(a_host) do
execute 'date'
execute :ls, '-l'
with rails_env: :production do
within '/tmp' do
as :root do
execute :touch, 'restart.txt'
end
end
end
end.run
command_lines = @output.lines.select { |line| line.start_with?('Command:') }
assert_equal <<-EOEXPECTED.unindent, command_lines.join
Command: /usr/bin/env date
Command: /usr/bin/env ls -l
Command: if test ! -d /tmp; then echo \"Directory does not exist '/tmp'\" 1>&2; false; fi
Command: if ! sudo -u root whoami > /dev/null; then echo \"You cannot switch to user 'root' using sudo, please check the sudoers file\" 1>&2; false; fi
Command: cd /tmp && ( export RAILS_ENV="production" ; sudo -u root RAILS_ENV="production" -- sh -c '/usr/bin/env touch restart.txt' )
EOEXPECTED
end
def test_capture
captured_command_result = nil
Netssh.new(a_host) do |_host|
captured_command_result = capture(:uname)
end.run
assert_includes %W(Linux Darwin), captured_command_result
end
def test_ssh_option_merge
a_host.ssh_options = { paranoid: true }
host_ssh_options = {}
SSHKit::Backend::Netssh.config.ssh_options = { forward_agent: false }
Netssh.new(a_host) do |host|
capture(:uname)
host_ssh_options = host.ssh_options
end.run
assert_equal({ forward_agent: false, paranoid: true }, host_ssh_options)
end
def test_env_vars_substituion_in_subshell
captured_command_result = nil
Netssh.new(a_host) do |_host|
with some_env_var: :some_value do
captured_command_result = capture(:echo, '$SOME_ENV_VAR')
end
end.run
assert_equal "some_value", captured_command_result
end
def test_execute_raises_on_non_zero_exit_status_and_captures_stdout_and_stderr
err = assert_raises SSHKit::Command::Failed do
Netssh.new(a_host) do |_host|
execute :echo, "'Test capturing stderr' 1>&2; false"
end.run
end
assert_equal "echo exit status: 1\necho stdout: Nothing written\necho stderr: Test capturing stderr\n", err.message
end
def test_test_does_not_raise_on_non_zero_exit_status
Netssh.new(a_host) do |_host|
test :false
end.run
end
def test_upload_and_then_capture_file_contents
actual_file_contents = ""
file_name = File.join("/tmp", SecureRandom.uuid)
File.open file_name, 'w+' do |f|
f.write "Some Content\nWith a newline and trailing spaces \n "
end
Netssh.new(a_host) do
upload!(file_name, file_name)
actual_file_contents = capture(:cat, file_name, strip: false)
end.run
assert_equal "Some Content\nWith a newline and trailing spaces \n ", actual_file_contents
end
def test_upload_string_io
file_contents = ""
Netssh.new(a_host) do |_host|
file_name = File.join("/tmp", SecureRandom.uuid)
upload!(StringIO.new('example_io'), file_name)
file_contents = download!(file_name)
end.run
assert_equal "example_io", file_contents
end
def test_upload_large_file
size = 25
fills = SecureRandom.random_bytes(1024*1024)
file_name = "/tmp/file-#{size}.txt"
File.open(file_name, 'w') do |f|
(size).times {f.write(fills) }
end
file_contents = ""
Netssh.new(a_host) do
upload!(file_name, file_name)
file_contents = download!(file_name)
end.run
assert_equal File.open(file_name).read, file_contents
end
def test_interaction_handler
captured_command_result = nil
Netssh.new(a_host) do
command = 'echo Enter Data; read the_data; echo Captured $the_data;'
captured_command_result = capture(command, interaction_handler: {
"Enter Data\n" => "SOME DATA\n",
"Captured SOME DATA\n" => nil
})
end.run
assert_equal("Enter Data\nCaptured SOME DATA", captured_command_result)
end
end
end
end
|
