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
124
125
126
127
128
|
require 'test/unit'
require 'soap/rpc/driver'
require 'soap/rpc/standaloneServer'
require 'soap/header/simplehandler'
require 'logger'
require 'webrick'
require 'rbconfig'
module SOAP
module Header
class TestAuthHeaderCGI < Test::Unit::TestCase
# This test shuld be run after installing ruby.
RUBYBIN = File.join(
Config::CONFIG["bindir"],
Config::CONFIG["ruby_install_name"] + Config::CONFIG["EXEEXT"]
)
RUBYBIN << " -d" if $DEBUG
Port = 17171
PortName = 'http://tempuri.org/authHeaderPort'
SupportPortName = 'http://tempuri.org/authHeaderSupportPort'
MyHeaderName = XSD::QName.new("http://tempuri.org/authHeader", "auth")
class ClientAuthHeaderHandler < SOAP::Header::SimpleHandler
def initialize(userid, passwd)
super(MyHeaderName)
@sessionid = nil
@userid = userid
@passwd = passwd
end
def on_simple_outbound
if @sessionid
{ "sessionid" => @sessionid }
else
{ "userid" => @userid, "passwd" => @passwd }
end
end
def on_simple_inbound(my_header, mustunderstand)
@sessionid = my_header["sessionid"]
end
def sessionid
@sessionid
end
end
def setup
@endpoint = "http://localhost:#{Port}/"
setup_server
setup_client
end
def setup_server
@endpoint = "http://localhost:#{Port}/server.cgi"
logger = Logger.new(STDERR)
logger.level = Logger::Severity::ERROR
@server = WEBrick::HTTPServer.new(
:BindAddress => "0.0.0.0",
:Logger => logger,
:Port => Port,
:AccessLog => [],
:DocumentRoot => File.dirname(File.expand_path(__FILE__)),
:CGIPathEnv => ENV['PATH'],
:CGIInterpreter => RUBYBIN
)
@t = Thread.new {
Thread.current.abort_on_exception = true
@server.start
}
while @server.status != :Running
sleep 0.1
unless @t.alive?
@t.join
raise
end
end
end
def setup_client
@client = SOAP::RPC::Driver.new(@endpoint, PortName)
@client.wiredump_dev = STDERR if $DEBUG
@client.add_method('deposit', 'amt')
@client.add_method('withdrawal', 'amt')
@supportclient = SOAP::RPC::Driver.new(@endpoint, SupportPortName)
@supportclient.add_method('delete_sessiondb')
end
def teardown
@supportclient.delete_sessiondb
teardown_server
teardown_client
end
def teardown_server
@server.shutdown
@t.kill
@t.join
end
def teardown_client
@client.reset_stream
@supportclient.reset_stream
end
def test_success
h = ClientAuthHeaderHandler.new('NaHi', 'passwd')
@client.headerhandler << h
assert_equal("deposit 150 OK", @client.deposit(150))
assert_equal("withdrawal 120 OK", @client.withdrawal(120))
end
def test_authfailure
h = ClientAuthHeaderHandler.new('NaHi', 'pa')
@client.headerhandler << h
assert_raises(RuntimeError) do
@client.deposit(150)
end
end
end
end
end
|
