mirror of
https://github.com/labstack/echo.git
synced 2026-04-24 20:55:53 +02:00
fix: enhance IsWebSocket() test cases with Connection header validation
This commit is contained in:
+19
-2
@@ -1119,7 +1119,10 @@ func TestContext_IsWebSocket(t *testing.T) {
|
||||
{
|
||||
&Context{
|
||||
request: &http.Request{
|
||||
Header: http.Header{HeaderUpgrade: []string{"websocket"}},
|
||||
Header: http.Header{
|
||||
HeaderUpgrade: []string{"websocket"},
|
||||
HeaderConnection: []string{"upgrade"},
|
||||
},
|
||||
},
|
||||
},
|
||||
assert.True,
|
||||
@@ -1127,7 +1130,10 @@ func TestContext_IsWebSocket(t *testing.T) {
|
||||
{
|
||||
&Context{
|
||||
request: &http.Request{
|
||||
Header: http.Header{HeaderUpgrade: []string{"Websocket"}},
|
||||
Header: http.Header{
|
||||
HeaderUpgrade: []string{"Websocket"},
|
||||
HeaderConnection: []string{"Upgrade"},
|
||||
},
|
||||
},
|
||||
},
|
||||
assert.True,
|
||||
@@ -1146,6 +1152,17 @@ func TestContext_IsWebSocket(t *testing.T) {
|
||||
},
|
||||
assert.False,
|
||||
},
|
||||
{
|
||||
&Context{
|
||||
request: &http.Request{
|
||||
Header: http.Header{
|
||||
HeaderUpgrade: []string{"websocket"},
|
||||
HeaderConnection: []string{"close"},
|
||||
},
|
||||
},
|
||||
},
|
||||
assert.False,
|
||||
},
|
||||
}
|
||||
|
||||
for i, tt := range tests {
|
||||
|
||||
Reference in New Issue
Block a user