mirror of
https://github.com/vlang/v.git
synced 2023-08-10 21:13:21 +03:00
net.http: add a close method to immediatly close the listener of a started http.Server, add more tests (#11248)
This commit is contained in:
@ -1,16 +1,90 @@
|
||||
module http
|
||||
|
||||
import net.http
|
||||
import time
|
||||
|
||||
fn test_server_stop() ? {
|
||||
server := &Server{
|
||||
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
|
||||
}
|
||||
|
Reference in New Issue
Block a user