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
|
#--
# =============================================================================
# Copyright (c) 2004,2005 Jamis Buck (jamis@37signals.com)
# All rights reserved.
#
# This source file is distributed as part of the Net::SSH Secure Shell Client
# library for Ruby. This file (and the library as a whole) may be used only as
# allowed by either the BSD license, or the Ruby license (or, by association
# with the Ruby license, the GPL). See the "doc" subdirectory of the Net::SSH
# distribution for the texts of these licenses.
# -----------------------------------------------------------------------------
# net-ssh website : http://net-ssh.rubyforge.org
# project website: http://rubyforge.org/projects/net-ssh
# =============================================================================
#++
$:.unshift "#{File.dirname(__FILE__)}/../../../lib"
require 'net/ssh/service/forward/local-network-handler'
require 'test/unit'
class TC_LocalNetworkHandler < Test::Unit::TestCase
class Log
def debug?
true
end
def debug( msg )
end
def error( msg )
end
end
class Client
attr_reader :events
attr_reader :script
def initialize
@events = []
@script = []
end
def send( data, code )
@events << [ :send, data, code ]
end
def recv( block_size )
@events << [ :recv, block_size ]
@script.shift
end
end
class Channel
attr_reader :events
def initialize
@values = Hash.new
@events = []
end
def []( name )
@values[name]
end
def []=( name, value )
@values[name] = value
end
def close
@events << :close
end
def send_data( data )
@events << data
end
end
def setup
@client = Client.new
@channel = Channel.new
@handler = Net::SSH::Service::Forward::LocalNetworkHandler.new(
Log.new, 16, @client )
end
def test_on_receive
@handler.on_receive( @channel, "hello" )
assert_equal [ [ :send, "hello", 0 ] ], @client.events
end
def test_on_eof
assert !@channel[:eof]
@handler.on_eof( @channel )
assert @channel[:eof]
end
def test_process_no_eof
@client.script << "part 1"
@client.script << "part 2"
@client.script << "part 3"
@handler.process @channel
assert_equal [ "part 1", "part 2", "part 3", :close ], @channel.events
assert_equal [ [ :recv, 16 ], [ :recv, 16 ], [ :recv, 16 ], [ :recv, 16 ] ],
@client.events
end
def test_process_eof
@channel[:eof] = true
@handler.process @channel
assert_equal [ :close ], @channel.events
assert_equal [], @client.events
end
def test_process_error
assert_nothing_raised do
@handler.process nil
end
end
end
|
