m5r
abb97cce56
- replace global hardcodedRecords/mutex with instance fields - add initialRecords() factory for fresh record copies per instance - rename initHardcodedRecords to initNameServers, pass nameservers explicitly - add TestInstanceIsolation to verify instances don't share state - fix unassigned err in certs/persistFiles
132 lines
3.1 KiB
Go
132 lines
3.1 KiB
Go
package xip
|
|
|
|
import (
|
|
"fmt"
|
|
"os/exec"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestResolveDashUnit(t *testing.T) {
|
|
xip := NewXip(
|
|
WithDomain("local-ip.sh"),
|
|
WithDnsPort(9053),
|
|
WithNameServers([]string{"1.2.3.4", "5.6.7.8"}),
|
|
)
|
|
|
|
A := xip.fqdnToA("192-168-1-29.local-ip.sh")
|
|
expected := "192.168.1.29"
|
|
received := A[0].A.String()
|
|
if received != expected {
|
|
t.Fatalf("Expected %s but received %s", expected, received)
|
|
}
|
|
|
|
A = xip.fqdnToA("192.168.1.29.local-ip.sh")
|
|
expected = "192.168.1.29"
|
|
received = A[0].A.String()
|
|
if received != expected {
|
|
t.Fatalf("Expected %s but received %s", expected, received)
|
|
}
|
|
|
|
A = xip.fqdnToA("prefixed.192.168.1.29.local-ip.sh")
|
|
expected = "192.168.1.29"
|
|
received = A[0].A.String()
|
|
if received != expected {
|
|
t.Fatalf("Expected %s but received %s", expected, received)
|
|
}
|
|
|
|
A = xip.fqdnToA("prefixed-192.168.1.29.local-ip.sh")
|
|
if A != nil {
|
|
t.Fatalf("Expected %v but received %s", nil, A)
|
|
}
|
|
}
|
|
|
|
func TestConstructor(t *testing.T) {
|
|
xip := NewXip(
|
|
WithDomain("local-ip.sh"),
|
|
WithDnsPort(9053),
|
|
WithNameServers([]string{"1.2.3.4", "5.6.7.8"}),
|
|
)
|
|
|
|
if xip.nameServers[0] != "ns1.local-ip.sh." {
|
|
t.Errorf("expected ns1.local-ip.sh., got %s", xip.nameServers[0])
|
|
}
|
|
if xip.nameServers[1] != "ns2.local-ip.sh." {
|
|
t.Errorf("expected ns2.local-ip.sh., got %s", xip.nameServers[1])
|
|
}
|
|
}
|
|
|
|
func TestResolveDashE2E(t *testing.T) {
|
|
xip := NewXip(
|
|
WithDomain("local-ip.sh"),
|
|
WithDnsPort(9053),
|
|
WithNameServers([]string{"1.2.3.4", "5.6.7.8"}),
|
|
)
|
|
go xip.StartServer()
|
|
|
|
cmd := exec.Command("dig", "@localhost", "-p", "9053", "192-168-1-29.local-ip.sh", "+short")
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if strings.TrimSpace(string(out)) != "192.168.1.29" {
|
|
t.Fatal(string(out))
|
|
}
|
|
}
|
|
|
|
func BenchmarkResolveDashBasic(b *testing.B) {
|
|
b.Skip()
|
|
|
|
for i := 0; i < b.N; i++ {
|
|
port := 9053 + i
|
|
xip := NewXip(
|
|
WithDomain("local-ip.sh"),
|
|
WithDnsPort(uint(port)),
|
|
WithNameServers([]string{"1.2.3.4", "5.6.7.8"}),
|
|
)
|
|
go xip.StartServer()
|
|
|
|
cmd := exec.Command("dig", "@localhost", "-p", fmt.Sprint(port), "192-168-1-29.local-ip.sh", "+short")
|
|
cmd.Run()
|
|
}
|
|
}
|
|
|
|
func TestInstanceIsolation(t *testing.T) {
|
|
xip1 := NewXip(
|
|
WithDomain("one.test"),
|
|
WithDnsPort(0),
|
|
WithNameServers([]string{"1.1.1.1"}),
|
|
)
|
|
xip2 := NewXip(
|
|
WithDomain("two.test"),
|
|
WithDnsPort(0),
|
|
WithNameServers([]string{"2.2.2.2"}),
|
|
)
|
|
|
|
if xip1.records == nil || xip2.records == nil {
|
|
t.Fatal("records not initialized")
|
|
}
|
|
|
|
if len(xip1.nameServers) != 1 || xip1.nameServers[0] != "ns1.one.test." {
|
|
t.Errorf("xip1 nameservers incorrect: %v", xip1.nameServers)
|
|
}
|
|
if len(xip2.nameServers) != 1 || xip2.nameServers[0] != "ns1.two.test." {
|
|
t.Errorf("xip2 nameservers incorrect: %v", xip2.nameServers)
|
|
}
|
|
|
|
if _, ok := xip1.records["ns1.one.test."]; !ok {
|
|
t.Error("xip1 missing ns1.one.test. record")
|
|
}
|
|
if _, ok := xip2.records["ns1.two.test."]; !ok {
|
|
t.Error("xip2 missing ns1.two.test. record")
|
|
}
|
|
|
|
if _, ok := xip1.records["ns1.two.test."]; ok {
|
|
t.Error("xip1 should not have xip2's records")
|
|
}
|
|
if _, ok := xip2.records["ns1.one.test."]; ok {
|
|
t.Error("xip2 should not have xip1's records")
|
|
}
|
|
}
|