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
|
#--
# =============================================================================
# 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/remote-network-handler'
require 'test/unit'
require 'socket'
class TC_RemoteNetworkHandler < Test::Unit::TestCase
class Log
def debug?
true
end
def debug( msg )
end
def error( msg )
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
HOST = "127.0.0.1"
PORT = 12248
class ScriptedServer
attr_reader :script
def initialize
@script = []
@thread = Thread.new {
server = TCPServer.new( HOST, PORT )
client = server.accept
until @script.empty?
client.send @script.shift, 0
sleep 0.1
end
server.shutdown rescue nil
}
end
def join
@thread.join
end
end
def setup
@channel = Channel.new
@handler = Net::SSH::Service::Forward::RemoteNetworkHandler.new(
Log.new, 16, PORT, HOST )
end
def test_on_eof
assert !@channel[:eof]
@handler.on_eof( @channel )
assert @channel[:eof]
end
def test_on_open
server = ScriptedServer.new
server.script << "1234567890123456"
server.script << "abcdefghijklmnop"
sleep 0.1
@handler.on_open( @channel, nil, nil, nil, nil )
server.join
@handler.on_eof( @channel )
sleep 0.1
assert_equal [ "1234567890123456", "abcdefghijklmnop", :close ],
@channel.events
end
end
|
