blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2025-10-16 21:32:11 +08:00
Code Issues Projects Releases Wiki Activity

Add more tests for validateHeader()

Browse Source
This commit is contained in:
Notealot 2021-10-26 23:30:29 +08:00
parent e540fcfeb6
commit d1c9d876fe
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
gin_test.go
View File

@ -657,5 +657,34 @@ func assertRoutePresent(t *testing.T, gotRoutes RoutesInfo, wantRoute RouteInfo)
t.Errorf("route not found: %v", wantRoute) t.Errorf("route not found: %v", wantRoute)
} }
func TestEngineValidateHeader(t *testing.T) {
_, e := CreateTestContext(httptest.NewRecorder())
_ = e.SetTrustedProxies([]string{"30.30.30.30", "40.40.40.40"})
{
// normal legal header
header := "123.123.123.123, 30.30.30.30, 40.40.40.40"
ip, ok := e.validateHeader(header)
assert.Equal(t, true, ok)
assert.Equal(t, "123.123.123.123", ip)
}
{
// partial legal header
header := "50.50.50.50, 123.123.123.123, 30.30.30.30"
ip, ok := e.validateHeader(header)
assert.Equal(t, true, ok)
assert.Equal(t, "123.123.123.123", ip)
}
{
// illegal header
header := "123.123.123.123, 30.30.30.30, 256.255.255.255"
ip, ok := e.validateHeader(header)
assert.Equal(t, false, ok)
assert.Empty(t, ip)
}
}
func handlerTest1(c *Context) {} func handlerTest1(c *Context) {}
func handlerTest2(c *Context) {} func handlerTest2(c *Context) {}