2021-03-31 02:36:57 -07:00
|
|
|
package aghnet
|
|
|
|
|
|
|
|
import (
|
|
|
|
"net"
|
2021-04-07 06:36:38 -07:00
|
|
|
"strings"
|
2021-03-31 02:36:57 -07:00
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestValidateHardwareAddress(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
|
|
name string
|
|
|
|
wantErrMsg string
|
|
|
|
in net.HardwareAddr
|
|
|
|
}{{
|
|
|
|
name: "success_eui_48",
|
|
|
|
wantErrMsg: "",
|
|
|
|
in: net.HardwareAddr{0x00, 0x01, 0x02, 0x03, 0x04, 0x05},
|
|
|
|
}, {
|
|
|
|
name: "success_eui_64",
|
|
|
|
wantErrMsg: "",
|
|
|
|
in: net.HardwareAddr{0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07},
|
|
|
|
}, {
|
|
|
|
name: "success_infiniband",
|
|
|
|
wantErrMsg: "",
|
|
|
|
in: net.HardwareAddr{
|
|
|
|
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
|
|
|
|
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
|
|
|
|
0x10, 0x11, 0x12, 0x13,
|
|
|
|
},
|
|
|
|
}, {
|
|
|
|
name: "error_nil",
|
|
|
|
wantErrMsg: `validating hardware address "": address is empty`,
|
|
|
|
in: nil,
|
|
|
|
}, {
|
|
|
|
name: "error_empty",
|
|
|
|
wantErrMsg: `validating hardware address "": address is empty`,
|
|
|
|
in: net.HardwareAddr{},
|
|
|
|
}, {
|
|
|
|
name: "error_bad",
|
|
|
|
wantErrMsg: `validating hardware address "00:01:02:03": bad len: 4`,
|
|
|
|
in: net.HardwareAddr{0x00, 0x01, 0x02, 0x03},
|
|
|
|
}}
|
|
|
|
|
|
|
|
for _, tc := range testCases {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
err := ValidateHardwareAddress(tc.in)
|
|
|
|
if tc.wantErrMsg == "" {
|
|
|
|
assert.NoError(t, err)
|
|
|
|
} else {
|
|
|
|
require.Error(t, err)
|
2021-04-07 06:36:38 -07:00
|
|
|
|
|
|
|
assert.Equal(t, tc.wantErrMsg, err.Error())
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func repeatStr(b *strings.Builder, s string, n int) {
|
|
|
|
for i := 0; i < n; i++ {
|
|
|
|
_, _ = b.WriteString(s)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestValidateDomainName(t *testing.T) {
|
|
|
|
b := &strings.Builder{}
|
|
|
|
repeatStr(b, "a", 255)
|
|
|
|
longDomainName := b.String()
|
|
|
|
|
|
|
|
b.Reset()
|
|
|
|
repeatStr(b, "a", 64)
|
|
|
|
longLabel := b.String()
|
|
|
|
|
|
|
|
_, _ = b.WriteString(".com")
|
|
|
|
longLabelDomainName := b.String()
|
|
|
|
|
|
|
|
testCases := []struct {
|
|
|
|
name string
|
|
|
|
in string
|
|
|
|
wantErrMsg string
|
|
|
|
}{{
|
|
|
|
name: "success",
|
|
|
|
in: "example.com",
|
|
|
|
wantErrMsg: "",
|
|
|
|
}, {
|
|
|
|
name: "success_idna",
|
|
|
|
in: "пример.рф",
|
|
|
|
wantErrMsg: "",
|
|
|
|
}, {
|
|
|
|
name: "bad_symbol",
|
|
|
|
in: "!!!",
|
|
|
|
wantErrMsg: `invalid domain name label at index 0: ` +
|
|
|
|
`invalid char '!' at index 0 in "!!!"`,
|
|
|
|
}, {
|
|
|
|
name: "bad_length",
|
|
|
|
in: longDomainName,
|
|
|
|
wantErrMsg: `"` + longDomainName + `" is too long, max: 253`,
|
|
|
|
}, {
|
|
|
|
name: "bad_label_length",
|
|
|
|
in: longLabelDomainName,
|
|
|
|
wantErrMsg: `invalid domain name label at index 0: "` + longLabel +
|
|
|
|
`" is too long, max: 63`,
|
|
|
|
}, {
|
|
|
|
name: "bad_label_empty",
|
|
|
|
in: "example..com",
|
|
|
|
wantErrMsg: `invalid domain name label at index 1: label is empty`,
|
|
|
|
}, {
|
|
|
|
name: "bad_label_first_symbol",
|
|
|
|
in: "example.-aa.com",
|
|
|
|
wantErrMsg: `invalid domain name label at index 1:` +
|
|
|
|
` invalid char '-' at index 0 in "-aa"`,
|
|
|
|
}, {
|
|
|
|
name: "bad_label_symbol",
|
|
|
|
in: "example.a!!!.com",
|
|
|
|
wantErrMsg: `invalid domain name label at index 1:` +
|
|
|
|
` invalid char '!' at index 1 in "a!!!"`,
|
|
|
|
}}
|
|
|
|
|
|
|
|
for _, tc := range testCases {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
err := ValidateDomainName(tc.in)
|
|
|
|
if tc.wantErrMsg == "" {
|
|
|
|
assert.NoError(t, err)
|
|
|
|
} else {
|
|
|
|
require.Error(t, err)
|
|
|
|
|
2021-03-31 02:36:57 -07:00
|
|
|
assert.Equal(t, tc.wantErrMsg, err.Error())
|
|
|
|
}
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
2021-04-15 06:59:05 -07:00
|
|
|
|
|
|
|
func TestGenerateHostName(t *testing.T) {
|
|
|
|
testCases := []struct {
|
|
|
|
name string
|
|
|
|
want string
|
|
|
|
ip net.IP
|
|
|
|
}{{
|
|
|
|
name: "good_ipv4",
|
|
|
|
want: "127-0-0-1",
|
|
|
|
ip: net.IP{127, 0, 0, 1},
|
|
|
|
}, {
|
|
|
|
name: "bad_ipv4",
|
|
|
|
want: "",
|
|
|
|
ip: net.IP{127, 0, 0, 1, 0},
|
|
|
|
}, {
|
|
|
|
name: "good_ipv6",
|
|
|
|
want: "fe00-0000-0000-0000-0000-0000-0000-0001",
|
|
|
|
ip: net.ParseIP("fe00::1"),
|
|
|
|
}, {
|
|
|
|
name: "bad_ipv6",
|
|
|
|
want: "",
|
|
|
|
ip: net.IP{
|
|
|
|
0xff, 0xff, 0xff, 0xff,
|
|
|
|
0xff, 0xff, 0xff, 0xff,
|
|
|
|
0xff, 0xff, 0xff, 0xff,
|
|
|
|
0xff, 0xff, 0xff,
|
|
|
|
},
|
|
|
|
}, {
|
|
|
|
name: "nil",
|
|
|
|
want: "",
|
|
|
|
ip: nil,
|
|
|
|
}}
|
|
|
|
|
|
|
|
for _, tc := range testCases {
|
|
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
|
|
hostname := GenerateHostName(tc.ip)
|
|
|
|
assert.Equal(t, tc.want, hostname)
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|