websocket: add tests (#5967)

vlib/net/websocket/ws_test.v

@@ -1,6 +1,66 @@
-module websocket
+import websocket
+import time
 
-// TODO This only checks that the client compiles, do real tests
+struct Test {
-fn test_compile() {
+mut:
-	new('http://examples.com')
+	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
 }