starred/godoxy-yusing
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-03-28 11:51:53 +01:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

eliminate SonarCloud hardcoded IP complains

Browse Source
This commit is contained in:
yusing
2024-12-16 10:59:36 +08:00
parent 2f275ca81e
commit d82594bf09
2 changed files with 27 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
internal/net/http/middleware/cidr_whitelist_test.go
View File

@@ -4,6 +4,7 @@ import (
_ "embed"
"net"
"net/http"
"strings"
"testing"
E "github.com/yusing/go-proxy/internal/error"
@@ -16,31 +17,32 @@ var testCIDRWhitelistCompose []byte
var deny, accept *Middleware
func TestCIDRWhitelistValidation(t *testing.T) {
const testMessage = "test-message"
t.Run("valid", func(t *testing.T) {
_, err := CIDRWhiteList.New(OptionsRaw{
"allow": []string{"1.2.3.4/32"},
"message": "test-message",
"allow": []string{"192.168.2.100/32"},
"message": testMessage,
})
ExpectNoError(t, err)
})
t.Run("missing allow", func(t *testing.T) {
_, err := CIDRWhiteList.New(OptionsRaw{
"message": "test-message",
"message": testMessage,
})
ExpectError(t, utils.ErrValidationError, err)
})
t.Run("invalid cidr", func(t *testing.T) {
_, err := CIDRWhiteList.New(OptionsRaw{
"allow": []string{"1.2.3.4/123"},
"message": "test-message",
"allow": []string{"192.168.2.100/123"},
"message": testMessage,
})
ExpectErrorT[*net.ParseError](t, err)
})
t.Run("invalid status code", func(t *testing.T) {
_, err := CIDRWhiteList.New(OptionsRaw{
"allow": []string{"1.2.3.4/32"},
"allow": []string{"192.168.2.100/32"},
"status_code": 600,
"message": "test-message",
"message": testMessage,
})
ExpectError(t, utils.ErrValidationError, err)
})
@@ -62,7 +64,7 @@ func TestCIDRWhitelist(t *testing.T) {
result, err := newMiddlewareTest(deny, nil)
ExpectNoError(t, err)
ExpectEqual(t, result.ResponseStatus, cidrWhitelistDefaults.StatusCode)
ExpectEqual(t, string(result.Data), cidrWhitelistDefaults.Message)
ExpectEqual(t, strings.TrimSpace(string(result.Data)), cidrWhitelistDefaults.Message)
}
})