зеркало из https://github.com/github/ruby.git
135 строки
3.3 KiB
Ruby
135 строки
3.3 KiB
Ruby
# coding: utf-8
|
|
|
|
require 'test/unit'
|
|
require 'webrick'
|
|
require_relative 'webrick_testing'
|
|
require "xmlrpc/server"
|
|
require 'xmlrpc/client'
|
|
require 'logger'
|
|
|
|
class Test_Webrick < Test::Unit::TestCase
|
|
include WEBrick_Testing
|
|
|
|
@@basic_auth = WEBrick::HTTPAuth::BasicAuth.new(
|
|
:Realm => 'auth',
|
|
:UserDB => WEBrick::HTTPAuth::Htpasswd.new(File.expand_path('./htpasswd', File.dirname(__FILE__))),
|
|
:Logger => Logger.new(File::NULL),
|
|
)
|
|
|
|
def create_servlet
|
|
s = XMLRPC::WEBrickServlet.new
|
|
|
|
def s.service(req, res)
|
|
@@basic_auth.authenticate(req, res)
|
|
super(req, res)
|
|
end
|
|
|
|
s.add_handler("test.add") do |a,b|
|
|
a + b
|
|
end
|
|
|
|
s.add_handler("test.div") do |a,b|
|
|
if b == 0
|
|
raise XMLRPC::FaultException.new(1, "division by zero")
|
|
else
|
|
a / b
|
|
end
|
|
end
|
|
|
|
s.set_default_handler do |name, *args|
|
|
raise XMLRPC::FaultException.new(-99, "Method #{name} missing" +
|
|
" or wrong number of parameters!")
|
|
end
|
|
|
|
s.add_introspection
|
|
|
|
return s
|
|
end
|
|
|
|
def setup_http_server(port, use_ssl)
|
|
option = {
|
|
:BindAddress => "localhost",
|
|
:Port => port,
|
|
:SSLEnable => use_ssl,
|
|
}
|
|
if use_ssl
|
|
require 'webrick/https'
|
|
option.update(
|
|
:SSLVerifyClient => ::OpenSSL::SSL::VERIFY_NONE,
|
|
:SSLCertName => []
|
|
)
|
|
end
|
|
|
|
start_server(option) {|w| w.mount('/RPC2', create_servlet) }
|
|
end
|
|
|
|
PORT = 8071
|
|
def test_client_server
|
|
# NOTE: I don't enable SSL testing as this hangs
|
|
[false].each do |use_ssl|
|
|
begin
|
|
setup_http_server(PORT, use_ssl)
|
|
@s = XMLRPC::Client.new3(:port => PORT, :use_ssl => use_ssl)
|
|
@s.user = 'admin'
|
|
@s.password = 'admin'
|
|
silent do
|
|
do_test
|
|
end
|
|
@s = XMLRPC::Client.new3(:port => PORT, :use_ssl => use_ssl)
|
|
@s.user = '01234567890123456789012345678901234567890123456789012345678901234567890123456789'
|
|
@s.password = 'guest'
|
|
silent do
|
|
do_test
|
|
end
|
|
ensure
|
|
stop_server
|
|
end
|
|
end
|
|
end
|
|
|
|
def silent
|
|
begin
|
|
back, $VERBOSE = $VERBOSE, nil
|
|
yield
|
|
ensure
|
|
$VERBOSE = back
|
|
end
|
|
end
|
|
|
|
def do_test
|
|
# simple call
|
|
assert_equal 9, @s.call('test.add', 4, 5)
|
|
|
|
# fault exception
|
|
assert_raise(XMLRPC::FaultException) { @s.call('test.div', 1, 0) }
|
|
|
|
# fault exception via call2
|
|
ok, param = @s.call2('test.div', 1, 0)
|
|
assert_equal false, ok
|
|
assert_instance_of XMLRPC::FaultException, param
|
|
assert_equal 1, param.faultCode
|
|
assert_equal 'division by zero', param.faultString
|
|
|
|
# call2 without fault exception
|
|
ok, param = @s.call2('test.div', 10, 5)
|
|
assert_equal true, ok
|
|
assert_equal param, 2
|
|
|
|
# introspection
|
|
assert_equal ["test.add", "test.div", "system.listMethods", "system.methodSignature", "system.methodHelp"], @s.call("system.listMethods")
|
|
|
|
# default handler (missing handler)
|
|
ok, param = @s.call2('test.nonexisting')
|
|
assert_equal false, ok
|
|
assert_equal(-99, param.faultCode)
|
|
|
|
# default handler (wrong number of arguments)
|
|
ok, param = @s.call2('test.add', 1, 2, 3)
|
|
assert_equal false, ok
|
|
assert_equal(-99, param.faultCode)
|
|
|
|
# multibyte characters
|
|
assert_equal "あいうえおかきくけこ", @s.call('test.add', "あいうえお", "かきくけこ")
|
|
end
|
|
end
|