nodemcu-firmware/lua_examples/httptest.lua

152 lines
4.6 KiB
Lua

--dofile"httptest.lua"
function assertEquals(a, b)
if a ~= b then
error(string.format("%q not equal to %q", tostring(a), tostring(b)), 2)
end
end
-- Skipping test because my internet connection always resolves invalid DNS...
function SKIP_test_bad_dns()
do return end
local c = http.createConnection("http://nope.invalid")
local seenConnect, seenFinish
c:on("connect", function()
seenConnect = true
end)
c:on("complete", function(statusCode)
seenFinish = statusCode
end)
local status, connected = c:request()
assert(status < 0)
assertEquals(connected, false)
assertEquals(seenConnect, false)
assert(seenFinish < 0)
end
function test_simple()
local c = http.createConnection("http://httpbin.org/", { headers = { Connection = "close" } })
local seenConnect, seenHeaders, headersStatusCode
local dataBytes = 0
c:on("connect", function() seenConnect = true end)
c:on("headers", function(statusCode, headers)
headersStatusCode = statusCode
seenHeaders = headers
end)
c:on("data", function(statusCode, data)
assertEquals(statusCode, headersStatusCode)
dataBytes = dataBytes + #data
end)
local status, connected = c:request()
assertEquals(seenConnect, true)
assertEquals(headersStatusCode, 200)
assert(seenHeaders ~= nil)
assert(dataBytes > 0)
assertEquals(status, 200)
assertEquals(connected, false)
end
function test_keepalive()
local c = http.createConnection("http://httpbin.org/")
local seenConnect, seenHeaders, seenData, seenFinish
c:on("connect", function() seenConnect = true end)
c:on("headers", function(status, headers)
seenHeaders = headers
end)
c:on("data", function(status, data)
seenData = data
end)
c:on("complete", function(status)
seenFinish = status
end)
local status, connected = c:request()
assertEquals(seenConnect, true)
assertEquals(status, 200)
assertEquals(connected, true)
assert(seenHeaders)
assertEquals(seenFinish, 200)
c:seturl("http://httpbin.org/user-agent")
c:setheader("Connection", "close")
seenConnect = false
seenData = nil
seenHeaders = nil
seenFinish = nil
status, connected = c:request()
assertEquals(status, 200)
assertEquals(connected, false)
assertEquals(seenConnect, false) -- You don't get another connect callback
assert(seenHeaders) -- But you do get new headers
assertEquals(seenData, '{\n "user-agent": "ESP32 HTTP Client/1.0"\n}\n')
assertEquals(seenFinish, 200)
c:close()
end
function test_oneshot_get()
local options = {
headers = {
["user-agent"] = "ESP32 HTTP Module User Agent test",
},
}
local code, data = http.get("http://httpbin.org/user-agent", options)
assertEquals(code, 200)
assertEquals(data, '{\n "user-agent": "ESP32 HTTP Module User Agent test"\n}\n')
-- And async version
print("test_oneshot_get async starting")
http.get("http://httpbin.org/user-agent", options, function(code, data)
assertEquals(code, 200)
assertEquals(data, '{\n "user-agent": "ESP32 HTTP Module User Agent test"\n}\n')
print("test_oneshot_get async completed")
end)
end
function test_404()
local status = http.get("http://httpbin.org/status/404")
assertEquals(status, 404)
end
function test_post()
local status, data = http.post("http://httpbin.org/post", nil, "hello=world&answer=42")
assertEquals(status, 200)
assert(data:match('"hello": "world"'))
assert(data:match('"answer": "42"'))
status, data = http.post("http://httpbin.org/post", {headers = {["Content-Type"] = "application/json"}}, '{"hello":"world"}')
assertEquals(status, 200)
assert(data:match('"hello": "world"'))
end
function test_redirects()
local status = http.createConnection("http://httpbin.org/redirect/3", { max_redirects = 0 }):request()
assertEquals(status, -28673) -- -ESP_ERR_HTTP_MAX_REDIRECT
status = http.createConnection("http://httpbin.org/redirect/3", { max_redirects = 2 }):request()
assertEquals(status, -28673)
status = http.createConnection("http://httpbin.org/redirect/3"):request()
assertEquals(status, 200)
end
-- Doesn't seem to work...
function SKIP_test_timeout()
local status = http.createConnection("http://httpbin.org/delay/10", { timeout = 2000 }):request()
assertEquals(status, -1)
end
local tests = {}
for k, v in pairs(_G) do
if type(v) == "function" and k:match("^test_") then
table.insert(tests, k)
end
end
table.sort(tests)
for _, t in ipairs(tests) do
print(t)
_G[t]()
end