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
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
require "helper"
require "train/transports/local"
class TransportHelper
attr_accessor :transport
def initialize(user_opts = {})
opts = { platform_name: "mock", family_hierarchy: ["mock"] }.merge(user_opts)
plat = Train::Platforms.name(opts[:platform_name])
plat.family_hierarchy = opts[:family_hierarchy]
plat.add_platform_methods
Train::Platforms::Detect.stubs(:scan).returns(plat)
@transport = Train::Transports::Local.new(user_opts)
end
end
describe "local transport" do
let(:transport) { TransportHelper.new.transport }
let(:connection) { transport.connection }
it "can be instantiated" do
_(transport).wont_be_nil
end
it "gets the connection" do
_(connection).must_be_kind_of Train::Transports::Local::Connection
end
it "provides a uri" do
_(connection.uri).must_equal "local://"
end
it "doesnt wait to be read" do
_(connection.wait_until_ready).must_be_nil
end
it "inspects with readable output" do
_(connection.inspect).must_match(/Train::Transports::Local::Connection\[\w+\]/)
end
it "can be closed" do
_(connection.close).must_be_nil
end
it "has no login command" do
_(connection.login_command).must_be_nil
end
it "provides a run_command_via_connection method" do
methods = connection.class.private_instance_methods(false)
_(methods.include?(:run_command_via_connection)).must_equal true
end
it "provides a file_via_connection method" do
methods = connection.class.private_instance_methods(false)
_(methods.include?(:file_via_connection)).must_equal true
end
describe "when overriding runner selection" do
it "can select the `GenericRunner`" do
Train::Transports::Local::Connection::GenericRunner
.expects(:new)
Train::Transports::Local::Connection::WindowsPipeRunner
.expects(:new)
.never
Train::Transports::Local::Connection::WindowsShellRunner
.expects(:new)
.never
Train::Transports::Local::Connection.new(command_runner: :generic)
end
it "can select the `WindowsPipeRunner`" do
Train::Transports::Local::Connection::GenericRunner
.expects(:new)
.never
Train::Transports::Local::Connection::WindowsPipeRunner
.expects(:new)
Train::Transports::Local::Connection::WindowsShellRunner
.expects(:new)
.never
Train::Transports::Local::Connection.new(command_runner: :windows_pipe)
end
it "can select the `WindowsShellRunner`" do
Train::Transports::Local::Connection::GenericRunner
.expects(:new)
.never
Train::Transports::Local::Connection::WindowsPipeRunner
.expects(:new)
.never
Train::Transports::Local::Connection::WindowsShellRunner
.expects(:new)
Train::Transports::Local::Connection.new(command_runner: :windows_shell)
end
it "throws a RuntimeError when an invalid runner type is passed" do
_ { Train::Transports::Local::Connection.new(command_runner: :nope ) }
.must_raise(RuntimeError, "Runner type `:nope` not supported")
end
end
describe "when running a local command" do
let(:cmd_runner) { Minitest::Mock.new }
def mock_run_cmd(cmd, &block)
cmd_runner.expect :run_command, nil
cmd_runner.expect :timeout=, nil, [nil]
Mixlib::ShellOut.stub :new, cmd_runner do |*args|
yield
end
end
it "gets stdout" do
mock_run_cmd(rand) do
x = rand
cmd_runner.expect :stdout, x
cmd_runner.expect :stderr, nil
cmd_runner.expect :exitstatus, nil
_(connection.run_command(rand).stdout).must_equal x
end
end
it "gets stderr" do
mock_run_cmd(rand) do
x = rand
cmd_runner.expect :stdout, nil
cmd_runner.expect :stderr, x
cmd_runner.expect :exitstatus, nil
_(connection.run_command(rand).stderr).must_equal x
end
end
it "gets exit_status" do
mock_run_cmd(rand) do
x = rand
cmd_runner.expect :stdout, nil
cmd_runner.expect :stderr, nil
cmd_runner.expect :exitstatus, x
_(connection.run_command(rand).exit_status).must_equal x
end
end
end
describe "when running on Windows" do
let(:connection) do
TransportHelper.new(family_hierarchy: ["windows"]).transport.connection
end
let(:runner) { mock }
it "uses `WindowsPipeRunner` by default" do
Train::Transports::Local::Connection::WindowsPipeRunner
.expects(:new)
.returns(runner)
Train::Transports::Local::Connection::WindowsShellRunner
.expects(:new)
.never
runner.expects(:run_command).with("not actually executed", {})
connection.run_command("not actually executed")
end
it "uses `WindowsShellRunner` when a named pipe is not available" do
Train::Transports::Local::Connection::WindowsPipeRunner
.expects(:new)
.raises(Train::Transports::Local::PipeError)
Train::Transports::Local::Connection::WindowsShellRunner
.expects(:new)
.returns(runner)
runner.expects(:run_command).with("not actually executed", {})
connection.run_command("not actually executed")
end
end
end
|
