Method readMessage

websocket.go:534–565

(opChain *chain)

}

func (ws *Websocket) readMessage(opChain *chain) *WebsocketMessage {
	wm := newEmptyWebsocketMessage(opChain)

	if !ws.setReadDeadline(opChain) {
		return nil
	}

	var err error
	wm.typ, wm.content, err = ws.conn.ReadMessage()

	if err != nil {
		closeErr, ok := err.(*websocket.CloseError)
		if !ok {
			opChain.fail(AssertionFailure{
				Type: AssertOperation,
				Errors: []error{
					errors.New("failed to read from websocket"),
					err,
				},
			})
			return nil
		}

		wm.typ = websocket.CloseMessage
		wm.closeCode = closeErr.Code
		wm.content = []byte(closeErr.Text)
	}

	ws.printRead(wm.typ, wm.content, wm.closeCode)

	return wm
}

func (ws *Websocket) writeMessage(
	opChain *chain, typ int, content []byte, closeCode ...int,