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
|
require 'webrick'
require 'webrick/httpservlet/abstract'
module NetHTTPSpecs
class NullWriter
def <<(s) end
def puts(*args) end
def print(*args) end
def printf(*args) end
end
class SpecServlet < WEBrick::HTTPServlet::AbstractServlet
def handle(req, res)
reply(req, res)
end
%w{ do_GET do_HEAD do_POST do_PUT do_PROPPATCH do_LOCK do_UNLOCK
do_OPTIONS do_PROPFIND do_DELETE do_MOVE do_COPY
do_MKCOL do_TRACE }.each do |method|
alias_method method.to_sym, :handle
end
end
class RequestServlet < SpecServlet
def reply(req, res)
res.content_type = "text/plain"
res.body = "Request type: #{req.request_method}"
end
end
class RequestBodyServlet < SpecServlet
def reply(req, res)
res.content_type = "text/plain"
res.body = req.body
end
end
class RequestHeaderServlet < SpecServlet
def reply(req, res)
res.content_type = "text/plain"
res.body = req.header.inspect
end
end
class << self
@server = nil
@server_thread = nil
def port
@server ? @server.config[:Port] : 3333
end
def start_server
server_config = {
BindAddress: "localhost",
Port: 0,
Logger: WEBrick::Log.new(NullWriter.new),
AccessLog: [],
ServerType: Thread
}
@server = WEBrick::HTTPServer.new(server_config)
@server.mount_proc('/') do |req, res|
res.content_type = "text/plain"
res.body = "This is the index page."
end
@server.mount('/request', RequestServlet)
@server.mount("/request/body", RequestBodyServlet)
@server.mount("/request/header", RequestHeaderServlet)
@server_thread = @server.start
end
def stop_server
if @server
begin
@server.shutdown
rescue Errno::EPIPE
# Because WEBrick is not thread-safe and only catches IOError
end
end
if @server_thread
@server_thread.join
end
timeout = WEBrick::Utils::TimeoutHandler
timeout.terminate if timeout.respond_to?(:terminate)
end
end
end
|