72 lines
1.8 KiB
V
72 lines
1.8 KiB
V
import websocket
|
|
import time
|
|
|
|
struct Test {
|
|
mut:
|
|
connected bool = false
|
|
sent_messages []string = []
|
|
received_messages []string = []
|
|
}
|
|
|
|
fn test_ws() {
|
|
$if !network ? {
|
|
return
|
|
}
|
|
ws_test('ws://echo.websocket.org')
|
|
ws_test('wss://echo.websocket.org')
|
|
}
|
|
|
|
fn ws_test(uri string) {
|
|
mut test := Test{}
|
|
println('connecting to $uri ...')
|
|
mut ws := websocket.new(uri)
|
|
ws.subscriber.subscribe_method('on_open', on_open, test)
|
|
ws.subscriber.subscribe_method('on_message', on_message, test)
|
|
ws.subscriber.subscribe('on_error', on_error)
|
|
ws.subscriber.subscribe_method('on_close', on_close, test)
|
|
ws.connect()
|
|
go ws.listen()
|
|
text := ['ws test', '{"vlang": "test0\n192"}']
|
|
for msg in text {
|
|
test.sent_messages << msg
|
|
len := ws.write(msg.str, msg.len, .text_frame)
|
|
assert msg.len <= len
|
|
// sleep to give time to recieve response before send a new one
|
|
time.sleep_ms(100)
|
|
}
|
|
// sleep to give time to recieve response before asserts
|
|
time.sleep_ms(500)
|
|
|
|
assert test.connected == true
|
|
assert test.sent_messages.len == test.received_messages.len
|
|
for x, msg in test.sent_messages {
|
|
assert msg == test.received_messages[x]
|
|
}
|
|
}
|
|
|
|
fn on_open(mut test Test, x voidptr, mut ws &websocket.Client) {
|
|
// Send PONG only for testing porposes
|
|
ws.send_pong()
|
|
println('websocket opened.')
|
|
test.connected = true
|
|
}
|
|
|
|
fn on_message(mut test Test, msg &websocket.Message, ws &websocket.Client) {
|
|
typ := msg.opcode
|
|
if typ == .text_frame {
|
|
println('Message: ${cstring_to_vstring(msg.payload)}')
|
|
test.received_messages << cstring_to_vstring(msg.payload)
|
|
} else {
|
|
println('Binary message: $msg')
|
|
}
|
|
}
|
|
|
|
fn on_close(x, y voidptr, ws &websocket.Client) {
|
|
println('websocket closed.')
|
|
}
|
|
|
|
fn on_error(x, y voidptr, ws &websocket.Client) {
|
|
println('we have an error.')
|
|
assert false
|
|
}
|