v/vlib/net/websocket/ws_test.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
}