From 1f8ae5d12ce869a8e173dbbbdaa4e6b0afa6e727 Mon Sep 17 00:00:00 2001
From: Carlos Esquerdo Bernat <esquerbatua@gmail.com>
Date: Sat, 25 Jul 2020 22:25:31 +0200
Subject: [PATCH] websocket: add tests (#5967)

---
 vlib/net/websocket/ws_test.v | 68 +++++++++++++++++++++++++++++++++---
 1 file changed, 64 insertions(+), 4 deletions(-)

diff --git a/vlib/net/websocket/ws_test.v b/vlib/net/websocket/ws_test.v
index 0c7aef3c78..d785f44b8d 100644
--- a/vlib/net/websocket/ws_test.v
+++ b/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
-fn test_compile() {
-	new('http://examples.com')
+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
 }