v_windows/v/vlib/net/websocket/tests/autobahn/autobahn_client.v


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

// use this test to test the websocket client in the autobahn test
module main

import net.websocket

fn main() {
	for i in 1 .. 304 {
		println('\ncase: $i')
		handle_case(i) or { println('error should be ok: $err') }
	}
	// update the reports
	uri := 'ws://autobahn_server:9001/updateReports?agent=v-client'
	mut ws := websocket.new_client(uri) ?
	ws.connect() ?
	ws.listen() ?
}

fn handle_case(case_nr int) ? {
	uri := 'ws://autobahn_server:9001/runCase?case=$case_nr&agent=v-client'
	mut ws := websocket.new_client(uri) ?
	ws.on_message(on_message)
	ws.connect() ?
	ws.listen() ?
}

fn on_message(mut ws websocket.Client, msg &websocket.Message) ? {
	// autobahn tests expects to send same message back
	if msg.opcode == .pong {
		// We just wanna pass text and binary message back to autobahn
		return
	}
	ws.write(msg.payload, msg.opcode) or { panic(err) }
}