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 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204
|
# DO NOT REMOVE THIS COMMENT - PART OF TESTMODEL.
# Copyright (c)2006 Ross Bamford. See LICENSE.
$CURB_TESTING = true
require 'uri'
$TOPDIR = File.expand_path(File.join(File.dirname(__FILE__), '..'))
#$EXTDIR = File.join($TOPDIR, 'ext')
$LIBDIR = File.join($TOPDIR, 'lib')
$:.unshift($LIBDIR)
#$:.unshift($EXTDIR)
require 'curb'
begin
require 'test/unit'
rescue LoadError
gem 'test/unit'
require 'test/unit'
end
require 'fileutils'
$TEST_URL = "file://#{URI.escape(File.expand_path(__FILE__).tr('\\','/').tr(':','|'))}"
require 'thread'
require 'webrick'
# set this to true to avoid testing with multiple threads
# or to test with multiple threads set it to false
# this is important since, some code paths will change depending
# on the presence of multiple threads
TEST_SINGLE_THREADED=false
# keep webrick quiet
class ::WEBrick::HTTPServer
def access_log(config, req, res)
# nop
end
end
class ::WEBrick::BasicLog
def log(level, data)
# nop
end
end
#
# Simple test server to record number of times a request is sent/recieved of a specific
# request type, e.g. GET,POST,PUT,DELETE
#
class TestServlet < WEBrick::HTTPServlet::AbstractServlet
def self.port=(p)
@port = p
end
def self.port
(@port or 9129)
end
def self.path
'/methods'
end
def self.url
"http://127.0.0.1:#{port}#{path}"
end
def respond_with(method,req,res)
res.body = method.to_s
$auth_header = req['Authorization']
res['Content-Type'] = "text/plain"
end
def do_GET(req,res)
if req.path.match /redirect$/
res.status = 302
res['Location'] = '/foo'
elsif req.path.match /not_here$/
res.status = 404
elsif req.path.match /error$/
res.status = 500
end
respond_with("GET#{req.query_string}",req,res)
end
def do_HEAD(req,res)
res['Location'] = "/nonexistent"
respond_with("HEAD#{req.query_string}",req,res)
end
def do_POST(req,res)
if req.query['filename'].nil?
if req.body
params = {}
req.body.split('&').map{|s| k,v=s.split('='); params[k] = v }
end
if params and params['s'] == '500'
res.status = 500
else
respond_with("POST\n#{req.body}",req,res)
end
else
respond_with(req.query['filename'],req,res)
end
end
def do_PUT(req,res)
res['X-Requested-Content-Type'] = req.content_type
respond_with("PUT\n#{req.body}",req,res)
end
def do_DELETE(req,res)
respond_with("DELETE#{req.query_string}",req,res)
end
def do_PURGE(req,res)
respond_with("PURGE#{req.query_string}",req,res)
end
def do_COPY(req,res)
respond_with("COPY#{req.query_string}",req,res)
end
def do_PATCH(req,res)
respond_with("PATCH\n#{req.body}",req,res)
end
def do_OPTIONS(req,res)
respond_with("OPTIONS#{req.query_string}",req,res)
end
end
module TestServerMethods
def locked_file
File.join(File.dirname(__FILE__),"server_lock-#{@__port}")
end
def server_setup(port=9129,servlet=TestServlet)
@__port = port
if @server.nil? and !File.exist?(locked_file)
File.open(locked_file,'w') {|f| f << 'locked' }
if TEST_SINGLE_THREADED
rd, wr = IO.pipe
@__pid = fork do
rd.close
rd = nil
# start up a webrick server for testing delete
server = WEBrick::HTTPServer.new :Port => port, :DocumentRoot => File.expand_path(File.dirname(__FILE__))
server.mount(servlet.path, servlet)
server.mount("/ext", WEBrick::HTTPServlet::FileHandler, File.join(File.dirname(__FILE__),'..','ext'))
trap("INT") { server.shutdown }
GC.start
wr.flush
wr.close
server.start
end
wr.close
rd.read
rd.close
else
# start up a webrick server for testing delete
@server = WEBrick::HTTPServer.new :Port => port, :DocumentRoot => File.expand_path(File.dirname(__FILE__))
@server.mount(servlet.path, servlet)
@server.mount("/ext", WEBrick::HTTPServlet::FileHandler, File.join(File.dirname(__FILE__),'..','ext'))
queue = Queue.new # synchronize the thread startup to the main thread
@test_thread = Thread.new { queue << 1; @server.start }
# wait for the queue
value = queue.pop
if !value
STDERR.puts "Failed to startup test server!"
exit(1)
end
end
exit_code = lambda do
begin
if File.exist?(locked_file)
File.unlink locked_file
if TEST_SINGLE_THREADED
Process.kill 'INT', @__pid
else
@server.shutdown unless @server.nil?
end
end
#@server.shutdown unless @server.nil?
rescue Object => e
puts "Error #{__FILE__}:#{__LINE__}\n#{e.message}"
end
end
trap("INT"){exit_code.call}
at_exit{exit_code.call}
end
rescue Errno::EADDRINUSE
end
end
|