poolparty 1.3.4 → 1.3.6

Sign up to get free protection for your applications and to get access to all the features.
Files changed (174) hide show
  1. data/Rakefile +2 -2
  2. data/VERSION.yml +1 -1
  3. data/bin/cloud-bootstrap +1 -0
  4. data/bin/cloud-configure +1 -0
  5. data/bin/cloud-contract +1 -0
  6. data/bin/cloud-misc +34 -0
  7. data/bin/cloud-setup +36 -0
  8. data/bin/cloud-ssh +4 -1
  9. data/config/jeweler.rb +4 -3
  10. data/examples/monitored_cloud.rb +1 -1
  11. data/examples/thrift/thrift_example.rb +5 -3
  12. data/examples/vmware.rb +28 -0
  13. data/lib/cloud_providers/cloud_provider_instance.rb +14 -5
  14. data/lib/cloud_providers/connections.rb +1 -1
  15. data/lib/core/file.rb +12 -0
  16. data/lib/core/object.rb +2 -2
  17. data/lib/dependency_resolvers/base.rb +1 -1
  18. data/lib/dependency_resolvers/chef.rb +9 -7
  19. data/lib/dependency_resolvers/proxy_object.rb +11 -3
  20. data/lib/mixins/askable.rb +16 -7
  21. data/lib/poolparty/base.rb +8 -7
  22. data/lib/poolparty/cloud.rb +77 -7
  23. data/lib/poolparty/default.rb +1 -0
  24. data/lib/poolparty/installer.rb +8 -4
  25. data/lib/poolparty/installers/ec2.rb +75 -5
  26. data/lib/poolparty/installers/vmware.rb +17 -5
  27. data/lib/poolparty/plugin.rb +1 -5
  28. data/lib/poolparty/plugins/apache.rb +10 -7
  29. data/lib/poolparty/plugins/apache2/base.conf.erb +2 -2
  30. data/lib/poolparty/plugins/apache2/browser_fixes.conf.erb +1 -1
  31. data/lib/poolparty/plugins/apache2/passenger_site.rb +2 -2
  32. data/lib/poolparty/plugins/collectd/templates/collectd.conf.erb +369 -0
  33. data/lib/poolparty/plugins/collectd.rb +24 -0
  34. data/lib/poolparty/plugins/hermes.rb +89 -0
  35. data/lib/poolparty/pool.rb +33 -3
  36. data/lib/poolparty/resource.rb +32 -18
  37. data/lib/poolparty/resources/directory.rb +5 -1
  38. data/lib/poolparty/resources/exec.rb +2 -2
  39. data/lib/poolparty/resources/file.rb +8 -2
  40. data/lib/poolparty/resources/gem_package.rb +2 -2
  41. data/lib/poolparty/resources/line.rb +23 -6
  42. data/lib/poolparty/resources/mount.rb +2 -2
  43. data/lib/poolparty/resources/package.rb +2 -2
  44. data/lib/poolparty/resources/service.rb +2 -2
  45. data/lib/poolparty/resources/user.rb +2 -2
  46. data/lib/poolparty/resources/variable.rb +4 -3
  47. data/lib/poolparty.rb +5 -3
  48. data/lib/proto/command_interface_handler.rb +17 -1
  49. data/lib/proto/gen-py/cloudthrift/CommandInterface.pyc +0 -0
  50. data/lib/proto/gen-py/cloudthrift/__init__.pyc +0 -0
  51. data/lib/proto/gen-py/cloudthrift/constants.pyc +0 -0
  52. data/lib/proto/gen-py/cloudthrift/ttypes.pyc +0 -0
  53. data/lib/proto/gen-py/thrift/Thrift.pyc +0 -0
  54. data/lib/proto/gen-py/thrift/__init__.pyc +0 -0
  55. data/lib/proto/gen-py/thrift/protocol/TBinaryProtocol.pyc +0 -0
  56. data/lib/proto/gen-py/thrift/protocol/TProtocol.pyc +0 -0
  57. data/lib/proto/gen-py/thrift/protocol/__init__.pyc +0 -0
  58. data/lib/proto/gen-py/thrift/transport/TSocket.pyc +0 -0
  59. data/lib/proto/gen-py/thrift/transport/TTransport.pyc +0 -0
  60. data/lib/proto/gen-py/thrift/transport/__init__.pyc +0 -0
  61. data/test/lib/dependency_resolvers/chef_test.rb +92 -100
  62. data/test/lib/poolparty/base_test.rb +13 -0
  63. data/test/lib/poolparty/cloud_test.rb +50 -2
  64. data/test/lib/poolparty/monitor_test.rb +2 -2
  65. data/test/lib/poolparty/resource_test.rb +5 -0
  66. data/test/lib/poolparty/resources/line_test.rb +3 -3
  67. data/test/lib/poolparty/resources/service_test.rb +1 -1
  68. data/test/lib/poolparty/resources/variable_test.rb +33 -10
  69. data/vendor/gems/net-ssh/CHANGELOG.rdoc +127 -0
  70. data/vendor/gems/net-ssh/Manifest +104 -0
  71. data/vendor/gems/net-ssh/README.rdoc +110 -0
  72. data/vendor/gems/net-ssh/Rakefile +26 -0
  73. data/vendor/gems/net-ssh/THANKS.rdoc +16 -0
  74. data/vendor/gems/net-ssh/lib/net/ssh/authentication/agent.rb +176 -0
  75. data/vendor/gems/net-ssh/lib/net/ssh/authentication/constants.rb +18 -0
  76. data/vendor/gems/net-ssh/lib/net/ssh/authentication/key_manager.rb +193 -0
  77. data/vendor/gems/net-ssh/lib/net/ssh/authentication/methods/abstract.rb +60 -0
  78. data/vendor/gems/net-ssh/lib/net/ssh/authentication/methods/hostbased.rb +71 -0
  79. data/vendor/gems/net-ssh/lib/net/ssh/authentication/methods/keyboard_interactive.rb +66 -0
  80. data/vendor/gems/net-ssh/lib/net/ssh/authentication/methods/password.rb +39 -0
  81. data/vendor/gems/net-ssh/lib/net/ssh/authentication/methods/publickey.rb +92 -0
  82. data/vendor/gems/net-ssh/lib/net/ssh/authentication/pageant.rb +183 -0
  83. data/vendor/gems/net-ssh/lib/net/ssh/authentication/session.rb +134 -0
  84. data/vendor/gems/net-ssh/lib/net/ssh/buffer.rb +340 -0
  85. data/vendor/gems/net-ssh/lib/net/ssh/buffered_io.rb +149 -0
  86. data/vendor/gems/net-ssh/lib/net/ssh/config.rb +181 -0
  87. data/vendor/gems/net-ssh/lib/net/ssh/connection/channel.rb +625 -0
  88. data/vendor/gems/net-ssh/lib/net/ssh/connection/constants.rb +33 -0
  89. data/vendor/gems/net-ssh/lib/net/ssh/connection/session.rb +596 -0
  90. data/vendor/gems/net-ssh/lib/net/ssh/connection/term.rb +178 -0
  91. data/vendor/gems/net-ssh/lib/net/ssh/errors.rb +85 -0
  92. data/vendor/gems/net-ssh/lib/net/ssh/key_factory.rb +102 -0
  93. data/vendor/gems/net-ssh/lib/net/ssh/known_hosts.rb +129 -0
  94. data/vendor/gems/net-ssh/lib/net/ssh/loggable.rb +61 -0
  95. data/vendor/gems/net-ssh/lib/net/ssh/packet.rb +102 -0
  96. data/vendor/gems/net-ssh/lib/net/ssh/prompt.rb +93 -0
  97. data/vendor/gems/net-ssh/lib/net/ssh/proxy/errors.rb +14 -0
  98. data/vendor/gems/net-ssh/lib/net/ssh/proxy/http.rb +94 -0
  99. data/vendor/gems/net-ssh/lib/net/ssh/proxy/socks4.rb +70 -0
  100. data/vendor/gems/net-ssh/lib/net/ssh/proxy/socks5.rb +129 -0
  101. data/vendor/gems/net-ssh/lib/net/ssh/ruby_compat.rb +7 -0
  102. data/vendor/gems/net-ssh/lib/net/ssh/service/forward.rb +267 -0
  103. data/vendor/gems/net-ssh/lib/net/ssh/test/channel.rb +129 -0
  104. data/vendor/gems/net-ssh/lib/net/ssh/test/extensions.rb +152 -0
  105. data/vendor/gems/net-ssh/lib/net/ssh/test/kex.rb +44 -0
  106. data/vendor/gems/net-ssh/lib/net/ssh/test/local_packet.rb +51 -0
  107. data/vendor/gems/net-ssh/lib/net/ssh/test/packet.rb +81 -0
  108. data/vendor/gems/net-ssh/lib/net/ssh/test/remote_packet.rb +38 -0
  109. data/vendor/gems/net-ssh/lib/net/ssh/test/script.rb +157 -0
  110. data/vendor/gems/net-ssh/lib/net/ssh/test/socket.rb +59 -0
  111. data/vendor/gems/net-ssh/lib/net/ssh/test.rb +89 -0
  112. data/vendor/gems/net-ssh/lib/net/ssh/transport/algorithms.rb +384 -0
  113. data/vendor/gems/net-ssh/lib/net/ssh/transport/cipher_factory.rb +80 -0
  114. data/vendor/gems/net-ssh/lib/net/ssh/transport/constants.rb +30 -0
  115. data/vendor/gems/net-ssh/lib/net/ssh/transport/hmac/abstract.rb +78 -0
  116. data/vendor/gems/net-ssh/lib/net/ssh/transport/hmac/md5.rb +12 -0
  117. data/vendor/gems/net-ssh/lib/net/ssh/transport/hmac/md5_96.rb +11 -0
  118. data/vendor/gems/net-ssh/lib/net/ssh/transport/hmac/none.rb +15 -0
  119. data/vendor/gems/net-ssh/lib/net/ssh/transport/hmac/sha1.rb +13 -0
  120. data/vendor/gems/net-ssh/lib/net/ssh/transport/hmac/sha1_96.rb +11 -0
  121. data/vendor/gems/net-ssh/lib/net/ssh/transport/hmac.rb +31 -0
  122. data/vendor/gems/net-ssh/lib/net/ssh/transport/identity_cipher.rb +55 -0
  123. data/vendor/gems/net-ssh/lib/net/ssh/transport/kex/diffie_hellman_group1_sha1.rb +208 -0
  124. data/vendor/gems/net-ssh/lib/net/ssh/transport/kex/diffie_hellman_group_exchange_sha1.rb +77 -0
  125. data/vendor/gems/net-ssh/lib/net/ssh/transport/kex.rb +13 -0
  126. data/vendor/gems/net-ssh/lib/net/ssh/transport/openssl.rb +128 -0
  127. data/vendor/gems/net-ssh/lib/net/ssh/transport/packet_stream.rb +230 -0
  128. data/vendor/gems/net-ssh/lib/net/ssh/transport/server_version.rb +60 -0
  129. data/vendor/gems/net-ssh/lib/net/ssh/transport/session.rb +276 -0
  130. data/vendor/gems/net-ssh/lib/net/ssh/transport/state.rb +201 -0
  131. data/vendor/gems/net-ssh/lib/net/ssh/verifiers/lenient.rb +30 -0
  132. data/vendor/gems/net-ssh/lib/net/ssh/verifiers/null.rb +12 -0
  133. data/vendor/gems/net-ssh/lib/net/ssh/verifiers/strict.rb +53 -0
  134. data/vendor/gems/net-ssh/lib/net/ssh/version.rb +62 -0
  135. data/vendor/gems/net-ssh/lib/net/ssh.rb +215 -0
  136. data/vendor/gems/net-ssh/net-ssh.gemspec +33 -0
  137. data/vendor/gems/net-ssh/setup.rb +1585 -0
  138. data/vendor/gems/net-ssh/test/authentication/methods/common.rb +28 -0
  139. data/vendor/gems/net-ssh/test/authentication/methods/test_abstract.rb +51 -0
  140. data/vendor/gems/net-ssh/test/authentication/methods/test_hostbased.rb +114 -0
  141. data/vendor/gems/net-ssh/test/authentication/methods/test_keyboard_interactive.rb +98 -0
  142. data/vendor/gems/net-ssh/test/authentication/methods/test_password.rb +50 -0
  143. data/vendor/gems/net-ssh/test/authentication/methods/test_publickey.rb +127 -0
  144. data/vendor/gems/net-ssh/test/authentication/test_agent.rb +205 -0
  145. data/vendor/gems/net-ssh/test/authentication/test_key_manager.rb +105 -0
  146. data/vendor/gems/net-ssh/test/authentication/test_session.rb +93 -0
  147. data/vendor/gems/net-ssh/test/common.rb +106 -0
  148. data/vendor/gems/net-ssh/test/configs/eqsign +3 -0
  149. data/vendor/gems/net-ssh/test/configs/exact_match +8 -0
  150. data/vendor/gems/net-ssh/test/configs/wild_cards +14 -0
  151. data/vendor/gems/net-ssh/test/connection/test_channel.rb +452 -0
  152. data/vendor/gems/net-ssh/test/connection/test_session.rb +488 -0
  153. data/vendor/gems/net-ssh/test/test_all.rb +6 -0
  154. data/vendor/gems/net-ssh/test/test_buffer.rb +336 -0
  155. data/vendor/gems/net-ssh/test/test_buffered_io.rb +63 -0
  156. data/vendor/gems/net-ssh/test/test_config.rb +84 -0
  157. data/vendor/gems/net-ssh/test/test_key_factory.rb +67 -0
  158. data/vendor/gems/net-ssh/test/transport/hmac/test_md5.rb +39 -0
  159. data/vendor/gems/net-ssh/test/transport/hmac/test_md5_96.rb +25 -0
  160. data/vendor/gems/net-ssh/test/transport/hmac/test_none.rb +34 -0
  161. data/vendor/gems/net-ssh/test/transport/hmac/test_sha1.rb +34 -0
  162. data/vendor/gems/net-ssh/test/transport/hmac/test_sha1_96.rb +25 -0
  163. data/vendor/gems/net-ssh/test/transport/kex/test_diffie_hellman_group1_sha1.rb +146 -0
  164. data/vendor/gems/net-ssh/test/transport/kex/test_diffie_hellman_group_exchange_sha1.rb +92 -0
  165. data/vendor/gems/net-ssh/test/transport/test_algorithms.rb +302 -0
  166. data/vendor/gems/net-ssh/test/transport/test_cipher_factory.rb +171 -0
  167. data/vendor/gems/net-ssh/test/transport/test_hmac.rb +34 -0
  168. data/vendor/gems/net-ssh/test/transport/test_identity_cipher.rb +40 -0
  169. data/vendor/gems/net-ssh/test/transport/test_packet_stream.rb +435 -0
  170. data/vendor/gems/net-ssh/test/transport/test_server_version.rb +57 -0
  171. data/vendor/gems/net-ssh/test/transport/test_session.rb +315 -0
  172. data/vendor/gems/net-ssh/test/transport/test_state.rb +173 -0
  173. metadata +116 -4
  174. data/bin/install-poolparty +0 -20
@@ -0,0 +1,336 @@
1
+ require 'common'
2
+ require 'net/ssh/buffer'
3
+
4
+ class TestBuffer < Test::Unit::TestCase
5
+ def test_constructor_should_initialize_buffer_to_empty_by_default
6
+ buffer = new
7
+ assert buffer.empty?
8
+ assert_equal 0, buffer.position
9
+ end
10
+
11
+ def test_constructor_with_string_should_initialize_buffer_to_the_string
12
+ buffer = new("hello")
13
+ assert !buffer.empty?
14
+ assert_equal "hello", buffer.to_s
15
+ assert_equal 0, buffer.position
16
+ end
17
+
18
+ def test_from_should_require_an_even_number_of_arguments
19
+ assert_raises(ArgumentError) { Net::SSH::Buffer.from("this") }
20
+ end
21
+
22
+ def test_from_should_build_new_buffer_from_definition
23
+ buffer = Net::SSH::Buffer.from(:byte, 1, :long, 2, :int64, 3, :string, "4", :bool, true, :bool, false, :bignum, OpenSSL::BN.new("1234567890", 10), :raw, "something")
24
+ assert_equal "\1\0\0\0\2\0\0\0\0\0\0\0\3\0\0\0\0014\1\0\000\000\000\004I\226\002\322something", buffer.to_s
25
+ end
26
+
27
+ def test_from_with_array_argument_should_write_multiple_of_the_given_type
28
+ buffer = Net::SSH::Buffer.from(:byte, [1,2,3,4,5])
29
+ assert_equal "\1\2\3\4\5", buffer.to_s
30
+ end
31
+
32
+ def test_read_without_argument_should_read_to_end
33
+ buffer = new("hello world")
34
+ assert_equal "hello world", buffer.read
35
+ assert buffer.eof?
36
+ assert_equal 11, buffer.position
37
+ end
38
+
39
+ def test_read_with_argument_that_is_less_than_length_should_read_that_many_bytes
40
+ buffer = new "hello world"
41
+ assert_equal "hello", buffer.read(5)
42
+ assert_equal 5, buffer.position
43
+ end
44
+
45
+ def test_read_with_argument_that_is_more_than_length_should_read_no_more_than_length
46
+ buffer = new "hello world"
47
+ assert_equal "hello world", buffer.read(500)
48
+ assert_equal 11, buffer.position
49
+ end
50
+
51
+ def test_read_at_eof_should_return_empty_string
52
+ buffer = new "hello"
53
+ buffer.position = 5
54
+ assert_equal "", buffer.read
55
+ end
56
+
57
+ def test_consume_without_argument_should_resize_buffer_to_start_at_position
58
+ buffer = new "hello world"
59
+ buffer.read(5)
60
+ assert_equal 5, buffer.position
61
+ assert_equal 11, buffer.length
62
+ buffer.consume!
63
+ assert_equal 0, buffer.position
64
+ assert_equal 6, buffer.length
65
+ assert_equal " world", buffer.to_s
66
+ end
67
+
68
+ def test_consume_with_argument_should_resize_buffer_starting_at_n
69
+ buffer = new "hello world"
70
+ assert_equal 0, buffer.position
71
+ buffer.consume!(5)
72
+ assert_equal 0, buffer.position
73
+ assert_equal 6, buffer.length
74
+ assert_equal " world", buffer.to_s
75
+ end
76
+
77
+ def test_read_bang_should_read_and_consume_and_return_read_portion
78
+ buffer = new "hello world"
79
+ assert_equal "hello", buffer.read!(5)
80
+ assert_equal 0, buffer.position
81
+ assert_equal 6, buffer.length
82
+ assert_equal " world", buffer.to_s
83
+ end
84
+
85
+ def test_available_should_return_length_after_position_to_end_of_string
86
+ buffer = new "hello world"
87
+ buffer.read(5)
88
+ assert_equal 6, buffer.available
89
+ end
90
+
91
+ def test_clear_bang_should_reset_buffer_contents_and_counters
92
+ buffer = new "hello world"
93
+ buffer.read(5)
94
+ buffer.clear!
95
+ assert_equal 0, buffer.length
96
+ assert_equal 0, buffer.position
97
+ assert_equal "", buffer.to_s
98
+ end
99
+
100
+ def test_append_should_append_argument_without_changing_position_and_should_return_self
101
+ buffer = new "hello world"
102
+ buffer.read(5)
103
+ buffer.append(" again")
104
+ assert_equal 5, buffer.position
105
+ assert_equal 12, buffer.available
106
+ assert_equal 17, buffer.length
107
+ assert_equal "hello world again", buffer.to_s
108
+ end
109
+
110
+ def test_remainder_as_buffer_should_return_a_new_buffer_filled_with_the_text_after_the_current_position
111
+ buffer = new "hello world"
112
+ buffer.read(6)
113
+ b2 = buffer.remainder_as_buffer
114
+ assert_equal 6, buffer.position
115
+ assert_equal 0, b2.position
116
+ assert_equal "world", b2.to_s
117
+ end
118
+
119
+ def test_read_int64_should_return_8_byte_integer
120
+ buffer = new "\xff\xee\xdd\xcc\xbb\xaa\x99\x88"
121
+ assert_equal 0xffeeddccbbaa9988, buffer.read_int64
122
+ assert_equal 8, buffer.position
123
+ end
124
+
125
+ def test_read_int64_should_return_nil_on_partial_read
126
+ buffer = new "\0\0\0\0\0\0\0"
127
+ assert_nil buffer.read_int64
128
+ assert buffer.eof?
129
+ end
130
+
131
+ def test_read_long_should_return_4_byte_integer
132
+ buffer = new "\xff\xee\xdd\xcc\xbb\xaa\x99\x88"
133
+ assert_equal 0xffeeddcc, buffer.read_long
134
+ assert_equal 4, buffer.position
135
+ end
136
+
137
+ def test_read_long_should_return_nil_on_partial_read
138
+ buffer = new "\0\0\0"
139
+ assert_nil buffer.read_long
140
+ assert buffer.eof?
141
+ end
142
+
143
+ def test_read_byte_should_return_single_byte_integer
144
+ buffer = new "\xfe\xdc"
145
+ assert_equal 0xfe, buffer.read_byte
146
+ assert_equal 1, buffer.position
147
+ end
148
+
149
+ def test_read_byte_should_return_nil_at_eof
150
+ assert_nil new.read_byte
151
+ end
152
+
153
+ def test_read_string_should_read_length_and_data_from_buffer
154
+ buffer = new "\0\0\0\x0bhello world"
155
+ assert_equal "hello world", buffer.read_string
156
+ end
157
+
158
+ def test_read_string_should_return_nil_if_4_byte_length_cannot_be_read
159
+ assert_nil new("\0\1").read_string
160
+ end
161
+
162
+ def test_read_bool_should_return_true_if_non_zero_byte_is_read
163
+ buffer = new "\1\2\3\4\5\6"
164
+ 6.times { assert_equal true, buffer.read_bool }
165
+ end
166
+
167
+ def test_read_bool_should_return_false_if_zero_byte_is_read
168
+ buffer = new "\0"
169
+ assert_equal false, buffer.read_bool
170
+ end
171
+
172
+ def test_read_bool_should_return_nil_at_eof
173
+ assert_nil new.read_bool
174
+ end
175
+
176
+ def test_read_bignum_should_read_openssl_formatted_bignum
177
+ buffer = new("\000\000\000\004I\226\002\322")
178
+ assert_equal OpenSSL::BN.new("1234567890", 10), buffer.read_bignum
179
+ end
180
+
181
+ def test_read_bignum_should_return_nil_if_length_cannot_be_read
182
+ assert_nil new("\0\1\2").read_bignum
183
+ end
184
+
185
+ def test_read_key_blob_should_read_dsa_keys
186
+ random_dss { |buffer| buffer.read_keyblob("ssh-dss") }
187
+ end
188
+
189
+ def test_read_key_blob_should_read_rsa_keys
190
+ random_rsa { |buffer| buffer.read_keyblob("ssh-rsa") }
191
+ end
192
+
193
+ def test_read_key_should_read_dsa_key_type_and_keyblob
194
+ random_dss do |buffer|
195
+ b2 = Net::SSH::Buffer.from(:string, "ssh-dss", :raw, buffer)
196
+ b2.read_key
197
+ end
198
+ end
199
+
200
+ def test_read_key_should_read_rsa_key_type_and_keyblob
201
+ random_rsa do |buffer|
202
+ b2 = Net::SSH::Buffer.from(:string, "ssh-rsa", :raw, buffer)
203
+ b2.read_key
204
+ end
205
+ end
206
+
207
+ def test_read_buffer_should_read_a_string_and_return_it_wrapped_in_a_buffer
208
+ buffer = new("\0\0\0\x0bhello world")
209
+ b2 = buffer.read_buffer
210
+ assert_equal 0, b2.position
211
+ assert_equal 11, b2.length
212
+ assert_equal "hello world", b2.read
213
+ end
214
+
215
+ def test_read_to_should_return_nil_if_pattern_does_not_exist_in_buffer
216
+ buffer = new("one two three")
217
+ assert_nil buffer.read_to("\n")
218
+ end
219
+
220
+ def test_read_to_should_grok_string_patterns
221
+ buffer = new("one two three")
222
+ assert_equal "one tw", buffer.read_to("tw")
223
+ assert_equal 6, buffer.position
224
+ end
225
+
226
+ def test_read_to_should_grok_regex_patterns
227
+ buffer = new("one two three")
228
+ assert_equal "one tw", buffer.read_to(/tw/)
229
+ assert_equal 6, buffer.position
230
+ end
231
+
232
+ def test_read_to_should_grok_fixnum_patterns
233
+ buffer = new("one two three")
234
+ assert_equal "one tw", buffer.read_to(?w)
235
+ assert_equal 6, buffer.position
236
+ end
237
+
238
+ def test_reset_bang_should_reset_position_to_0
239
+ buffer = new("hello world")
240
+ buffer.read(5)
241
+ assert_equal 5, buffer.position
242
+ buffer.reset!
243
+ assert_equal 0, buffer.position
244
+ end
245
+
246
+ def test_write_should_write_arguments_directly_to_end_buffer
247
+ buffer = new("start")
248
+ buffer.write "hello", " ", "world"
249
+ assert_equal "starthello world", buffer.to_s
250
+ assert_equal 0, buffer.position
251
+ end
252
+
253
+ def test_write_int64_should_write_arguments_as_8_byte_integers_to_end_of_buffer
254
+ buffer = new("start")
255
+ buffer.write_int64 0xffeeddccbbaa9988, 0x7766554433221100
256
+ assert_equal "start\xff\xee\xdd\xcc\xbb\xaa\x99\x88\x77\x66\x55\x44\x33\x22\x11\x00", buffer.to_s
257
+ end
258
+
259
+ def test_write_long_should_write_arguments_as_4_byte_integers_to_end_of_buffer
260
+ buffer = new("start")
261
+ buffer.write_long 0xffeeddcc, 0xbbaa9988
262
+ assert_equal "start\xff\xee\xdd\xcc\xbb\xaa\x99\x88", buffer.to_s
263
+ end
264
+
265
+ def test_write_byte_should_write_arguments_as_1_byte_integers_to_end_of_buffer
266
+ buffer = new("start")
267
+ buffer.write_byte 1, 2, 3, 4, 5
268
+ assert_equal "start\1\2\3\4\5", buffer.to_s
269
+ end
270
+
271
+ def test_write_bool_should_write_arguments_as_1_byte_boolean_values_to_end_of_buffer
272
+ buffer = new("start")
273
+ buffer.write_bool nil, false, true, 1, Object.new
274
+ assert_equal "start\0\0\1\1\1", buffer.to_s
275
+ end
276
+
277
+ def test_write_bignum_should_write_arguments_as_ssh_formatted_bignum_values_to_end_of_buffer
278
+ buffer = new("start")
279
+ buffer.write_bignum OpenSSL::BN.new('1234567890', 10)
280
+ assert_equal "start\000\000\000\004I\226\002\322", buffer.to_s
281
+ end
282
+
283
+ def test_write_dss_key_should_write_argument_to_end_of_buffer
284
+ buffer = new("start")
285
+
286
+ key = OpenSSL::PKey::DSA.new
287
+ key.p = 0xffeeddccbbaa9988
288
+ key.q = 0x7766554433221100
289
+ key.g = 0xffddbb9977553311
290
+ key.pub_key = 0xeeccaa8866442200
291
+
292
+ buffer.write_key(key)
293
+ assert_equal "start\0\0\0\7ssh-dss\0\0\0\011\0\xff\xee\xdd\xcc\xbb\xaa\x99\x88\0\0\0\010\x77\x66\x55\x44\x33\x22\x11\x00\0\0\0\011\0\xff\xdd\xbb\x99\x77\x55\x33\x11\0\0\0\011\0\xee\xcc\xaa\x88\x66\x44\x22\x00", buffer.to_s
294
+ end
295
+
296
+ def test_write_rsa_key_should_write_argument_to_end_of_buffer
297
+ buffer = new("start")
298
+
299
+ key = OpenSSL::PKey::RSA.new
300
+ key.e = 0xffeeddccbbaa9988
301
+ key.n = 0x7766554433221100
302
+
303
+ buffer.write_key(key)
304
+ assert_equal "start\0\0\0\7ssh-rsa\0\0\0\011\0\xff\xee\xdd\xcc\xbb\xaa\x99\x88\0\0\0\010\x77\x66\x55\x44\x33\x22\x11\x00", buffer.to_s
305
+ end
306
+
307
+ private
308
+
309
+ def random_rsa
310
+ n1 = OpenSSL::BN.new(rand(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF).to_s, 10)
311
+ n2 = OpenSSL::BN.new(rand(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF).to_s, 10)
312
+ buffer = Net::SSH::Buffer.from(:bignum, [n1, n2])
313
+ key = yield(buffer)
314
+ assert_equal "ssh-rsa", key.ssh_type
315
+ assert_equal n1, key.e
316
+ assert_equal n2, key.n
317
+ end
318
+
319
+ def random_dss
320
+ n1 = OpenSSL::BN.new(rand(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF).to_s, 10)
321
+ n2 = OpenSSL::BN.new(rand(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF).to_s, 10)
322
+ n3 = OpenSSL::BN.new(rand(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF).to_s, 10)
323
+ n4 = OpenSSL::BN.new(rand(0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF).to_s, 10)
324
+ buffer = Net::SSH::Buffer.from(:bignum, [n1, n2, n3, n4])
325
+ key = yield(buffer)
326
+ assert_equal "ssh-dss", key.ssh_type
327
+ assert_equal n1, key.p
328
+ assert_equal n2, key.q
329
+ assert_equal n3, key.g
330
+ assert_equal n4, key.pub_key
331
+ end
332
+
333
+ def new(*args)
334
+ Net::SSH::Buffer.new(*args)
335
+ end
336
+ end
@@ -0,0 +1,63 @@
1
+ require 'common'
2
+ require 'net/ssh/buffered_io'
3
+
4
+ class TestBufferedIo < Test::Unit::TestCase
5
+ def test_fill_should_pull_from_underlying_io
6
+ io.expects(:recv).with(8192).returns("here is some data")
7
+ assert_equal 17, io.fill
8
+ assert_equal 17, io.available
9
+ assert_equal "here is some data", io.read_available(20)
10
+ end
11
+
12
+ def test_enqueue_should_not_write_to_underlying_io
13
+ assert !io.pending_write?
14
+ io.expects(:send).never
15
+ io.enqueue("here is some data")
16
+ assert io.pending_write?
17
+ end
18
+
19
+ def test_send_pending_should_not_fail_when_no_writes_are_pending
20
+ assert !io.pending_write?
21
+ io.expects(:send).never
22
+ assert_nothing_raised { io.send_pending }
23
+ end
24
+
25
+ def test_send_pending_with_pending_writes_should_write_to_underlying_io
26
+ io.enqueue("here is some data")
27
+ io.expects(:send).with("here is some data", 0).returns(17)
28
+ assert io.pending_write?
29
+ assert_nothing_raised { io.send_pending }
30
+ assert !io.pending_write?
31
+ end
32
+
33
+ def test_wait_for_pending_sends_should_write_only_once_if_all_can_be_written_at_once
34
+ io.enqueue("here is some data")
35
+ io.expects(:send).with("here is some data", 0).returns(17)
36
+ assert io.pending_write?
37
+ assert_nothing_raised { io.wait_for_pending_sends }
38
+ assert !io.pending_write?
39
+ end
40
+
41
+ def test_wait_for_pending_sends_should_write_multiple_times_if_first_write_was_partial
42
+ io.enqueue("here is some data")
43
+
44
+ io.expects(:send).with("here is some data", 0).returns(10)
45
+ io.expects(:send).with("me data", 0).returns(4)
46
+ io.expects(:send).with("ata", 0).returns(3)
47
+
48
+ IO.expects(:select).times(2).with(nil, [io]).returns([[], [io]])
49
+
50
+ assert_nothing_raised { io.wait_for_pending_sends }
51
+ assert !io.pending_write?
52
+ end
53
+
54
+ private
55
+
56
+ def io
57
+ @io ||= begin
58
+ io = mock("io")
59
+ io.extend(Net::SSH::BufferedIo)
60
+ io
61
+ end
62
+ end
63
+ end
@@ -0,0 +1,84 @@
1
+ require 'common'
2
+ require 'net/ssh/config'
3
+
4
+ class TestConfig < Test::Unit::TestCase
5
+ def test_load_for_non_existant_file_should_return_empty_hash
6
+ File.expects(:readable?).with("/bogus/file").returns(false)
7
+ assert_equal({}, Net::SSH::Config.load("/bogus/file", "host.name"))
8
+ end
9
+
10
+ def test_load_should_expand_path
11
+ expected = File.expand_path("~/.ssh/config")
12
+ File.expects(:readable?).with(expected).returns(false)
13
+ Net::SSH::Config.load("~/.ssh/config", "host.name")
14
+ end
15
+
16
+ def test_load_with_exact_host_match_should_load_that_section
17
+ config = Net::SSH::Config.load(config(:exact_match), "test.host")
18
+ assert config['compression']
19
+ assert config['forwardagent']
20
+ assert_equal 1234, config['port']
21
+ end
22
+
23
+ def test_load_with_wild_card_matches_should_load_all_matches_with_first_match_taking_precedence
24
+ config = Net::SSH::Config.load(config(:wild_cards), "test.host")
25
+ assert_equal 1234, config['port']
26
+ assert !config['compression']
27
+ assert config['forwardagent']
28
+ assert_equal %w(~/.ssh/id_dsa), config['identityfile']
29
+ assert !config.key?('rekeylimit')
30
+ end
31
+
32
+ def test_for_should_load_all_files_and_translate_to_net_ssh_options
33
+ config = Net::SSH::Config.for("test.host", [config(:exact_match), config(:wild_cards)])
34
+ assert_equal 1234, config[:port]
35
+ assert config[:compression]
36
+ assert config[:forward_agent]
37
+ assert_equal %w(~/.ssh/id_dsa), config[:keys]
38
+ assert !config.key?(:rekey_limit)
39
+ end
40
+
41
+ def test_load_should_parse_equal_sign_delimiters
42
+ config = Net::SSH::Config.load(config(:eqsign), "test.test")
43
+ assert config['compression']
44
+ assert_equal 1234, config['port']
45
+ end
46
+
47
+ def test_translate_should_correctly_translate_from_openssh_to_net_ssh_names
48
+ open_ssh = {
49
+ 'ciphers' => "a,b,c",
50
+ 'compression' => true,
51
+ 'compressionlevel' => 6,
52
+ 'connecttimeout' => 100,
53
+ 'forwardagent' => true,
54
+ 'hostbasedauthentication' => true,
55
+ 'hostkeyalgorithms' => "d,e,f",
56
+ 'identityfile' => %w(g h i),
57
+ 'macs' => "j,k,l",
58
+ 'passwordauthentication' => true,
59
+ 'port' => 1234,
60
+ 'pubkeyauthentication' => true,
61
+ 'rekeylimit' => 1024
62
+ }
63
+
64
+ net_ssh = Net::SSH::Config.translate(open_ssh)
65
+
66
+ assert_equal %w(a b c), net_ssh[:encryption]
67
+ assert_equal true, net_ssh[:compression]
68
+ assert_equal 6, net_ssh[:compression_level]
69
+ assert_equal 100, net_ssh[:timeout]
70
+ assert_equal true, net_ssh[:forward_agent]
71
+ assert_equal %w(hostbased password publickey), net_ssh[:auth_methods].sort
72
+ assert_equal %w(d e f), net_ssh[:host_key]
73
+ assert_equal %w(g h i), net_ssh[:keys]
74
+ assert_equal %w(j k l), net_ssh[:hmac]
75
+ assert_equal 1234, net_ssh[:port]
76
+ assert_equal 1024, net_ssh[:rekey_limit]
77
+ end
78
+
79
+ private
80
+
81
+ def config(name)
82
+ "test/configs/#{name}"
83
+ end
84
+ end
@@ -0,0 +1,67 @@
1
+ require 'common'
2
+ require 'net/ssh/key_factory'
3
+
4
+ class TestKeyFactory < Test::Unit::TestCase
5
+ def test_load_unencrypted_private_RSA_key_should_return_key
6
+ File.expects(:read).with("/key-file").returns(rsa_key.export)
7
+ assert_equal rsa_key.to_der, Net::SSH::KeyFactory.load_private_key("/key-file").to_der
8
+ end
9
+
10
+ def test_load_unencrypted_private_DSA_key_should_return_key
11
+ File.expects(:read).with("/key-file").returns(dsa_key.export)
12
+ assert_equal dsa_key.to_der, Net::SSH::KeyFactory.load_private_key("/key-file").to_der
13
+ end
14
+
15
+ def test_load_encrypted_private_RSA_key_should_prompt_for_password_and_return_key
16
+ File.expects(:read).with("/key-file").returns(encrypted(rsa_key, "password"))
17
+ Net::SSH::KeyFactory.expects(:prompt).with("Enter passphrase for /key-file:", false).returns("password")
18
+ assert_equal rsa_key.to_der, Net::SSH::KeyFactory.load_private_key("/key-file").to_der
19
+ end
20
+
21
+ def test_load_encrypted_private_RSA_key_with_password_should_not_prompt_and_return_key
22
+ File.expects(:read).with("/key-file").returns(encrypted(rsa_key, "password"))
23
+ assert_equal rsa_key.to_der, Net::SSH::KeyFactory.load_private_key("/key-file", "password").to_der
24
+ end
25
+
26
+ def test_load_encrypted_private_DSA_key_should_prompt_for_password_and_return_key
27
+ File.expects(:read).with("/key-file").returns(encrypted(dsa_key, "password"))
28
+ Net::SSH::KeyFactory.expects(:prompt).with("Enter passphrase for /key-file:", false).returns("password")
29
+ assert_equal dsa_key.to_der, Net::SSH::KeyFactory.load_private_key("/key-file").to_der
30
+ end
31
+
32
+ def test_load_encrypted_private_DSA_key_with_password_should_not_prompt_and_return_key
33
+ File.expects(:read).with("/key-file").returns(encrypted(dsa_key, "password"))
34
+ assert_equal dsa_key.to_der, Net::SSH::KeyFactory.load_private_key("/key-file", "password").to_der
35
+ end
36
+
37
+ def test_load_encrypted_private_key_should_give_three_tries_for_the_password_and_then_raise_exception
38
+ File.expects(:read).with("/key-file").returns(encrypted(rsa_key, "password"))
39
+ Net::SSH::KeyFactory.expects(:prompt).times(3).with("Enter passphrase for /key-file:", false).returns("passwod","passphrase","passwd")
40
+ assert_raises(OpenSSL::PKey::RSAError) { Net::SSH::KeyFactory.load_private_key("/key-file") }
41
+ end
42
+
43
+ def test_load_public_rsa_key_should_return_key
44
+ File.expects(:read).with("/key-file").returns(public(rsa_key))
45
+ assert_equal rsa_key.to_blob, Net::SSH::KeyFactory.load_public_key("/key-file").to_blob
46
+ end
47
+
48
+ private
49
+
50
+ def rsa_key
51
+ @rsa_key ||= OpenSSL::PKey::RSA.new("0@\002\001\000\002\t\000\300\030\317\2132\340 \267\002\003\001\000\001\002\t\000\236~\232\025\350Y=\341\002\005\000\352D\217\a\002\005\000\321\352\304\321\002\005\000\242\350\206%\002\005\000\270\021\217\361\002\004~\253\214j")
52
+ end
53
+
54
+ def dsa_key
55
+ @dsa_key ||= OpenSSL::PKey::DSA.new("0\201\367\002\001\000\002A\000\203\316/\037u\272&J\265\003l3\315d\324h\372{\t8\252#\331_\026\006\035\270\266\255\343\353Z\302\276\335\336\306\220\375\202L\244\244J\206>\346\b\315\211\302L\246x\247u\a\376\366\345\302\016#\002\025\000\244\274\302\221Og\275/\302+\356\346\360\024\373wI\2573\361\002@\027\215\270r*\f\213\350C\245\021:\350 \006\\\376\345\022`\210b\262\3643\023XLKS\320\370\002\276\347A\nU\204\276\324\256`=\026\240\330\306J\316V\213\024\e\030\215\355\006\037q\337\356ln\002@\017\257\034\f\260\333'S\271#\237\230E\321\312\027\021\226\331\251Vj\220\305\316\036\v\266+\000\230\270\177B\003?t\a\305]e\344\261\334\023\253\323\251\223M\2175)a(\004\"lI8\312\303\307\a\002\024_\aznW\345\343\203V\326\246ua\203\376\201o\350\302\002")
56
+ end
57
+
58
+ def encrypted(key, password)
59
+ key.export(OpenSSL::Cipher::Cipher.new("des-ede3-cbc"), password)
60
+ end
61
+
62
+ def public(key)
63
+ result = "#{key.ssh_type} "
64
+ result << [Net::SSH::Buffer.from(:key, key).to_s].pack("m*").strip.tr("\n\r\t ", "")
65
+ result << " joe@host.test"
66
+ end
67
+ end
@@ -0,0 +1,39 @@
1
+ require 'common'
2
+ require 'net/ssh/transport/hmac/md5'
3
+
4
+ module Transport; module HMAC
5
+
6
+ class TestMD5 < Test::Unit::TestCase
7
+ def test_expected_digest_class
8
+ assert_equal OpenSSL::Digest::MD5, subject.digest_class
9
+ assert_equal OpenSSL::Digest::MD5, subject.new.digest_class
10
+ end
11
+
12
+ def test_expected_key_length
13
+ assert_equal 16, subject.key_length
14
+ assert_equal 16, subject.new.key_length
15
+ end
16
+
17
+ def test_expected_mac_length
18
+ assert_equal 16, subject.mac_length
19
+ assert_equal 16, subject.new.mac_length
20
+ end
21
+
22
+ def test_expected_digest
23
+ hmac = subject.new("1234567890123456")
24
+ assert_equal "\275\345\006\307y~Oi\035<.\341\031\250<\257", hmac.digest("hello world")
25
+ end
26
+
27
+ def test_key_should_be_truncated_to_required_length
28
+ hmac = subject.new("12345678901234567890")
29
+ assert_equal "1234567890123456", hmac.key
30
+ end
31
+
32
+ private
33
+
34
+ def subject
35
+ Net::SSH::Transport::HMAC::MD5
36
+ end
37
+ end
38
+
39
+ end; end
@@ -0,0 +1,25 @@
1
+ require 'common'
2
+ require 'transport/hmac/test_md5'
3
+ require 'net/ssh/transport/hmac/md5_96'
4
+
5
+ module Transport; module HMAC
6
+
7
+ class TestMD5_96 < TestMD5
8
+ def test_expected_mac_length
9
+ assert_equal 12, subject.mac_length
10
+ assert_equal 12, subject.new.mac_length
11
+ end
12
+
13
+ def test_expected_digest
14
+ hmac = subject.new("1234567890123456")
15
+ assert_equal "\275\345\006\307y~Oi\035<.\341", hmac.digest("hello world")
16
+ end
17
+
18
+ private
19
+
20
+ def subject
21
+ Net::SSH::Transport::HMAC::MD5_96
22
+ end
23
+ end
24
+
25
+ end; end
@@ -0,0 +1,34 @@
1
+ require 'common'
2
+ require 'net/ssh/transport/hmac/none'
3
+
4
+ module Transport; module HMAC
5
+
6
+ class TestNone < Test::Unit::TestCase
7
+ def test_expected_digest_class
8
+ assert_equal nil, subject.digest_class
9
+ assert_equal nil, subject.new.digest_class
10
+ end
11
+
12
+ def test_expected_key_length
13
+ assert_equal 0, subject.key_length
14
+ assert_equal 0, subject.new.key_length
15
+ end
16
+
17
+ def test_expected_mac_length
18
+ assert_equal 0, subject.mac_length
19
+ assert_equal 0, subject.new.mac_length
20
+ end
21
+
22
+ def test_expected_digest
23
+ hmac = subject.new("1234567890123456")
24
+ assert_equal "", hmac.digest("hello world")
25
+ end
26
+
27
+ private
28
+
29
+ def subject
30
+ Net::SSH::Transport::HMAC::None
31
+ end
32
+ end
33
+
34
+ end; end
@@ -0,0 +1,34 @@
1
+ require 'common'
2
+ require 'net/ssh/transport/hmac/sha1'
3
+
4
+ module Transport; module HMAC
5
+
6
+ class TestSHA1 < Test::Unit::TestCase
7
+ def test_expected_digest_class
8
+ assert_equal OpenSSL::Digest::SHA1, subject.digest_class
9
+ assert_equal OpenSSL::Digest::SHA1, subject.new.digest_class
10
+ end
11
+
12
+ def test_expected_key_length
13
+ assert_equal 20, subject.key_length
14
+ assert_equal 20, subject.new.key_length
15
+ end
16
+
17
+ def test_expected_mac_length
18
+ assert_equal 20, subject.mac_length
19
+ assert_equal 20, subject.new.mac_length
20
+ end
21
+
22
+ def test_expected_digest
23
+ hmac = subject.new("1234567890123456")
24
+ assert_equal "\000\004W\202\204+&\335\311\251P\266\250\214\276\206;\022U\365", hmac.digest("hello world")
25
+ end
26
+
27
+ private
28
+
29
+ def subject
30
+ Net::SSH::Transport::HMAC::SHA1
31
+ end
32
+ end
33
+
34
+ end; end