homelab/traefik
1
0
Fork 0
Code Activity

Add rejectStatusCode option to IPAllowList middleware

Browse source
This commit is contained in:
Jeremy Fleischman 2024-01-09 11:26:05 -08:00 • committed by GitHub
parent fea94a3393
commit ccf3a9995a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
12 changed files with 108 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

26
pkg/middlewares/ipallowlist/ip_allowlist_test.go
View file

@ -30,6 +30,14 @@ func TestNewIPAllowLister(t *testing.T) {
SourceRange: []string{"10.10.10.10"},
},
},
{
desc: "invalid HTTP status code",
allowList: dynamic.IPAllowList{
SourceRange: []string{"10.10.10.10"},
RejectStatusCode: 600,
},
expectedError: true,
},
}
for _, test := range testCases {
@ -73,6 +81,24 @@ func TestIPAllowLister_ServeHTTP(t *testing.T) {
remoteAddr: "20.20.20.21:1234",
expected: 403,
},
{
desc: "authorized with remote address, reject 404",
allowList: dynamic.IPAllowList{
SourceRange: []string{"20.20.20.20"},
RejectStatusCode: 404,
},
remoteAddr: "20.20.20.20:1234",
expected: 200,
},
{
desc: "non authorized with remote address, reject 404",
allowList: dynamic.IPAllowList{
SourceRange: []string{"20.20.20.20"},
RejectStatusCode: 404,
},
remoteAddr: "20.20.20.21:1234",
expected: 404,
},
}
for _, test := range testCases {