blob: 790da30b5601078dc8c2dc5ab185a3e961bbe20c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
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
}
|
