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
|
require 'test/unit'
require 'mocha'
require 'net/ssh/gateway'
class GatewayTest < Test::Unit::TestCase
def teardown
Thread.list { |t| t.kill unless Thread.current == t }
end
def test_shutdown_without_any_open_connections_should_terminate_session
session, gateway = new_gateway
session.expects(:close)
gateway.shutdown!
assert !gateway.active?
assert session.forward.active_locals.empty?
end
def test_open_should_start_local_ports_at_65535
gateway_session, gateway = new_gateway
assert_equal 65535, gateway.open("app1", 22)
assert_equal [65535, "app1", 22], gateway_session.forward.active_locals[65535]
end
def test_open_should_decrement_port_and_retry_if_ports_are_in_use
gateway_session, gateway = new_gateway(:reserved => lambda { |n| n > 65000 })
assert_equal 65000, gateway.open("app1", 22)
assert_equal [65000, "app1", 22], gateway_session.forward.active_locals[65000]
end
def test_ssh_should_return_connection_when_no_block_is_given
gateway_session, gateway = new_gateway
expect_connect_to("127.0.0.1", "user", :port => 65535).returns(result = mock("session"))
newsess = gateway.ssh("app1", "user")
assert_equal result, newsess
assert_equal [65535, "app1", 22], gateway_session.forward.active_locals[65535]
end
def test_ssh_with_block_should_yield_session_and_then_close_port
gateway_session, gateway = new_gateway
expect_connect_to("127.0.0.1", "user", :port => 65535).yields(result = mock("session"))
yielded = false
gateway.ssh("app1", "user") do |newsess|
yielded = true
assert_equal result, newsess
end
assert yielded
assert gateway_session.forward.active_locals.empty?
end
def test_shutdown_should_cancel_active_forwarded_ports
gateway_session, gateway = new_gateway
gateway.open("app1", 80)
assert !gateway_session.forward.active_locals.empty?
gateway.shutdown!
assert gateway_session.forward.active_locals.empty?
end
private
def expect_connect_to(host, user, options={})
Net::SSH.expects(:start).with do |real_host, real_user, real_options|
host == real_host &&
user == real_user &&
options[:port] == real_options[:port]
end
end
def new_gateway(options={})
session = MockSession.new(options)
expect_connect_to("test.host", "tester").returns(session)
[session, Net::SSH::Gateway.new("test.host", "tester")]
end
class MockForward
attr_reader :active_locals
def initialize(options)
@options = options
@active_locals = {}
end
def cancel_local(port)
@active_locals.delete(port)
end
def local(lport, host, rport)
raise Errno::EADDRINUSE if @options[:reserved] && @options[:reserved][lport]
@active_locals[lport] = [lport, host, rport]
end
end
class MockSession
attr_reader :forward
def initialize(options={})
@forward = MockForward.new(options)
end
def close
end
def process(wait=nil)
true
end
def looping?
@looping
end
def loop
@looping = true
sleep 0.1 while yield
@looping = false
end
end
end
|
