Function TestWebsocketMessage_MatchCodes

websocket_message_test.go:536–590

(t *testing.T)

}

func TestWebsocketMessage_MatchCodes(t *testing.T) {
	cases := []struct {
		name        string
		typ         int
		code        int
		testCodes   []int
		wantCode    chainResult
		wantNotCode chainResult
	}{
		{
			name:        "close message with close code 10 and with test codes 10 and 20",
			typ:         websocket.CloseMessage,
			code:        10,
			testCodes:   []int{10, 20},
			wantCode:    success,
			wantNotCode: failure,
		},
		{
			name:        "close message with close code 10 and with test codes 20 and 10",
			typ:         websocket.CloseMessage,
			code:        10,
			testCodes:   []int{20, 10},
			wantCode:    success,
			wantNotCode: failure,
		},
		{
			name:        "close message with close code 10 and with test codes 30 and 20",
			typ:         websocket.CloseMessage,
			code:        10,
			testCodes:   []int{30, 20},
			wantCode:    failure,
			wantNotCode: success,
		},
		{
			name:        "close message with close code 10 and with test codes 20 and 30",
			typ:         websocket.CloseMessage,
			code:        10,
			testCodes:   []int{20, 30},
			wantCode:    failure,
			wantNotCode: success,
		},
	}

	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			reporter := newMockReporter(t)

			NewWebsocketMessage(reporter, tc.typ, nil, tc.code).Code(tc.testCodes...).
				chain.assert(t, tc.wantCode)

			NewWebsocketMessage(reporter, tc.typ, nil, tc.code).NotCode(tc.testCodes...).
				chain.assert(t, tc.wantNotCode)
		})
	}
}

func TestWebsocketMessage_CodeAndType(t *testing.T) {
	cases := []struct {