Method writeMessage

websocket.go:567–619

(
	opChain *chain, typ int, content []byte, closeCode ...int,
)

}

func (ws *Websocket) writeMessage(
	opChain *chain, typ int, content []byte, closeCode ...int,
) {
	switch typ {
	case websocket.TextMessage, websocket.BinaryMessage:
		ws.printWrite(typ, content, 0)

	case websocket.CloseMessage:
		if len(closeCode) > 1 {
			opChain.fail(AssertionFailure{
				Type: AssertUsage,
				Errors: []error{
					errors.New("unexpected multiple closeCode arguments"),
				},
			})
			return
		}

		code := websocket.CloseNormalClosure
		if len(closeCode) > 0 {
			code = closeCode[0]
		}

		ws.printWrite(typ, content, code)

		content = websocket.FormatCloseMessage(code, string(content))

	default:
		opChain.fail(AssertionFailure{
			Type: AssertUsage,
			Errors: []error{
				fmt.Errorf("unexpected websocket message type %s",
					wsMessageType(typ)),
			},
		})
		return
	}

	if !ws.setWriteDeadline(opChain) {
		return
	}

	if err := ws.conn.WriteMessage(typ, content); err != nil {
		opChain.fail(AssertionFailure{
			Type: AssertOperation,
			Errors: []error{
				errors.New("failed to write to websocket"),
				err,
			},
		})
		return
	}
}

func (ws *Websocket) setReadDeadline(opChain *chain) bool {
	deadline := infiniteTime