1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
package wire
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestProbingFrames(t *testing.T) {
testCases := map[Frame]bool{
&AckFrame{}: false,
&ConnectionCloseFrame{}: false,
&DataBlockedFrame{}: false,
&PingFrame{}: false,
&ResetStreamFrame{}: false,
&StreamFrame{}: false,
&DatagramFrame{}: false,
&MaxDataFrame{}: false,
&MaxStreamDataFrame{}: false,
&StopSendingFrame{}: false,
&PathChallengeFrame{}: true,
&PathResponseFrame{}: true,
&NewConnectionIDFrame{}: true,
}
for f, expected := range testCases {
require.Equal(t, expected, IsProbingFrame(f))
}
}
func TestIsProbingFrameType(t *testing.T) {
tests := map[FrameType]bool{
FrameTypePathChallenge: true,
FrameTypePathResponse: true,
FrameTypeNewConnectionID: true,
FrameType(0x01): false,
FrameType(0xFF): false,
}
for ft, expected := range tests {
require.Equal(t, expected, IsProbingFrameType(ft))
}
}
|
