devproje/headscale
1
0
Fork 0
mirror of https://github.com/juanfont/headscale.git synced 2025-01-31 14:51:23 +09:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

nextdns mapper unit test

Browse source
This commit is contained in:
Niels Bouma 2025-01-06 10:32:25 +01:00
parent 5e916404aa
commit f870c78e6a
1 changed files with 108 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

108
hscontrol/mapper/mapper_test.go
View file

@ -130,6 +130,114 @@ func TestDNSConfigMapResponse(t *testing.T) {
} }
} }
func TestAddNextDNSMetadata(t *testing.T) {
tests := []struct {
name string
attrs []string
in []*dnstype.Resolver
want []*dnstype.Resolver
}{
{
name: "With NextDNS resolver, without nodeattrs",
attrs: []string{},
in: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/abcdef",
},
},
want: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/abcdef?device_ip=100.64.0.1&device_model=linux&device_name=testnode",
},
},
},
{
name: "With NextDNS resolver, with nodeattrs [nextdns:fedcba]",
attrs: []string{
"nextdns:fedcba",
},
in: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/abcdef",
},
},
want: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/fedcba?device_ip=100.64.0.1&device_model=linux&device_name=testnode",
},
},
},
{
name: "With NextDNS resolver, with nodeattrs [nextdns:no-device-info]",
attrs: []string{
"nextdns:no-device-info",
},
in: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/abcdef",
},
},
want: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/abcdef",
},
},
},
{
name: "With NextDNS resolver, with nodeattrs: [nextdns:fedcba, nextdns:no-device-info]",
attrs: []string{
"nextdns:fedcba",
"nextdns:no-device-info",
},
in: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/abcdef",
},
},
want: []*dnstype.Resolver{
{
Addr: "https://dns.nextdns.io/fedcba",
},
},
},
{
name: "No NextDNS resolver, with nodeattrs: [nextdns:fedcba, nextdns:no-device-info]",
attrs: []string{
"nextdns:fedcba",
"nextdns:no-device-info",
},
in: []*dnstype.Resolver{
{
Addr: "1.1.1.1",
},
},
want: []*dnstype.Resolver{
{
Addr: "1.1.1.1",
},
},
},
}
node := &types.Node{
Hostname: "testnode",
Hostinfo: &tailcfg.Hostinfo{
OS: "linux",
},
IPv4: iap("100.64.0.1"),
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
addNextDNSMetadata(tt.in, node, tt.attrs)
if diff := cmp.Diff(tt.want, tt.in, cmpopts.EquateEmpty()); diff != "" {
t.Errorf("addNextDNSMetadata() unexpected result (-want +got):\n%s", diff)
}
})
}
}
func Test_fullMapResponse(t *testing.T) { func Test_fullMapResponse(t *testing.T) {
mustNK := func(str string) key.NodePublic { mustNK := func(str string) key.NodePublic {
var k key.NodePublic var k key.NodePublic