v/vlib/net/http/server_test.v

91 lines
2.2 KiB
V

import net.http
import time
fn test_server_stop() ? {
mut server := &http.Server{
accept_timeout: 1 * time.second
}
t := go server.listen_and_serve()
time.sleep(250 * time.millisecond)
mut watch := time.new_stopwatch()
server.stop()
assert server.status() == .stopped
assert watch.elapsed() < 100 * time.millisecond
t.wait() ?
assert watch.elapsed() < 999 * time.millisecond
}
fn test_server_close() ? {
mut server := &http.Server{
accept_timeout: 1 * time.second
handler: MyHttpHandler{}
}
t := go server.listen_and_serve()
time.sleep(250 * time.millisecond)
mut watch := time.new_stopwatch()
server.close()
assert server.status() == .closed
assert watch.elapsed() < 100 * time.millisecond
t.wait() ?
assert watch.elapsed() < 999 * time.millisecond
}
struct MyHttpHandler {
mut:
counter int
oks int
not_founds int
}
fn (mut handler MyHttpHandler) handle(req http.Request) http.Response {
handler.counter++
// eprintln('$time.now() | counter: $handler.counter | $req.method $req.url\n$req.header\n$req.data - 200 OK\n')
mut r := http.Response{
text: req.data + ', $req.url'
header: req.header
}
match req.url.all_before('?') {
'/endpoint', '/another/endpoint' {
r.set_status(.ok)
handler.oks++
}
else {
r.set_status(.not_found)
handler.not_founds++
}
}
r.set_version(req.version)
return r
}
const cport = 8198
fn test_server_custom_handler() ? {
mut handler := MyHttpHandler{}
mut server := &http.Server{
accept_timeout: 1 * time.second
handler: handler
port: cport
}
t := go server.listen_and_serve()
for server.status() != .running {
time.sleep(10 * time.millisecond)
}
x := http.fetch(url: 'http://localhost:$cport/endpoint?abc=xyz', data: 'my data') ?
assert x.text == 'my data, /endpoint?abc=xyz'
assert x.status_code == 200
assert x.http_version == '1.1'
y := http.fetch(url: 'http://localhost:$cport/another/endpoint', data: 'abcde') ?
assert y.text == 'abcde, /another/endpoint'
assert y.status_code == 200
assert y.status() == .ok
assert y.http_version == '1.1'
//
http.fetch(url: 'http://localhost:$cport/something/else') ?
server.stop()
t.wait() ?
assert handler.counter == 3
assert handler.oks == 2
assert handler.not_founds == 1
}