diff options
author | 2025-01-27 19:21:13 +0100 | |
---|---|---|
committer | 2025-01-27 19:21:13 +0100 | |
commit | 904829094816fb38d8f1e1d2c19c4c9c014baa88 (patch) | |
tree | ef481d04b884011b838a03c8b3dd58b955c7eaec /internal/middleware/ratelimit_test.go | |
parent | [chore] some tidy ups (#3677) (diff) | |
download | gotosocial-904829094816fb38d8f1e1d2c19c4c9c014baa88.tar.xz |
[chore] skip `trusted-proxies` warning if ip excepted from rate limiting (#3699)
* [chore] skip `trusted-proxies` warning if ip excepted from rate limiting
* weep
* typo
* fix env parsing test
Diffstat (limited to 'internal/middleware/ratelimit_test.go')
-rw-r--r-- | internal/middleware/ratelimit_test.go | 34 |
1 files changed, 6 insertions, 28 deletions
diff --git a/internal/middleware/ratelimit_test.go b/internal/middleware/ratelimit_test.go index ad9891d79..e5afd40a6 100644 --- a/internal/middleware/ratelimit_test.go +++ b/internal/middleware/ratelimit_test.go @@ -20,6 +20,7 @@ package middleware_test import ( "net/http" "net/http/httptest" + "net/netip" "strconv" "testing" "time" @@ -47,60 +48,37 @@ func (suite *RateLimitTestSuite) TestRateLimit() { type rlTest struct { limit int - exceptions []string + exceptions []netip.Prefix clientIP string - shouldPanic bool shouldExcept bool } for _, test := range []rlTest{ { limit: 10, - exceptions: []string{}, + exceptions: nil, clientIP: "192.0.2.0", - shouldPanic: false, shouldExcept: false, }, { limit: 10, - exceptions: []string{}, + exceptions: nil, clientIP: "192.0.2.0", - shouldPanic: false, shouldExcept: false, }, { limit: 10, - exceptions: []string{"192.0.2.0/24"}, + exceptions: []netip.Prefix{netip.MustParsePrefix("192.0.2.0/24")}, clientIP: "192.0.2.0", - shouldPanic: false, shouldExcept: true, }, { limit: 10, - exceptions: []string{"192.0.2.0/32"}, + exceptions: []netip.Prefix{netip.MustParsePrefix("192.0.2.0/32")}, clientIP: "192.0.2.1", - shouldPanic: false, - shouldExcept: false, - }, - { - limit: 10, - exceptions: []string{"Ceci n'est pas une CIDR"}, - clientIP: "192.0.2.0", - shouldPanic: true, shouldExcept: false, }, } { - if test.shouldPanic { - // Try to trigger panic. - suite.Panics(func() { - _ = middleware.RateLimit( - test.limit, - test.exceptions, - ) - }) - continue - } - rlMiddleware := middleware.RateLimit( test.limit, test.exceptions, |