net-ssh-net-ssh 2.0.12

Sign up to get free protection for your applications and to get access to all the features.
Files changed (105) hide show
  1. data/CHANGELOG.rdoc +137 -0
  2. data/Manifest +104 -0
  3. data/README.rdoc +110 -0
  4. data/Rakefile +79 -0
  5. data/THANKS.rdoc +16 -0
  6. data/lib/net/ssh.rb +215 -0
  7. data/lib/net/ssh/authentication/agent.rb +176 -0
  8. data/lib/net/ssh/authentication/constants.rb +18 -0
  9. data/lib/net/ssh/authentication/key_manager.rb +193 -0
  10. data/lib/net/ssh/authentication/methods/abstract.rb +60 -0
  11. data/lib/net/ssh/authentication/methods/hostbased.rb +71 -0
  12. data/lib/net/ssh/authentication/methods/keyboard_interactive.rb +66 -0
  13. data/lib/net/ssh/authentication/methods/password.rb +39 -0
  14. data/lib/net/ssh/authentication/methods/publickey.rb +92 -0
  15. data/lib/net/ssh/authentication/pageant.rb +183 -0
  16. data/lib/net/ssh/authentication/session.rb +134 -0
  17. data/lib/net/ssh/buffer.rb +340 -0
  18. data/lib/net/ssh/buffered_io.rb +149 -0
  19. data/lib/net/ssh/config.rb +181 -0
  20. data/lib/net/ssh/connection/channel.rb +625 -0
  21. data/lib/net/ssh/connection/constants.rb +33 -0
  22. data/lib/net/ssh/connection/session.rb +596 -0
  23. data/lib/net/ssh/connection/term.rb +178 -0
  24. data/lib/net/ssh/errors.rb +85 -0
  25. data/lib/net/ssh/key_factory.rb +102 -0
  26. data/lib/net/ssh/known_hosts.rb +129 -0
  27. data/lib/net/ssh/loggable.rb +61 -0
  28. data/lib/net/ssh/packet.rb +102 -0
  29. data/lib/net/ssh/prompt.rb +93 -0
  30. data/lib/net/ssh/proxy/errors.rb +14 -0
  31. data/lib/net/ssh/proxy/http.rb +94 -0
  32. data/lib/net/ssh/proxy/socks4.rb +70 -0
  33. data/lib/net/ssh/proxy/socks5.rb +129 -0
  34. data/lib/net/ssh/ruby_compat.rb +7 -0
  35. data/lib/net/ssh/service/forward.rb +267 -0
  36. data/lib/net/ssh/test.rb +89 -0
  37. data/lib/net/ssh/test/channel.rb +129 -0
  38. data/lib/net/ssh/test/extensions.rb +152 -0
  39. data/lib/net/ssh/test/kex.rb +44 -0
  40. data/lib/net/ssh/test/local_packet.rb +51 -0
  41. data/lib/net/ssh/test/packet.rb +81 -0
  42. data/lib/net/ssh/test/remote_packet.rb +38 -0
  43. data/lib/net/ssh/test/script.rb +157 -0
  44. data/lib/net/ssh/test/socket.rb +59 -0
  45. data/lib/net/ssh/transport/algorithms.rb +384 -0
  46. data/lib/net/ssh/transport/cipher_factory.rb +84 -0
  47. data/lib/net/ssh/transport/constants.rb +30 -0
  48. data/lib/net/ssh/transport/hmac.rb +31 -0
  49. data/lib/net/ssh/transport/hmac/abstract.rb +78 -0
  50. data/lib/net/ssh/transport/hmac/md5.rb +12 -0
  51. data/lib/net/ssh/transport/hmac/md5_96.rb +11 -0
  52. data/lib/net/ssh/transport/hmac/none.rb +15 -0
  53. data/lib/net/ssh/transport/hmac/sha1.rb +13 -0
  54. data/lib/net/ssh/transport/hmac/sha1_96.rb +11 -0
  55. data/lib/net/ssh/transport/identity_cipher.rb +55 -0
  56. data/lib/net/ssh/transport/kex.rb +13 -0
  57. data/lib/net/ssh/transport/kex/diffie_hellman_group1_sha1.rb +208 -0
  58. data/lib/net/ssh/transport/kex/diffie_hellman_group_exchange_sha1.rb +77 -0
  59. data/lib/net/ssh/transport/openssl.rb +128 -0
  60. data/lib/net/ssh/transport/packet_stream.rb +230 -0
  61. data/lib/net/ssh/transport/server_version.rb +69 -0
  62. data/lib/net/ssh/transport/session.rb +276 -0
  63. data/lib/net/ssh/transport/state.rb +206 -0
  64. data/lib/net/ssh/verifiers/lenient.rb +30 -0
  65. data/lib/net/ssh/verifiers/null.rb +12 -0
  66. data/lib/net/ssh/verifiers/strict.rb +53 -0
  67. data/lib/net/ssh/version.rb +62 -0
  68. data/net-ssh.gemspec +128 -0
  69. data/setup.rb +1585 -0
  70. data/test/authentication/methods/common.rb +28 -0
  71. data/test/authentication/methods/test_abstract.rb +51 -0
  72. data/test/authentication/methods/test_hostbased.rb +114 -0
  73. data/test/authentication/methods/test_keyboard_interactive.rb +98 -0
  74. data/test/authentication/methods/test_password.rb +50 -0
  75. data/test/authentication/methods/test_publickey.rb +127 -0
  76. data/test/authentication/test_agent.rb +205 -0
  77. data/test/authentication/test_key_manager.rb +105 -0
  78. data/test/authentication/test_session.rb +93 -0
  79. data/test/common.rb +106 -0
  80. data/test/configs/eqsign +3 -0
  81. data/test/configs/exact_match +8 -0
  82. data/test/configs/wild_cards +14 -0
  83. data/test/connection/test_channel.rb +452 -0
  84. data/test/connection/test_session.rb +488 -0
  85. data/test/test_all.rb +6 -0
  86. data/test/test_buffer.rb +336 -0
  87. data/test/test_buffered_io.rb +63 -0
  88. data/test/test_config.rb +84 -0
  89. data/test/test_key_factory.rb +67 -0
  90. data/test/transport/hmac/test_md5.rb +39 -0
  91. data/test/transport/hmac/test_md5_96.rb +25 -0
  92. data/test/transport/hmac/test_none.rb +34 -0
  93. data/test/transport/hmac/test_sha1.rb +34 -0
  94. data/test/transport/hmac/test_sha1_96.rb +25 -0
  95. data/test/transport/kex/test_diffie_hellman_group1_sha1.rb +146 -0
  96. data/test/transport/kex/test_diffie_hellman_group_exchange_sha1.rb +92 -0
  97. data/test/transport/test_algorithms.rb +302 -0
  98. data/test/transport/test_cipher_factory.rb +171 -0
  99. data/test/transport/test_hmac.rb +34 -0
  100. data/test/transport/test_identity_cipher.rb +40 -0
  101. data/test/transport/test_packet_stream.rb +435 -0
  102. data/test/transport/test_server_version.rb +68 -0
  103. data/test/transport/test_session.rb +315 -0
  104. data/test/transport/test_state.rb +173 -0
  105. metadata +162 -0
@@ -0,0 +1,488 @@
1
+ require 'common'
2
+ require 'net/ssh/connection/session'
3
+
4
+ module Connection
5
+
6
+ class TestSession < Test::Unit::TestCase
7
+ include Net::SSH::Connection::Constants
8
+
9
+ def test_constructor_should_set_defaults
10
+ assert session.channels.empty?
11
+ assert session.pending_requests.empty?
12
+ assert_equal({ socket => nil }, session.listeners)
13
+ end
14
+
15
+ def test_on_open_channel_should_register_block_with_given_channel_type
16
+ flag = false
17
+ session.on_open_channel("testing") { flag = true }
18
+ assert_not_nil session.channel_open_handlers["testing"]
19
+ session.channel_open_handlers["testing"].call
20
+ assert flag, "callback should have been invoked"
21
+ end
22
+
23
+ def test_forward_should_create_and_cache_instance_of_forward_service
24
+ assert_instance_of Net::SSH::Service::Forward, session.forward
25
+ assert_equal session.forward.object_id, session.forward.object_id
26
+ end
27
+
28
+ def test_listen_to_without_callback_should_add_argument_as_listener
29
+ io = stub("io")
30
+ session.listen_to(io)
31
+ assert session.listeners.key?(io)
32
+ assert_nil session.listeners[io]
33
+ end
34
+
35
+ def test_listen_to_should_add_argument_to_listeners_list_if_block_is_given
36
+ io = stub("io", :pending_write? => true)
37
+ flag = false
38
+ session.listen_to(io) { flag = true }
39
+ assert !flag, "callback should not be invoked immediately"
40
+ assert session.listeners.key?(io)
41
+ session.listeners[io].call
42
+ assert flag, "callback should have been invoked"
43
+ end
44
+
45
+ def test_stop_listening_to_should_remove_argument_from_listeners
46
+ io = stub("io", :pending_write? => true)
47
+
48
+ session.listen_to(io)
49
+ assert session.listeners.key?(io)
50
+
51
+ session.stop_listening_to(io)
52
+ assert !session.listeners.key?(io)
53
+ end
54
+
55
+ def test_send_message_should_enqueue_message_at_transport_layer
56
+ packet = P(:byte, REQUEST_SUCCESS)
57
+ session.send_message(packet)
58
+ assert_equal packet.to_s, socket.write_buffer
59
+ end
60
+
61
+ def test_open_channel_defaults_should_use_session_channel
62
+ flag = false
63
+ channel = session.open_channel { flag = true }
64
+ assert !flag, "callback should not be invoked immediately"
65
+ channel.do_open_confirmation(1,2,3)
66
+ assert flag, "callback should have been invoked"
67
+ assert_equal "session", channel.type
68
+ assert_equal 0, channel.local_id
69
+ assert_equal channel, session.channels[channel.local_id]
70
+
71
+ packet = P(:byte, CHANNEL_OPEN, :string, "session", :long, channel.local_id,
72
+ :long, channel.local_maximum_window_size, :long, channel.local_maximum_packet_size)
73
+ assert_equal packet.to_s, socket.write_buffer
74
+ end
75
+
76
+ def test_open_channel_with_type_should_use_type
77
+ channel = session.open_channel("direct-tcpip")
78
+ assert_equal "direct-tcpip", channel.type
79
+ packet = P(:byte, CHANNEL_OPEN, :string, "direct-tcpip", :long, channel.local_id,
80
+ :long, channel.local_maximum_window_size, :long, channel.local_maximum_packet_size)
81
+ assert_equal packet.to_s, socket.write_buffer
82
+ end
83
+
84
+ def test_open_channel_with_extras_should_append_extras_to_packet
85
+ channel = session.open_channel("direct-tcpip", :string, "other.host", :long, 1234)
86
+ packet = P(:byte, CHANNEL_OPEN, :string, "direct-tcpip", :long, channel.local_id,
87
+ :long, channel.local_maximum_window_size, :long, channel.local_maximum_packet_size,
88
+ :string, "other.host", :long, 1234)
89
+ assert_equal packet.to_s, socket.write_buffer
90
+ end
91
+
92
+ def test_send_global_request_without_callback_should_not_expect_reply
93
+ packet = P(:byte, GLOBAL_REQUEST, :string, "testing", :bool, false)
94
+ session.send_global_request("testing")
95
+ assert_equal packet.to_s, socket.write_buffer
96
+ assert session.pending_requests.empty?
97
+ end
98
+
99
+ def test_send_global_request_with_callback_should_expect_reply
100
+ packet = P(:byte, GLOBAL_REQUEST, :string, "testing", :bool, true)
101
+ proc = Proc.new {}
102
+ session.send_global_request("testing", &proc)
103
+ assert_equal packet.to_s, socket.write_buffer
104
+ assert_equal [proc], session.pending_requests
105
+ end
106
+
107
+ def test_send_global_request_with_extras_should_append_extras_to_packet
108
+ packet = P(:byte, GLOBAL_REQUEST, :string, "testing", :bool, false, :string, "other.host", :long, 1234)
109
+ session.send_global_request("testing", :string, "other.host", :long, 1234)
110
+ assert_equal packet.to_s, socket.write_buffer
111
+ end
112
+
113
+ def test_process_should_exit_immediately_if_block_is_false
114
+ session.channels[0] = stub("channel", :closing? => false)
115
+ session.channels[0].expects(:process).never
116
+ process_times(0)
117
+ end
118
+
119
+ def test_process_should_exit_after_processing_if_block_is_true_then_false
120
+ session.channels[0] = stub("channel", :closing? => false)
121
+ session.channels[0].expects(:process)
122
+ IO.expects(:select).never
123
+ process_times(2)
124
+ end
125
+
126
+ def test_process_should_not_process_channels_that_are_closing
127
+ session.channels[0] = stub("channel", :closing? => true)
128
+ session.channels[0].expects(:process).never
129
+ IO.expects(:select).never
130
+ process_times(2)
131
+ end
132
+
133
+ def test_global_request_packets_should_be_silently_handled_if_no_handler_exists_for_them
134
+ transport.return(GLOBAL_REQUEST, :string, "testing", :bool, false)
135
+ process_times(2)
136
+ assert transport.queue.empty?
137
+ assert !socket.pending_write?
138
+ end
139
+
140
+ def test_global_request_packets_should_be_auto_replied_to_even_if_no_handler_exists
141
+ transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
142
+ process_times(2)
143
+ assert_equal P(:byte, REQUEST_FAILURE).to_s, socket.write_buffer
144
+ end
145
+
146
+ def test_global_request_handler_should_not_trigger_auto_reply_if_no_reply_is_wanted
147
+ flag = false
148
+ session.on_global_request("testing") { flag = true }
149
+ assert !flag, "callback should not be invoked yet"
150
+ transport.return(GLOBAL_REQUEST, :string, "testing", :bool, false)
151
+ process_times(2)
152
+ assert transport.queue.empty?
153
+ assert !socket.pending_write?
154
+ assert flag, "callback should have been invoked"
155
+ end
156
+
157
+ def test_global_request_handler_returning_true_should_trigger_success_auto_reply
158
+ flag = false
159
+ session.on_global_request("testing") { flag = true }
160
+ transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
161
+ process_times(2)
162
+ assert_equal P(:byte, REQUEST_SUCCESS).to_s, socket.write_buffer
163
+ assert flag
164
+ end
165
+
166
+ def test_global_request_handler_returning_false_should_trigger_failure_auto_reply
167
+ flag = false
168
+ session.on_global_request("testing") { flag = true; false }
169
+ transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
170
+ process_times(2)
171
+ assert_equal P(:byte, REQUEST_FAILURE).to_s, socket.write_buffer
172
+ assert flag
173
+ end
174
+
175
+ def test_global_request_handler_returning_sent_should_not_trigger_auto_reply
176
+ flag = false
177
+ session.on_global_request("testing") { flag = true; :sent }
178
+ transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
179
+ process_times(2)
180
+ assert !socket.pending_write?
181
+ assert flag
182
+ end
183
+
184
+ def test_global_request_handler_returning_other_value_should_raise_error
185
+ session.on_global_request("testing") { "bug" }
186
+ transport.return(GLOBAL_REQUEST, :string, "testing", :bool, true)
187
+ assert_raises(RuntimeError) { process_times(2) }
188
+ end
189
+
190
+ def test_request_success_packets_should_invoke_next_pending_request_with_true
191
+ result = nil
192
+ session.pending_requests << Proc.new { |*args| result = args }
193
+ transport.return(REQUEST_SUCCESS)
194
+ process_times(2)
195
+ assert_equal [true, P(:byte, REQUEST_SUCCESS)], result
196
+ assert session.pending_requests.empty?
197
+ end
198
+
199
+ def test_request_failure_packets_should_invoke_next_pending_request_with_false
200
+ result = nil
201
+ session.pending_requests << Proc.new { |*args| result = args }
202
+ transport.return(REQUEST_FAILURE)
203
+ process_times(2)
204
+ assert_equal [false, P(:byte, REQUEST_FAILURE)], result
205
+ assert session.pending_requests.empty?
206
+ end
207
+
208
+ def test_channel_open_packet_without_corresponding_channel_open_handler_should_result_in_channel_open_failure
209
+ transport.return(CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20000, :long, 0x10000)
210
+ process_times(2)
211
+ assert_equal P(:byte, CHANNEL_OPEN_FAILURE, :long, 14, :long, 3, :string, "unknown channel type auth-agent", :string, "").to_s, socket.write_buffer
212
+ end
213
+
214
+ def test_channel_open_packet_with_corresponding_handler_should_result_in_channel_open_failure_when_handler_returns_an_error
215
+ transport.return(CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20000, :long, 0x10000)
216
+ session.on_open_channel "auth-agent" do |s, ch, p|
217
+ raise Net::SSH::ChannelOpenFailed.new(1234, "we iz in ur channelz!")
218
+ end
219
+ process_times(2)
220
+ assert_equal P(:byte, CHANNEL_OPEN_FAILURE, :long, 14, :long, 1234, :string, "we iz in ur channelz!", :string, "").to_s, socket.write_buffer
221
+ end
222
+
223
+ def test_channel_open_packet_with_corresponding_handler_should_result_in_channel_open_confirmation_when_handler_succeeds
224
+ transport.return(CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20001, :long, 0x10001)
225
+ result = nil
226
+ session.on_open_channel("auth-agent") { |*args| result = args }
227
+ process_times(2)
228
+ assert_equal P(:byte, CHANNEL_OPEN_CONFIRMATION, :long, 14, :long, 0, :long, 0x20000, :long, 0x10000).to_s, socket.write_buffer
229
+ assert_not_nil(ch = session.channels[0])
230
+ assert_equal [session, ch, P(:byte, CHANNEL_OPEN, :string, "auth-agent", :long, 14, :long, 0x20001, :long, 0x10001)], result
231
+ assert_equal 0, ch.local_id
232
+ assert_equal 14, ch.remote_id
233
+ assert_equal 0x20001, ch.remote_maximum_window_size
234
+ assert_equal 0x10001, ch.remote_maximum_packet_size
235
+ assert_equal 0x20000, ch.local_maximum_window_size
236
+ assert_equal 0x10000, ch.local_maximum_packet_size
237
+ assert_equal "auth-agent", ch.type
238
+ end
239
+
240
+ def test_channel_open_failure_should_remove_channel_and_tell_channel_that_open_failed
241
+ session.channels[1] = stub("channel")
242
+ session.channels[1].expects(:do_open_failed).with(1234, "some reason")
243
+ transport.return(CHANNEL_OPEN_FAILURE, :long, 1, :long, 1234, :string, "some reason", :string, "lang tag")
244
+ process_times(2)
245
+ assert session.channels.empty?
246
+ end
247
+
248
+ def test_channel_open_confirmation_packet_should_be_routed_to_corresponding_channel
249
+ channel_at(14).expects(:do_open_confirmation).with(1234, 0x20001, 0x10001)
250
+ transport.return(CHANNEL_OPEN_CONFIRMATION, :long, 14, :long, 1234, :long, 0x20001, :long, 0x10001)
251
+ process_times(2)
252
+ end
253
+
254
+ def test_channel_window_adjust_packet_should_be_routed_to_corresponding_channel
255
+ channel_at(14).expects(:do_window_adjust).with(5000)
256
+ transport.return(CHANNEL_WINDOW_ADJUST, :long, 14, :long, 5000)
257
+ process_times(2)
258
+ end
259
+
260
+ def test_channel_request_for_nonexistant_channel_should_be_ignored
261
+ transport.return(CHANNEL_REQUEST, :long, 14, :string, "testing", :bool, false)
262
+ assert_nothing_raised { process_times(2) }
263
+ end
264
+
265
+ def test_channel_request_packet_should_be_routed_to_corresponding_channel
266
+ channel_at(14).expects(:do_request).with("testing", false, Net::SSH::Buffer.new)
267
+ transport.return(CHANNEL_REQUEST, :long, 14, :string, "testing", :bool, false)
268
+ process_times(2)
269
+ end
270
+
271
+ def test_channel_data_packet_should_be_routed_to_corresponding_channel
272
+ channel_at(14).expects(:do_data).with("bring it on down")
273
+ transport.return(CHANNEL_DATA, :long, 14, :string, "bring it on down")
274
+ process_times(2)
275
+ end
276
+
277
+ def test_channel_extended_data_packet_should_be_routed_to_corresponding_channel
278
+ channel_at(14).expects(:do_extended_data).with(1, "bring it on down")
279
+ transport.return(CHANNEL_EXTENDED_DATA, :long, 14, :long, 1, :string, "bring it on down")
280
+ process_times(2)
281
+ end
282
+
283
+ def test_channel_eof_packet_should_be_routed_to_corresponding_channel
284
+ channel_at(14).expects(:do_eof).with()
285
+ transport.return(CHANNEL_EOF, :long, 14)
286
+ process_times(2)
287
+ end
288
+
289
+ def test_channel_success_packet_should_be_routed_to_corresponding_channel
290
+ channel_at(14).expects(:do_success).with()
291
+ transport.return(CHANNEL_SUCCESS, :long, 14)
292
+ process_times(2)
293
+ end
294
+
295
+ def test_channel_failure_packet_should_be_routed_to_corresponding_channel
296
+ channel_at(14).expects(:do_failure).with()
297
+ transport.return(CHANNEL_FAILURE, :long, 14)
298
+ process_times(2)
299
+ end
300
+
301
+ def test_channel_close_packet_should_be_routed_to_corresponding_channel_and_channel_should_be_closed_and_removed
302
+ channel_at(14).expects(:do_close).with()
303
+ session.channels[14].expects(:close).with()
304
+ transport.return(CHANNEL_CLOSE, :long, 14)
305
+ process_times(2)
306
+ assert session.channels.empty?
307
+ end
308
+
309
+ def test_multiple_pending_dispatches_should_be_dispatched_together
310
+ channel_at(14).expects(:do_eof).with()
311
+ session.channels[14].expects(:do_success).with()
312
+ transport.return(CHANNEL_SUCCESS, :long, 14)
313
+ transport.return(CHANNEL_EOF, :long, 14)
314
+ process_times(2)
315
+ end
316
+
317
+ def test_writers_without_pending_writes_should_not_be_considered_for_select
318
+ IO.expects(:select).with([socket],[],nil,nil).returns([[],[],[]])
319
+ session.process
320
+ end
321
+
322
+ def test_writers_with_pending_writes_should_be_considered_for_select
323
+ socket.enqueue("laksdjflasdkf")
324
+ IO.expects(:select).with([socket],[socket],nil,nil).returns([[],[],[]])
325
+ session.process
326
+ end
327
+
328
+ def test_ready_readers_should_be_filled
329
+ socket.expects(:recv).returns("this is some data")
330
+ IO.expects(:select).with([socket],[],nil,nil).returns([[socket],[],[]])
331
+ session.process
332
+ assert_equal [socket], session.listeners.keys
333
+ end
334
+
335
+ def test_ready_readers_that_cant_be_filled_should_be_removed
336
+ socket.expects(:recv).returns("")
337
+ socket.expects(:close)
338
+ IO.expects(:select).with([socket],[],nil,nil).returns([[socket],[],[]])
339
+ session.process
340
+ assert session.listeners.empty?
341
+ end
342
+
343
+ def test_ready_readers_that_are_registered_with_a_block_should_call_block_instead_of_fill
344
+ io = stub("io", :pending_write? => false)
345
+ flag = false
346
+ session.stop_listening_to(socket) # so that we only have to test the presence of a single IO object
347
+ session.listen_to(io) { flag = true }
348
+ IO.expects(:select).with([io],[],nil,nil).returns([[io],[],[]])
349
+ session.process
350
+ assert flag, "callback should have been invoked"
351
+ end
352
+
353
+ def test_ready_writers_should_call_send_pending
354
+ socket.enqueue("laksdjflasdkf")
355
+ socket.expects(:send).with("laksdjflasdkf", 0).returns(13)
356
+ IO.expects(:select).with([socket],[socket],nil,nil).returns([[],[socket],[]])
357
+ session.process
358
+ end
359
+
360
+ def test_process_should_call_rekey_as_needed
361
+ transport.expects(:rekey_as_needed)
362
+ IO.expects(:select).with([socket],[],nil,nil).returns([[],[],[]])
363
+ session.process
364
+ end
365
+
366
+ def test_loop_should_call_process_until_process_returns_false
367
+ IO.stubs(:select).with([socket],[],nil,nil).returns([[],[],[]])
368
+ session.expects(:process).with(nil).times(4).returns(true,true,true,false).yields
369
+ n = 0
370
+ session.loop { n += 1 }
371
+ assert_equal 4, n
372
+ end
373
+
374
+ def test_exec_should_open_channel_and_configure_default_callbacks
375
+ prep_exec("ls", :stdout, "data packet", :stderr, "extended data packet")
376
+
377
+ call = :first
378
+ session.exec "ls" do |channel, type, data|
379
+ if call == :first
380
+ assert_equal :stdout, type
381
+ assert_equal "data packet", data
382
+ call = :second
383
+ elsif call == :second
384
+ assert_equal :stderr, type
385
+ assert_equal "extended data packet", data
386
+ call = :third
387
+ else
388
+ flunk "should never get here, call == #{call.inspect}"
389
+ end
390
+ end
391
+
392
+ session.loop
393
+ assert_equal :third, call
394
+ end
395
+
396
+ def test_exec_without_block_should_use_print_to_display_result
397
+ prep_exec("ls", :stdout, "data packet", :stderr, "extended data packet")
398
+ $stdout.expects(:print).with("data packet")
399
+ $stderr.expects(:print).with("extended data packet")
400
+
401
+ session.exec "ls"
402
+ session.loop
403
+ end
404
+
405
+ def test_exec_bang_should_block_until_command_finishes
406
+ prep_exec("ls", :stdout, "some data")
407
+ called = false
408
+ session.exec! "ls" do |channel, type, data|
409
+ called = true
410
+ assert_equal :stdout, type
411
+ assert_equal "some data", data
412
+ end
413
+ assert called
414
+ end
415
+
416
+ def test_exec_bang_without_block_should_return_data_as_string
417
+ prep_exec("ls", :stdout, "some data")
418
+ assert_equal "some data", session.exec!("ls")
419
+ end
420
+
421
+ private
422
+
423
+ def prep_exec(command, *data)
424
+ transport.mock_enqueue = true
425
+ transport.expect do |t, p|
426
+ assert_equal CHANNEL_OPEN, p.type
427
+ t.return(CHANNEL_OPEN_CONFIRMATION, :long, p[:remote_id], :long, 0, :long, 0x20000, :long, 0x10000)
428
+ t.expect do |t2, p2|
429
+ assert_equal CHANNEL_REQUEST, p2.type
430
+ assert_equal "exec", p2[:request]
431
+ assert_equal true, p2[:want_reply]
432
+ assert_equal "ls", p2.read_string
433
+
434
+ t2.return(CHANNEL_SUCCESS, :long, p[:remote_id])
435
+
436
+ 0.step(data.length-1, 2) do |index|
437
+ type = data[index]
438
+ datum = data[index+1]
439
+
440
+ if type == :stdout
441
+ t2.return(CHANNEL_DATA, :long, p[:remote_id], :string, datum)
442
+ else
443
+ t2.return(CHANNEL_EXTENDED_DATA, :long, p[:remote_id], :long, 1, :string, datum)
444
+ end
445
+ end
446
+
447
+ t2.return(CHANNEL_CLOSE, :long, p[:remote_id])
448
+ t2.expect { |t3,p3| assert_equal CHANNEL_CLOSE, p3.type }
449
+ end
450
+ end
451
+ end
452
+
453
+ module MockSocket
454
+ # so that we can easily test the contents that were enqueued, without
455
+ # worrying about all the packet stream overhead
456
+ def enqueue_packet(message)
457
+ enqueue(message.to_s)
458
+ end
459
+ end
460
+
461
+ def socket
462
+ @socket ||= begin
463
+ socket ||= Object.new
464
+ socket.extend(Net::SSH::Transport::PacketStream)
465
+ socket.extend(MockSocket)
466
+ socket
467
+ end
468
+ end
469
+
470
+ def channel_at(local_id)
471
+ session.channels[local_id] = stub("channel", :process => true, :closing? => false)
472
+ end
473
+
474
+ def transport(options={})
475
+ @transport ||= MockTransport.new(options.merge(:socket => socket))
476
+ end
477
+
478
+ def session(options={})
479
+ @session ||= Net::SSH::Connection::Session.new(transport, options)
480
+ end
481
+
482
+ def process_times(n)
483
+ i = 0
484
+ session.process { (i += 1) < n }
485
+ end
486
+ end
487
+
488
+ end