luasocket/test/httptest.lua

467 lines
11 KiB
Lua
Raw Normal View History

-- needs Alias from /home/c/diego/tec/luasocket/test to
-- "/luasocket-test" and "/luasocket-test/"
-- needs ScriptAlias from /home/c/diego/tec/luasocket/test/cgi
-- to "/luasocket-test-cgi" and "/luasocket-test-cgi/"
-- needs "AllowOverride AuthConfig" on /home/c/diego/tec/luasocket/test/auth
dofile("noglobals.lua")
local host, proxy, request, response, index_file
local ignore, expect, index, prefix, cgiprefix, index_crlf
socket.http.TIMEOUT = 5
local t = socket.time()
host = host or "diego.student.dyn.cs.princeton.edu"
proxy = proxy or "http://localhost:3128"
prefix = prefix or "/luasocket-test"
cgiprefix = cgiprefix or "/luasocket-test-cgi"
index_file = "test/index.html"
local readfile = function(name)
local f = io.open(name, "r")
if not f then return nil end
local s = f:read("*a")
f:close()
return s
end
-- read index with CRLF convention
index = readfile(index_file)
2001-09-25 23:34:43 +02:00
local similar = function(s1, s2)
return string.lower(string.gsub(s1 or "", "%s", "")) ==
string.lower(string.gsub(s2 or "", "%s", ""))
2001-09-25 23:34:43 +02:00
end
2001-01-25 22:59:39 +01:00
2001-09-25 23:34:43 +02:00
local fail = function(s)
s = s or "failed!"
print(s)
os.exit()
2001-01-25 22:59:39 +01:00
end
2001-09-25 23:34:43 +02:00
local check = function (v, e)
if v then print("ok")
2003-05-25 03:54:13 +02:00
else fail(e) end
2001-09-25 23:34:43 +02:00
end
local check_result = function(response, expect, ignore)
2001-09-25 23:34:43 +02:00
for i,v in response do
if not ignore[i] then
if v ~= expect[i] then
v = string.sub(type(v) == "string" and v or "", 1, 70)
print(v)
fail(i .. " differs!")
end
2001-09-25 23:34:43 +02:00
end
end
for i,v in expect do
if not ignore[i] then
if v ~= response[i] then
v = string.sub(type(v) == "string" and v or "", 1, 70)
print(v)
fail(i .. " differs!")
end
2001-09-25 23:34:43 +02:00
end
end
print("ok")
end
local check_request = function(request, expect, ignore)
local response = socket.http.request(request)
check_result(response, expect, ignore)
end
local check_request_cb = function(request, response, expect, ignore)
local response = socket.http.request_cb(request, response)
check_result(response, expect, ignore)
end
------------------------------------------------------------------------
io.write("testing request uri correctness: ")
2001-09-25 23:34:43 +02:00
local forth = cgiprefix .. "/request-uri?" .. "this+is+the+query+string"
local back, h, c, e = socket.http.get("http://" .. host .. forth)
if not back then fail(e) end
back = socket.url.parse(back)
if similar(back.query, "this+is+the+query+string") then print("ok")
else fail() end
2001-01-25 22:59:39 +01:00
------------------------------------------------------------------------
io.write("testing query string correctness: ")
2001-09-25 23:34:43 +02:00
forth = "this+is+the+query+string"
back = socket.http.get("http://" .. host .. cgiprefix ..
"/query-string?" .. forth)
2001-09-25 23:34:43 +02:00
if similar(back, forth) then print("ok")
else fail("failed!") end
2001-01-25 22:59:39 +01:00
------------------------------------------------------------------------
io.write("testing document retrieval: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://" .. host .. prefix .. "/index.html"
2001-09-25 23:34:43 +02:00
}
expect = {
body = index,
code = 200
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing redirect loop: ")
request = {
url = "http://" .. host .. cgiprefix .. "/redirect-loop"
}
expect = {
code = 302
}
ignore = {
status = 1,
headers = 1,
body = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing post method: ")
-- wanted to test chunked post, but apache doesn't support it...
request = {
url = "http://" .. host .. cgiprefix .. "/cat",
method = "POST",
body = index,
-- remove content-length header to send chunked body
headers = { ["content-length"] = string.len(index) }
}
expect = {
body = index,
code = 200
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing proxy with post method: ")
request = {
url = "http://" .. host .. cgiprefix .. "/cat",
method = "POST",
body = index,
headers = { ["content-length"] = string.len(index) },
proxy= proxy
}
expect = {
body = index,
code = 200
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing simple post function: ")
back = socket.http.post("http://" .. host .. cgiprefix .. "/cat", index)
check(back == index)
------------------------------------------------------------------------
io.write("testing send.file and receive.file callbacks: ")
request = {
url = "http://" .. host .. cgiprefix .. "/cat",
method = "POST",
body_cb = socket.callback.send.file(io.open(index_file, "r")),
headers = { ["content-length"] = string.len(index) }
}
response = {
body_cb = socket.callback.receive.file(io.open(index_file .. "-back", "w"))
}
expect = {
code = 200
}
ignore = {
body_cb = 1,
status = 1,
headers = 1
}
check_request_cb(request, response, expect, ignore)
back = readfile(index_file .. "-back")
check(back == index)
os.remove(index_file .. "-back")
------------------------------------------------------------------------
io.write("testing send.chain and receive.chain callbacks: ")
local function b64length(len)
local a = math.ceil(len/3)*4
local l = math.ceil(a/76)
return a + l*2
end
local req_cb = socket.callback.send.chain(
socket.callback.send.file(io.open(index_file, "r")),
socket.mime.chain(
socket.mime.encode("base64"),
socket.mime.wrap("base64")
)
)
local resp_cb = socket.callback.receive.chain(
socket.mime.decode("base64"),
socket.callback.receive.file(io.open(index_file .. "-back", "w"))
)
request = {
url = "http://" .. host .. cgiprefix .. "/cat",
method = "POST",
body_cb = req_cb,
headers = { ["content-length"] = b64length(string.len(index)) }
}
response = { body_cb = resp_cb }
expect = {
code = 200
}
ignore = {
body_cb = 1,
status = 1,
headers = 1
}
check_request_cb(request, response, expect, ignore)
back = readfile(index_file .. "-back")
check(back == index)
os.remove(index_file .. "-back")
------------------------------------------------------------------------
io.write("testing simple post function with table args: ")
back = socket.http.post {
url = "http://" .. host .. cgiprefix .. "/cat",
body = index
}
check(back == index)
------------------------------------------------------------------------
io.write("testing http redirection: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://" .. host .. prefix
2001-09-25 23:34:43 +02:00
}
expect = {
body = index,
code = 200
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing proxy with redirection: ")
request = {
url = "http://" .. host .. prefix,
proxy = proxy
}
expect = {
body = index,
code = 200
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing automatic auth failure: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://really:wrong@" .. host .. prefix .. "/auth/index.html"
2001-09-25 23:34:43 +02:00
}
expect = {
code = 401
}
ignore = {
body = 1,
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing http redirection failure: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://" .. host .. prefix,
redirect = false
2001-09-25 23:34:43 +02:00
}
expect = {
code = 301
}
ignore = {
body = 1,
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing host not found: ")
2001-09-25 23:34:43 +02:00
request = {
url = "http://wronghost/does/not/exist"
}
local c, e = socket.connect("wronghost", 80)
2001-09-25 23:34:43 +02:00
expect = {
error = e
}
ignore = {}
check_request(request, expect, ignore)
2001-01-25 22:59:39 +01:00
------------------------------------------------------------------------
io.write("testing invalid url: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = host .. prefix
2001-09-25 23:34:43 +02:00
}
local c, e = socket.connect("", 80)
2001-09-25 23:34:43 +02:00
expect = {
error = e
}
ignore = {}
check_request(request, expect, ignore)
2001-01-25 22:59:39 +01:00
------------------------------------------------------------------------
io.write("testing document not found: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://" .. host .. "/wrongdocument.html"
2001-09-25 23:34:43 +02:00
}
expect = {
code = 404
}
ignore = {
body = 1,
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing auth failure: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://" .. host .. prefix .. "/auth/index.html"
2001-09-25 23:34:43 +02:00
}
expect = {
code = 401
}
ignore = {
body = 1,
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing manual basic auth: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://" .. host .. prefix .. "/auth/index.html",
2001-09-25 23:34:43 +02:00
headers = {
authorization = "Basic " .. (socket.mime.b64("luasocket:password"))
2001-09-25 23:34:43 +02:00
}
}
expect = {
code = 200,
body = index
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing automatic basic auth: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://luasocket:password@" .. host .. prefix .. "/auth/index.html"
2001-09-25 23:34:43 +02:00
}
expect = {
code = 200,
body = index
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing auth info overriding: ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://really:wrong@" .. host .. prefix .. "/auth/index.html",
2001-09-25 23:34:43 +02:00
user = "luasocket",
password = "password"
}
expect = {
code = 200,
body = index
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing cgi output retrieval (probably chunked...): ")
2001-09-25 23:34:43 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "http://" .. host .. cgiprefix .. "/cat-index-html"
2001-09-25 23:34:43 +02:00
}
expect = {
body = index,
code = 200
}
ignore = {
status = 1,
headers = 1
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
io.write("testing wrong scheme: ")
2001-09-26 22:29:18 +02:00
request = {
2003-03-28 23:41:26 +01:00
url = "wrong://" .. host .. cgiprefix .. "/cat",
2001-09-26 22:29:18 +02:00
method = "GET"
}
expect = {
error = "unknown scheme 'wrong'"
}
ignore = {
}
check_request(request, expect, ignore)
------------------------------------------------------------------------
2001-09-25 23:34:43 +02:00
local body
io.write("testing simple get function: ")
2003-03-28 23:41:26 +01:00
body = socket.http.get("http://" .. host .. prefix .. "/index.html")
2001-09-25 23:34:43 +02:00
check(body == index)
------------------------------------------------------------------------
io.write("testing simple get function with table args: ")
body = socket.http.get {
2003-03-28 23:41:26 +01:00
url = "http://really:wrong@" .. host .. prefix .. "/auth/index.html",
2001-09-25 23:34:43 +02:00
user = "luasocket",
password = "password"
}
check(body == index)
------------------------------------------------------------------------
io.write("testing HEAD method: ")
socket.http.TIMEOUT = 1
response = socket.http.request {
2001-09-25 23:34:43 +02:00
method = "HEAD",
url = "http://www.cs.princeton.edu/~diego/"
2001-09-25 23:34:43 +02:00
}
check(response and response.headers)
2001-01-25 22:59:39 +01:00
------------------------------------------------------------------------
2001-01-25 22:59:39 +01:00
print("passed all tests")
2001-06-09 00:40:09 +02:00
2003-05-25 03:54:13 +02:00
print(string.format("done in %.2fs", socket.time() - t))