1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
package dhcpv6
import (
"net"
"testing"
"github.com/stretchr/testify/require"
)
func TestParseOptDNS(t *testing.T) {
data := []byte{
0x2a, 0x03, 0x28, 0x80, 0xff, 0xfe, 0x00, 0x0c, 0xfa, 0xce, 0xb0, 0x0c, 0x00, 0x00, 0x00, 0x35,
}
expected := []net.IP{
net.IP(data),
}
opt, err := parseOptDNS(data)
require.NoError(t, err)
require.Equal(t, expected, opt.NameServers)
require.Equal(t, OptionDNSRecursiveNameServer, opt.Code())
require.Contains(t, opt.String(), "DNS: [2a03:2880:fffe:c:face:b00c:0:35]", "String() should contain the correct nameservers output")
}
func TestOptDNSRecursiveNameServerToBytes(t *testing.T) {
ns1 := net.ParseIP("2a03:2880:fffe:c:face:b00c:0:35")
ns2 := net.ParseIP("2001:4860:4860::8888")
nameservers := []net.IP{ns1, ns2}
expected := append([]byte{}, []byte(ns1)...)
expected = append(expected, []byte(ns2)...)
opt := OptDNS(nameservers...)
require.Equal(t, expected, opt.ToBytes())
}
func TestParseOptDNSBogus(t *testing.T) {
data := []byte{
0x2a, 0x03, 0x28, 0x80, 0xff, 0xfe, 0x00, 0x0c, // invalid IPv6 address
}
_, err := parseOptDNS(data)
require.Error(t, err, "An invalid nameserver IPv6 address should return an error")
}
|
