67 lines
1.6 KiB
V
67 lines
1.6 KiB
V
import websocket
|
|
import time
|
|
|
|
struct Test {
|
|
mut:
|
|
connected bool = false
|
|
sended_msg []string = []
|
|
recieved_msg []string = []
|
|
}
|
|
|
|
fn test_ws() {
|
|
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.sended_msg << 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.sended_msg.len == test.recieved_msg.len
|
|
for x, msg in test.sended_msg {
|
|
assert msg == test.recieved_msg[x]
|
|
}
|
|
}
|
|
|
|
fn on_open(mut test Test, y voidptr, ws &websocket.Client) {
|
|
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.recieved_msg << cstring_to_vstring(msg.payload)
|
|
} else {
|
|
println('Binary message: $msg')
|
|
}
|
|
}
|
|
|
|
fn on_close(mut test Test, y voidptr, ws &websocket.Client) {
|
|
println('websocket closed.')
|
|
}
|
|
|
|
fn on_error(ws &websocket.Client, x, y voidptr) {
|
|
println('we have an error.')
|
|
assert false
|
|
}
|