mirror of
https://github.com/StackExchange/dnscontrol.git
synced 2024-09-20 23:06:23 +08:00
225 lines
6.2 KiB
Go
225 lines
6.2 KiB
Go
package transform
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestIPToUint(t *testing.T) {
|
|
ip := net.ParseIP("1.2.3.4")
|
|
u, err := ipToUint(ip)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if u != 16909060 {
|
|
t.Fatalf("I to uint conversion failed. Should be 16909060. Got %d", u)
|
|
}
|
|
ip2 := UintToIP(u)
|
|
if !ip.Equal(ip2) {
|
|
t.Fatalf("IPs should be equal. %s is not %s", ip2, ip)
|
|
}
|
|
}
|
|
|
|
func TestDecodeTransformTableFailures(t *testing.T) {
|
|
result, err := DecodeTransformTable("1.2.3.4 ~ 3.4.5.6")
|
|
if result != nil {
|
|
t.Errorf("expected nil, got (%v)\n", result)
|
|
}
|
|
if err == nil {
|
|
t.Error("expect error, got none")
|
|
}
|
|
}
|
|
|
|
func testIP(t *testing.T, test string, expected string, actual net.IP) {
|
|
if !net.ParseIP(expected).Equal(actual) {
|
|
t.Errorf("Test %v: expected Low (%v), got (%v)\n", test, actual, expected)
|
|
}
|
|
}
|
|
|
|
func Test_DecodeTransformTable_0(t *testing.T) {
|
|
result, err := DecodeTransformTable("1.2.3.4 ~ 2.3.4.5 ~ 3.4.5.6 ~ ")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(result) != 1 {
|
|
t.Errorf("Test %v: expected col length (%v), got (%v)\n", 1, 1, len(result))
|
|
}
|
|
testIP(t, "low", "1.2.3.4", result[0].Low)
|
|
testIP(t, "high", "2.3.4.5", result[0].High)
|
|
testIP(t, "newBase", "3.4.5.6", result[0].NewBases[0])
|
|
// test_ip(t, "newIP", "", result[0].NewIPs)
|
|
}
|
|
|
|
func Test_DecodeTransformTable_1(t *testing.T) {
|
|
result, err := DecodeTransformTable("1.2.3.4~2.3.4.5~3.4.5.6 ~;8.7.6.5 ~ 9.8.7.6 ~ 7.6.5.4 ~ ")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(result) != 2 {
|
|
t.Errorf("Test %v: expected col length (%v), got (%v)\n", 1, 2, len(result))
|
|
}
|
|
testIP(t, "Low[0]", "1.2.3.4", result[0].Low)
|
|
testIP(t, "High[0]", "2.3.4.5", result[0].High)
|
|
testIP(t, "NewBase[0]", "3.4.5.6", result[0].NewBases[0])
|
|
// test_ip(t, "newIP[0]", "", result[0].NewIP)
|
|
testIP(t, "Low[1]", "8.7.6.5", result[1].Low)
|
|
testIP(t, "High[1]", "9.8.7.6", result[1].High)
|
|
testIP(t, "NewBase[1]", "7.6.5.4", result[1].NewBases[0])
|
|
// test_ip(t, "newIP[1]", "", result[0].NewIP)
|
|
}
|
|
func Test_DecodeTransformTable_NewIP(t *testing.T) {
|
|
result, err := DecodeTransformTable("1.2.3.4 ~ 2.3.4.5 ~ ~ 3.4.5.6 ")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(result) != 1 {
|
|
t.Errorf("Test %v: expected col length (%v), got (%v)\n", 1, 1, len(result))
|
|
}
|
|
testIP(t, "low", "1.2.3.4", result[0].Low)
|
|
testIP(t, "high", "2.3.4.5", result[0].High)
|
|
testIP(t, "newIP", "3.4.5.6", result[0].NewIPs[0])
|
|
}
|
|
|
|
func Test_DecodeTransformTable_order(t *testing.T) {
|
|
raw := "9.8.7.6 ~ 8.7.6.5 ~ 7.6.5.4 ~"
|
|
result, err := DecodeTransformTable(raw)
|
|
if result != nil {
|
|
t.Errorf("Invalid range not detected: (%v)\n", raw)
|
|
}
|
|
if err == nil {
|
|
t.Error("expect error, got none")
|
|
}
|
|
}
|
|
|
|
func Test_DecodeTransformTable_Base_and_IP(t *testing.T) {
|
|
raw := "1.1.1.1~ 8.7.6.5 ~ 7.6.5.4 ~ 4.4.4.4"
|
|
result, err := DecodeTransformTable(raw)
|
|
if result != nil {
|
|
t.Errorf("NewBase and NewIP should not both be specified: (%v)\n", raw)
|
|
}
|
|
if err == nil {
|
|
t.Error("expect error, got none")
|
|
}
|
|
}
|
|
|
|
func Test_IP(t *testing.T) {
|
|
|
|
var transforms1 = []IPConversion{{
|
|
Low: net.ParseIP("11.11.11.0"),
|
|
High: net.ParseIP("11.11.11.20"),
|
|
NewBases: []net.IP{net.ParseIP("99.99.99.0")},
|
|
}, {
|
|
Low: net.ParseIP("22.22.22.0"),
|
|
High: net.ParseIP("22.22.22.40"),
|
|
NewBases: []net.IP{net.ParseIP("99.99.99.100")},
|
|
}, {
|
|
Low: net.ParseIP("33.33.33.20"),
|
|
High: net.ParseIP("33.33.35.40"),
|
|
NewBases: []net.IP{net.ParseIP("100.100.100.0")},
|
|
}, {
|
|
Low: net.ParseIP("44.44.44.20"),
|
|
High: net.ParseIP("44.44.44.40"),
|
|
NewBases: []net.IP{net.ParseIP("100.100.100.40")},
|
|
}, {
|
|
Low: net.ParseIP("55.0.0.0"),
|
|
High: net.ParseIP("55.255.0.0"),
|
|
NewBases: []net.IP{net.ParseIP("66.0.0.0"), net.ParseIP("77.0.0.0")},
|
|
}}
|
|
// NO TRANSFORMS ON 99.x.x.x PLZ
|
|
|
|
var tests = []struct {
|
|
experiment string
|
|
expected string
|
|
}{
|
|
{"11.11.11.0", "99.99.99.0"},
|
|
{"11.11.11.1", "99.99.99.1"},
|
|
{"11.11.11.11", "99.99.99.11"},
|
|
{"11.11.11.19", "99.99.99.19"},
|
|
{"11.11.11.20", "99.99.99.20"},
|
|
{"11.11.11.21", "11.11.11.21"},
|
|
{"22.22.22.22", "99.99.99.122"},
|
|
{"22.22.22.255", "22.22.22.255"},
|
|
{"33.33.33.0", "33.33.33.0"},
|
|
{"33.33.33.19", "33.33.33.19"},
|
|
{"33.33.33.20", "100.100.100.0"},
|
|
{"33.33.33.21", "100.100.100.1"},
|
|
{"33.33.33.33", "100.100.100.13"},
|
|
{"33.33.35.39", "100.100.102.19"},
|
|
{"33.33.35.40", "100.100.102.20"},
|
|
{"33.33.35.41", "33.33.35.41"},
|
|
{"44.44.44.24", "100.100.100.44"},
|
|
{"44.44.44.44", "44.44.44.44"},
|
|
{"55.0.42.42", "66.0.42.42,77.0.42.42"},
|
|
{"99.0.0.42", "99.0.0.42"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
experiment := net.ParseIP(test.experiment)
|
|
actual, err := IPToList(experiment, transforms1)
|
|
if err != nil {
|
|
t.Errorf("%v: got an err: %v\n", experiment, err)
|
|
}
|
|
list := []string{}
|
|
for _, ip := range actual {
|
|
list = append(list, ip.String())
|
|
}
|
|
act := strings.Join(list, ",")
|
|
if test.expected != act {
|
|
t.Errorf("%v: expected (%v) got (%v)\n", experiment, test.expected, act)
|
|
}
|
|
}
|
|
}
|
|
|
|
func Test_IP_NewIP(t *testing.T) {
|
|
|
|
var transforms1 = []IPConversion{{
|
|
Low: net.ParseIP("11.11.11.0"),
|
|
High: net.ParseIP("11.11.11.20"),
|
|
NewIPs: []net.IP{net.ParseIP("1.1.1.1")},
|
|
}, {
|
|
Low: net.ParseIP("22.22.22.0"),
|
|
High: net.ParseIP("22.22.22.40"),
|
|
NewIPs: []net.IP{net.ParseIP("2.2.2.2")},
|
|
}, {
|
|
Low: net.ParseIP("33.33.33.20"),
|
|
High: net.ParseIP("33.33.35.40"),
|
|
NewIPs: []net.IP{net.ParseIP("3.3.3.3")},
|
|
},
|
|
}
|
|
|
|
var tests = []struct {
|
|
experiment string
|
|
expected string
|
|
}{
|
|
{"11.11.11.0", "1.1.1.1"},
|
|
{"11.11.11.1", "1.1.1.1"},
|
|
{"11.11.11.11", "1.1.1.1"},
|
|
{"11.11.11.19", "1.1.1.1"},
|
|
{"11.11.11.20", "1.1.1.1"},
|
|
{"11.11.11.21", "11.11.11.21"},
|
|
{"22.22.22.22", "2.2.2.2"},
|
|
{"22.22.22.255", "22.22.22.255"},
|
|
{"33.33.33.0", "33.33.33.0"},
|
|
{"33.33.33.19", "33.33.33.19"},
|
|
{"33.33.33.20", "3.3.3.3"},
|
|
{"33.33.33.21", "3.3.3.3"},
|
|
{"33.33.33.33", "3.3.3.3"},
|
|
{"33.33.35.39", "3.3.3.3"},
|
|
{"33.33.35.40", "3.3.3.3"},
|
|
{"33.33.35.41", "33.33.35.41"},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
experiment := net.ParseIP(test.experiment)
|
|
expected := net.ParseIP(test.expected)
|
|
actual, err := IP(experiment, transforms1)
|
|
if err != nil {
|
|
t.Errorf("%v: got an err: %v\n", experiment, err)
|
|
}
|
|
if !expected.Equal(actual) {
|
|
t.Errorf("%v: expected (%v) got (%v)\n", experiment, expected, actual)
|
|
}
|
|
}
|
|
}
|