import context
import net.websocket
import x.async as xasync

fn test_net_websocket_task_processes_message_in_memory() {
	msg := websocket.Message{
		opcode:  .text_frame
		payload: 'hello'.bytes()
	}
	mut task := xasync.run[string](fn [msg] (mut ctx context.Context) !string {
		_ = ctx
		return websocket_message_text(msg)!
	})!
	assert task.wait()! == 'hello'
}

fn test_net_websocket_callback_error_propagates_through_group() {
	mut group := xasync.new_group(context.background())
	observed := chan string{cap: 1}
	group.go(fn [observed] (mut ctx context.Context) ! {
		_ = ctx
		msg := websocket.Message{
			opcode: .close
		}
		text := websocket_message_text(msg) or {
			observed <- err.msg()
			return err
		}
		observed <- text
	})!
	group.wait() or {
		assert err.msg() == 'unsupported websocket opcode'
		assert <-observed == 'unsupported websocket opcode'
		return
	}
	assert false
}

fn websocket_message_text(msg websocket.Message) !string {
	if msg.opcode != .text_frame {
		return error('unsupported websocket opcode')
	}
	return msg.payload.bytestr()
}