mirror of
https://github.com/Control-D-Inc/ctrld.git
synced 2026-02-03 22:18:39 +00:00
78 lines
2.3 KiB
Go
78 lines
2.3 KiB
Go
package clientinfo
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func Test_hostsFile_LookupHostnameByIP(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
ip string
|
|
hostnames []string
|
|
expectedHostname string
|
|
}{
|
|
{"ipv4 loopback", "127.0.0.1", []string{ipv4LocalhostName}, ""},
|
|
{"ipv6 loopback", "::1", []string{ipv6LocalhostName, ipv6LoopbackName}, ""},
|
|
{"non-localhost", "::1", []string{"foo"}, "foo"},
|
|
{"multiple hostnames", "::1", []string{ipv4LocalhostName, "foo"}, "foo"},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
tc := tc
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
hf := &hostsFile{m: make(map[string][]string)}
|
|
hf.mu.Lock()
|
|
hf.m[tc.ip] = tc.hostnames
|
|
hf.mu.Unlock()
|
|
if got := hf.LookupHostnameByIP(tc.ip); got != tc.expectedHostname {
|
|
t.Errorf("unpexpected result, want: %q, got: %q", tc.expectedHostname, got)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func Test_parseHostEntriesConfFromReader(t *testing.T) {
|
|
const content = `local-zone: "localdomain" transparent
|
|
local-data-ptr: "127.0.0.1 localhost"
|
|
local-data: "localhost A 127.0.0.1"
|
|
local-data: "localhost.localdomain A 127.0.0.1"
|
|
local-data-ptr: "::1 localhost"
|
|
local-data: "localhost AAAA ::1"
|
|
local-data: "localhost.localdomain AAAA ::1"
|
|
local-data-ptr: "10.0.10.227 OPNsense.localdomain"
|
|
local-data: "OPNsense.localdomain A 10.0.10.227"
|
|
local-data: "OPNsense A 10.0.10.227"
|
|
local-data-ptr: "fe80::5a78:4e29:caa3:f9f7 OPNsense.localdomain"
|
|
local-data: "OPNsense.localdomain AAAA fe80::5a78:4e29:caa3:f9f7"
|
|
local-data: "OPNsense AAAA fe80::5a78:4e29:caa3:f9f7"
|
|
local-data-ptr: "1.1.1.1 banana-party.local.com"
|
|
local-data: "banana-party.local.com IN A 1.1.1.1"
|
|
local-data-ptr: "1.1.1.1 cheese-land.lan"
|
|
local-data: "cheese-land.lan IN A 1.1.1.1"
|
|
`
|
|
r := strings.NewReader(content)
|
|
hostsMap := parseHostEntriesConfFromReader(r)
|
|
if len(hostsMap) != 5 {
|
|
t.Fatalf("unexpected number of entries, want 5, got: %d", len(hostsMap))
|
|
}
|
|
for ip, names := range hostsMap {
|
|
switch ip {
|
|
case "1.1.1.1":
|
|
for _, name := range names {
|
|
if name != "banana-party.local.com" && name != "cheese-land.lan" {
|
|
t.Fatalf("unexpected names for 1.1.1.1: %v", names)
|
|
}
|
|
}
|
|
case "10.0.10.227":
|
|
if len(names) != 1 {
|
|
t.Fatalf("unexpected names for 10.0.10.227: %v", names)
|
|
}
|
|
if names[0] != "OPNsense" {
|
|
t.Fatalf("unexpected name: %s", names[0])
|
|
}
|
|
}
|
|
}
|
|
}
|