928de738dd
Rename the constants to IP4arpa and IP6arpa (shorter and exported) and make IsReverse return the type of the reverse zone which could be handy for some callers. Also add tests for IsReverse()
71 lines
1.3 KiB
Go
71 lines
1.3 KiB
Go
package dnsutil
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestExtractAddressFromReverse(t *testing.T) {
|
|
tests := []struct {
|
|
reverseName string
|
|
expectedAddress string
|
|
}{
|
|
{
|
|
"54.119.58.176.in-addr.arpa.",
|
|
"176.58.119.54",
|
|
},
|
|
{
|
|
".58.176.in-addr.arpa.",
|
|
"",
|
|
},
|
|
{
|
|
"b.a.9.8.7.6.5.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.8.b.d.0.1.0.0.2.in-addr.arpa.",
|
|
"",
|
|
},
|
|
{
|
|
"b.a.9.8.7.6.5.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.8.b.d.0.1.0.0.2.ip6.arpa.",
|
|
"2001:db8::567:89ab",
|
|
},
|
|
{
|
|
"d.0.1.0.0.2.ip6.arpa.",
|
|
"",
|
|
},
|
|
{
|
|
"54.119.58.176.ip6.arpa.",
|
|
"",
|
|
},
|
|
{
|
|
"NONAME",
|
|
"",
|
|
},
|
|
{
|
|
"",
|
|
"",
|
|
},
|
|
}
|
|
for i, test := range tests {
|
|
got := ExtractAddressFromReverse(test.reverseName)
|
|
if got != test.expectedAddress {
|
|
t.Errorf("Test %d, expected '%s', got '%s'", i, test.expectedAddress, got)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestIsReverse(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
expected int
|
|
}{
|
|
{"b.a.9.8.7.6.5.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.8.b.d.0.1.0.0.2.ip6.arpa.", 2},
|
|
{"d.0.1.0.0.2.in-addr.arpa.", 1},
|
|
{"example.com.", 0},
|
|
{"", 0},
|
|
{"in-addr.arpa.example.com.", 0},
|
|
}
|
|
for i, tc := range tests {
|
|
got := IsReverse(tc.name)
|
|
if got != tc.expected {
|
|
t.Errorf("Test %d, got %d, expected %d for %s", i, got, tc.expected, tc.name)
|
|
}
|
|
|
|
}
|
|
}
|