|
|
@@ -2,7 +2,6 @@ package util
|
|
|
|
|
|
import (
|
|
|
"errors"
|
|
|
- "golang.org/x/time/rate"
|
|
|
"io"
|
|
|
"net/netip"
|
|
|
"os"
|
|
|
@@ -11,6 +10,8 @@ import (
|
|
|
"testing"
|
|
|
"time"
|
|
|
|
|
|
+ "golang.org/x/time/rate"
|
|
|
+
|
|
|
"github.com/stretchr/testify/require"
|
|
|
)
|
|
|
|
|
|
@@ -49,6 +50,11 @@ func TestContains(t *testing.T) {
|
|
|
require.False(t, Contains(s, 3))
|
|
|
}
|
|
|
|
|
|
+func TestContainsAll(t *testing.T) {
|
|
|
+ require.True(t, ContainsAll([]int{1, 2, 3}, []int{2, 3}))
|
|
|
+ require.False(t, ContainsAll([]int{1, 1}, []int{1, 2}))
|
|
|
+}
|
|
|
+
|
|
|
func TestContainsIP(t *testing.T) {
|
|
|
require.True(t, ContainsIP([]netip.Prefix{netip.MustParsePrefix("fd00::/8"), netip.MustParsePrefix("1.1.0.0/16")}, netip.MustParseAddr("1.1.1.1")))
|
|
|
require.True(t, ContainsIP([]netip.Prefix{netip.MustParsePrefix("fd00::/8"), netip.MustParsePrefix("1.1.0.0/16")}, netip.MustParseAddr("fd12:1234:5678::9876")))
|