Function TestWebsocket_SetWriteDeadline

websocket_test.go:958–1000

(t *testing.T)

}

func TestWebsocket_SetWriteDeadline(t *testing.T) {
	type args struct {
		wsConn WebsocketConn
	}
	cases := []struct {
		name   string
		args   args
		result chainResult
	}{
		{
			name: "success",
			args: args{
				wsConn: &mockWebsocketConn{},
			},
			result: success,
		},
		{
			name: "conn.SetReadDeadline error",
			args: args{
				wsConn: &mockWebsocketConn{
					writeDlError: errors.New("Failed to set read deadline"),
				},
			},
			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).
				WithWriteTimeout(time.Second)

			opChain := ws.chain.enter("test")
			ws.setWriteDeadline(opChain)
			opChain.leave()

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

func TestWebsocket_Disconnect(t *testing.T) {
	type args struct {