Skip to content

Commit d6a55ec

Browse files
fix: enhance IsWebSocket() test cases with Connection header validation
1 parent e075bf4 commit d6a55ec

File tree

1 file changed

+19
-2
lines changed

1 file changed

+19
-2
lines changed

context_test.go

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1119,15 +1119,21 @@ func TestContext_IsWebSocket(t *testing.T) {
11191119
{
11201120
&Context{
11211121
request: &http.Request{
1122-
Header: http.Header{HeaderUpgrade: []string{"websocket"}},
1122+
Header: http.Header{
1123+
HeaderUpgrade: []string{"websocket"},
1124+
HeaderConnection: []string{"upgrade"},
1125+
},
11231126
},
11241127
},
11251128
assert.True,
11261129
},
11271130
{
11281131
&Context{
11291132
request: &http.Request{
1130-
Header: http.Header{HeaderUpgrade: []string{"Websocket"}},
1133+
Header: http.Header{
1134+
HeaderUpgrade: []string{"Websocket"},
1135+
HeaderConnection: []string{"Upgrade"},
1136+
},
11311137
},
11321138
},
11331139
assert.True,
@@ -1146,6 +1152,17 @@ func TestContext_IsWebSocket(t *testing.T) {
11461152
},
11471153
assert.False,
11481154
},
1155+
{
1156+
&context{
1157+
request: &http.Request{
1158+
Header: http.Header{
1159+
HeaderUpgrade: []string{"websocket"},
1160+
HeaderConnection: []string{"close"},
1161+
},
1162+
},
1163+
},
1164+
assert.False,
1165+
},
11491166
}
11501167

11511168
for i, tt := range tests {

0 commit comments

Comments
 (0)