William Petit
a50f926463
All checks were successful
Cadoles/bouncer/pipeline/head This commit looks good
78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
package cidr
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func TestMatchAny(t *testing.T) {
|
|
type testCase struct {
|
|
RemoteHostPort string
|
|
AuthorizedCIDRs []string
|
|
ExpectedResult bool
|
|
ExpectedError error
|
|
}
|
|
|
|
testCases := []testCase{
|
|
{
|
|
RemoteHostPort: "192.168.1.15",
|
|
AuthorizedCIDRs: []string{
|
|
"192.168.1.0/24",
|
|
},
|
|
ExpectedResult: true,
|
|
},
|
|
{
|
|
RemoteHostPort: "192.168.1.15:43349",
|
|
AuthorizedCIDRs: []string{
|
|
"192.168.1.0/24",
|
|
},
|
|
ExpectedResult: true,
|
|
},
|
|
{
|
|
RemoteHostPort: "192.168.1.15:43349",
|
|
AuthorizedCIDRs: []string{
|
|
"192.168.1.5/32",
|
|
},
|
|
ExpectedResult: false,
|
|
},
|
|
{
|
|
RemoteHostPort: "192.168.1.15:43349",
|
|
AuthorizedCIDRs: []string{
|
|
"192.168.1.5/32",
|
|
"192.168.1.0/24",
|
|
},
|
|
ExpectedResult: true,
|
|
},
|
|
{
|
|
RemoteHostPort: "192.168.1.15:43349",
|
|
AuthorizedCIDRs: []string{
|
|
"192.168.1.5/32",
|
|
"192.168.1.6/32",
|
|
"192.168.1.7/32",
|
|
},
|
|
ExpectedResult: false,
|
|
},
|
|
{
|
|
RemoteHostPort: "[2001:0db8:0000:85a3:0000:0000:ac1f:8001]:8001",
|
|
AuthorizedCIDRs: []string{"2000::/3"},
|
|
ExpectedResult: true,
|
|
},
|
|
}
|
|
|
|
for idx, tc := range testCases {
|
|
t.Run(fmt.Sprintf("Case #%d", idx), func(t *testing.T) {
|
|
result, err := MatchAny(tc.RemoteHostPort, tc.AuthorizedCIDRs...)
|
|
|
|
if g, e := result, tc.ExpectedResult; e != g {
|
|
t.Errorf("result: expected '%v', got '%v'", e, g)
|
|
}
|
|
|
|
if e, g := tc.ExpectedError, err; !errors.Is(err, tc.ExpectedError) {
|
|
t.Errorf("err: expected '%v', got '%v'", e, g)
|
|
}
|
|
})
|
|
}
|
|
}
|