Function TestWebsocket_WriteText

websocket_test.go:766–822

(t *testing.T)

}

func TestWebsocket_WriteText(t *testing.T) {
	type args struct {
		wsConn     WebsocketConn
		wsPreSteps func(*Websocket)
		content    string
	}
	cases := []struct {
		name   string
		args   args
		result chainResult
	}{
		{
			name: "success",
			args: args{
				wsPreSteps: noWsPreSteps,
				wsConn:     &mockWebsocketConn{},
				content:    "random message...",
			},
			result: success,
		},
		{
			name: "conn is nil",
			args: args{
				wsConn:     nil,
				wsPreSteps: noWsPreSteps,
				content:    "random message...",
			},
			result: failure,
		},
		{
			name: "websocket unusable",
			args: args{
				wsConn: &mockWebsocketConn{},
				wsPreSteps: func(ws *Websocket) {
					ws.Disconnect()
				},
				content: "random message...",
			},
			result: failure,
		},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			reporter := newMockReporter(t)
			chain := newChainWithDefaults("test", reporter)
			config := newMockConfig(reporter)

			ws := newWebsocket(chain, config, tc.args.wsConn)

			tc.args.wsPreSteps(ws)

			ws.WriteText(tc.args.content)

			ws.chain.assert(t, tc.result)
		})
	}
}