Function TestWebsocket_WriteMessage

websocket_test.go:511–648

(t *testing.T)

}

func TestWebsocket_WriteMessage(t *testing.T) {
	type args struct {
		wsConn     WebsocketConn
		wsPreSteps func(*Websocket)
		typ        int
		content    []byte
		closeCode  []int
	}
	cases := []struct {
		name   string
		args   args
		result chainResult
	}{
		{
			name: "text message success",
			args: args{
				wsPreSteps: noWsPreSteps,
				wsConn:     &mockWebsocketConn{},
				typ:        websocket.TextMessage,
				content:    []byte("random message..."),
				closeCode:  []int{},
			},
			result: success,
		},
		{
			name: "text message fail nil conn",
			args: args{
				wsConn:     nil,
				wsPreSteps: noWsPreSteps,
				typ:        websocket.TextMessage,
				content:    []byte("random message..."),
				closeCode:  []int{},
			},
			result: failure,
		},
		{
			name: "text message fail unusable",
			args: args{
				wsConn: &mockWebsocketConn{},
				wsPreSteps: func(ws *Websocket) {
					ws.Disconnect()
				},
				typ:       websocket.TextMessage,
				content:   []byte("random message..."),
				closeCode: []int{},
			},
			result: failure,
		},
		{
			name: "text message failed to set write deadline",
			args: args{
				wsConn: &mockWebsocketConn{
					writeDlError: errors.New("failed to set write deadline"),
				},
				wsPreSteps: noWsPreSteps,
				typ:        websocket.TextMessage,
				content:    []byte("random message..."),
				closeCode:  []int{},