mirror of
https://github.com/lunarmodules/luasocket.git
synced 2024-11-08 14:28:21 +01:00
434 lines
10 KiB
Lua
434 lines
10 KiB
Lua
-- 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
|
|
|
|
require("luasocket")
|
|
require("http")
|
|
|
|
dofile("testsupport.lua")
|
|
|
|
local host, proxy, request, response, index_file
|
|
local ignore, expect, index, prefix, cgiprefix, index_crlf
|
|
|
|
socket.http.TIMEOUT = 10
|
|
|
|
local t = socket.time()
|
|
|
|
host = host or "diego.student.princeton.edu"
|
|
proxy = proxy or "http://localhost:3128"
|
|
prefix = prefix or "/luasocket-test"
|
|
cgiprefix = cgiprefix or "/luasocket-test-cgi"
|
|
index_file = "test/index.html"
|
|
|
|
-- read index with CRLF convention
|
|
index = readfile(index_file)
|
|
|
|
local check_result = function(response, expect, ignore)
|
|
for i,v in response do
|
|
if not ignore[i] then
|
|
if v ~= expect[i] then
|
|
local f = io.open("err", "w")
|
|
f:write(tostring(v), "\n\n versus\n\n", tostring(expect[i]))
|
|
f:close()
|
|
fail(i .. " differs!")
|
|
end
|
|
end
|
|
end
|
|
for i,v in expect do
|
|
if not ignore[i] then
|
|
if v ~= response[i] then
|
|
local f = io.open("err", "w")
|
|
f:write(tostring(response[i]), "\n\n versus\n\n", tostring(v))
|
|
v = string.sub(type(v) == "string" and v or "", 1, 70)
|
|
f:close()
|
|
fail(i .. " differs!")
|
|
end
|
|
end
|
|
end
|
|
print("ok")
|
|
end
|
|
|
|
local check_request = function(request, expect, ignore)
|
|
local t
|
|
if not request.sink then
|
|
request.sink, t = ltn12.sink.table(t)
|
|
end
|
|
request.source = request.source or
|
|
(request.body and ltn12.source.string(request.body))
|
|
local response = socket.http.request(request)
|
|
if t and table.getn(t) > 0 then response.body = table.concat(t) end
|
|
check_result(response, expect, ignore)
|
|
end
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing request uri correctness: ")
|
|
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(back.query) end
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing query string correctness: ")
|
|
forth = "this+is+the+query+string"
|
|
back = socket.http.get("http://" .. host .. cgiprefix ..
|
|
"/query-string?" .. forth)
|
|
if similar(back, forth) then print("ok")
|
|
else fail("failed!") end
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing document retrieval: ")
|
|
request = {
|
|
url = "http://" .. host .. prefix .. "/index.html"
|
|
}
|
|
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)
|
|
assert(back == index)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing ltn12.(sink|source).file: ")
|
|
request = {
|
|
url = "http://" .. host .. cgiprefix .. "/cat",
|
|
method = "POST",
|
|
source = ltn12.source.file(io.open(index_file, "r")),
|
|
sink = ltn12.sink.file(io.open(index_file .. "-back", "w")),
|
|
headers = { ["content-length"] = string.len(index) }
|
|
}
|
|
expect = {
|
|
code = 200
|
|
}
|
|
ignore = {
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
back = readfile(index_file .. "-back")
|
|
assert(back == index)
|
|
os.remove(index_file .. "-back")
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing ltn12.(sink|source).chain and mime.(encode|decode): ")
|
|
|
|
local function b64length(len)
|
|
local a = math.ceil(len/3)*4
|
|
local l = math.ceil(a/76)
|
|
return a + l*2
|
|
end
|
|
|
|
local source = ltn12.source.chain(
|
|
ltn12.source.file(io.open(index_file, "r")),
|
|
ltn12.filter.chain(
|
|
mime.encode("base64"),
|
|
mime.wrap("base64")
|
|
)
|
|
)
|
|
|
|
local sink = ltn12.sink.chain(
|
|
mime.decode("base64"),
|
|
ltn12.sink.file(io.open(index_file .. "-back", "w"))
|
|
)
|
|
|
|
request = {
|
|
url = "http://" .. host .. cgiprefix .. "/cat",
|
|
method = "POST",
|
|
source = source,
|
|
sink = sink,
|
|
headers = { ["content-length"] = b64length(string.len(index)) }
|
|
}
|
|
expect = {
|
|
code = 200
|
|
}
|
|
ignore = {
|
|
body_cb = 1,
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
back = readfile(index_file .. "-back")
|
|
assert(back == index)
|
|
os.remove(index_file .. "-back")
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing http redirection: ")
|
|
request = {
|
|
url = "http://" .. host .. prefix
|
|
}
|
|
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: ")
|
|
request = {
|
|
url = "http://really:wrong@" .. host .. prefix .. "/auth/index.html"
|
|
}
|
|
expect = {
|
|
code = 401
|
|
}
|
|
ignore = {
|
|
body = 1,
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing http redirection failure: ")
|
|
request = {
|
|
url = "http://" .. host .. prefix,
|
|
redirect = false
|
|
}
|
|
expect = {
|
|
code = 301
|
|
}
|
|
ignore = {
|
|
body = 1,
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing host not found: ")
|
|
request = {
|
|
url = "http://wronghost/does/not/exist"
|
|
}
|
|
local c, e = socket.connect("wronghost", 80)
|
|
expect = {
|
|
error = e
|
|
}
|
|
ignore = {}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing invalid url: ")
|
|
request = {
|
|
url = host .. prefix
|
|
}
|
|
local c, e = socket.connect("", 80)
|
|
expect = {
|
|
error = e
|
|
}
|
|
ignore = {}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing document not found: ")
|
|
request = {
|
|
url = "http://" .. host .. "/wrongdocument.html"
|
|
}
|
|
expect = {
|
|
code = 404
|
|
}
|
|
ignore = {
|
|
body = 1,
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing auth failure: ")
|
|
request = {
|
|
url = "http://" .. host .. prefix .. "/auth/index.html"
|
|
}
|
|
expect = {
|
|
code = 401
|
|
}
|
|
ignore = {
|
|
body = 1,
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing manual basic auth: ")
|
|
request = {
|
|
url = "http://" .. host .. prefix .. "/auth/index.html",
|
|
headers = {
|
|
authorization = "Basic " .. (mime.b64("luasocket:password"))
|
|
}
|
|
}
|
|
expect = {
|
|
code = 200,
|
|
body = index
|
|
}
|
|
ignore = {
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing automatic basic auth: ")
|
|
request = {
|
|
url = "http://luasocket:password@" .. host .. prefix .. "/auth/index.html"
|
|
}
|
|
expect = {
|
|
code = 200,
|
|
body = index
|
|
}
|
|
ignore = {
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing auth info overriding: ")
|
|
request = {
|
|
url = "http://really:wrong@" .. host .. prefix .. "/auth/index.html",
|
|
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...): ")
|
|
request = {
|
|
url = "http://" .. host .. cgiprefix .. "/cat-index-html"
|
|
}
|
|
expect = {
|
|
body = index,
|
|
code = 200
|
|
}
|
|
ignore = {
|
|
status = 1,
|
|
headers = 1
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing wrong scheme: ")
|
|
request = {
|
|
url = "wrong://" .. host .. cgiprefix .. "/cat",
|
|
method = "GET"
|
|
}
|
|
expect = {
|
|
error = "unknown scheme 'wrong'"
|
|
}
|
|
ignore = {
|
|
}
|
|
check_request(request, expect, ignore)
|
|
|
|
------------------------------------------------------------------------
|
|
local body
|
|
io.write("testing simple get function: ")
|
|
body = socket.http.get("http://" .. host .. prefix .. "/index.html")
|
|
assert(body == index)
|
|
print("ok")
|
|
|
|
------------------------------------------------------------------------
|
|
io.write("testing HEAD method: ")
|
|
socket.http.TIMEOUT = 1
|
|
response = socket.http.request {
|
|
method = "HEAD",
|
|
url = "http://www.cs.princeton.edu/~diego/"
|
|
}
|
|
assert(response and response.headers)
|
|
print("ok")
|
|
|
|
------------------------------------------------------------------------
|
|
print("passed all tests")
|
|
|
|
print(string.format("done in %.2fs", socket.time() - t))
|