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
|
require_relative '../common'
require 'net/ssh'
module NetSSH
class TestConnection < NetSSHTest
attr_reader :connection_session
def setup
authentication_session = mock('authentication_session')
authentication_session.stubs(:authenticate).returns(true)
Net::SSH::Authentication::Session.stubs(:new).returns(authentication_session)
@transport_session = mock('transport_session')
Net::SSH::Transport::Session.stubs(:new).returns(@transport_session)
@connection_session = mock('connection_session')
Net::SSH::Connection::Session.expects(new: connection_session)
end
def test_close_connection_on_exception
@connection_session.expects(:closed?).returns(false)
@connection_session.expects(:close).once
begin
Net::SSH.start('localhost', 'testuser') { raise "error" }
rescue RuntimeError
# We aren't interested in the exception
end
end
def test_close_connection_on_exception_only_if_still_open
conn_open = states('conn').starts_as(true)
@connection_session.expects(:close).then(conn_open.is(false)).once
@connection_session.expects(:closed?).when(conn_open.is(false)).returns(true)
begin
Net::SSH.start('localhost', 'testuser') do |ssh|
ssh.close
raise "error"
end
rescue RuntimeError
# We aren't interested in the exception
end
end
def test_return_value_is_returned
@connection_session.expects(:closed?).returns(false)
@connection_session.expects(:close).once
val = 1
retval = Net::SSH.start('localhost', 'testuser') { val }
assert_equal(val, retval)
end
end
end
|
