mirror of
https://github.com/datarhei/core.git
synced 2025-11-01 03:42:51 +08:00
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package net
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestIPLimiterNew(t *testing.T) {
|
|
var err error
|
|
|
|
_, err = NewIPLimiter([]string{}, []string{})
|
|
require.Nil(t, err)
|
|
|
|
_, err = NewIPLimiter([]string{"::1/128", "127.0.0.1/32"}, []string{})
|
|
require.Nil(t, err)
|
|
|
|
_, err = NewIPLimiter([]string{}, []string{"::1/128", "127.0.0.1/32"})
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestIPLimiterError(t *testing.T) {
|
|
var err error
|
|
|
|
_, err = NewIPLimiter([]string{}, []string{})
|
|
require.Nil(t, err)
|
|
|
|
_, err = NewIPLimiter([]string{"::1"}, []string{})
|
|
require.Nil(t, err, "Should accept IP")
|
|
|
|
_, err = NewIPLimiter([]string{}, []string{"::1"})
|
|
require.Nil(t, err, "Should accept IP")
|
|
}
|
|
|
|
func TestIPLimiterInvalidIPs(t *testing.T) {
|
|
limiter, _ := NewIPLimiter([]string{}, []string{})
|
|
|
|
require.False(t, limiter.IsAllowed(""), "Invalid IP shouldn't be allowed")
|
|
}
|
|
|
|
func TestIPLimiterNoIPs(t *testing.T) {
|
|
limiter, _ := NewIPLimiter([]string{}, []string{})
|
|
|
|
require.True(t, limiter.IsAllowed("127.0.0.1"), "IP should be allowed")
|
|
}
|
|
|
|
func TestIPLimiterAllowlist(t *testing.T) {
|
|
limiter, _ := NewIPLimiter([]string{}, []string{"::1/128"})
|
|
|
|
require.False(t, limiter.IsAllowed("127.0.0.1"), "Unallowed IP shouldn't be allowed")
|
|
require.True(t, limiter.IsAllowed("::1"), "Allowed IP should be allowed")
|
|
}
|
|
|
|
func TestIPLimiterBlocklist(t *testing.T) {
|
|
limiter, _ := NewIPLimiter([]string{"::1/128"}, []string{})
|
|
|
|
require.True(t, limiter.IsAllowed("127.0.0.1"), "Allowed IP should be allowed")
|
|
require.False(t, limiter.IsAllowed("::1"), "Unallowed IP shouldn't be allowed")
|
|
}
|