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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
package dhcpv4
import (
"testing"
"github.com/insomniacslk/dhcp/iana"
"github.com/insomniacslk/dhcp/rfc1035label"
"github.com/stretchr/testify/require"
)
func TestGetDomainSearch(t *testing.T) {
data := []byte{
7, 'e', 'x', 'a', 'm', 'p', 'l', 'e', 3, 'c', 'o', 'm', 0,
6, 's', 'u', 'b', 'n', 'e', 't', 7, 'e', 'x', 'a', 'm', 'p', 'l', 'e', 3, 'o', 'r', 'g', 0,
}
m, _ := New(WithGeneric(OptionDNSDomainSearchList, data))
labels := m.DomainSearch()
require.NotNil(t, labels)
require.Equal(t, 2, len(labels.Labels))
require.Equal(t, data, labels.ToBytes())
require.Equal(t, labels.Labels[0], "example.com")
require.Equal(t, labels.Labels[1], "subnet.example.org")
}
func TestOptDomainSearchToBytes(t *testing.T) {
expected := []byte{
7, 'e', 'x', 'a', 'm', 'p', 'l', 'e', 3, 'c', 'o', 'm', 0,
6, 's', 'u', 'b', 'n', 'e', 't', 7, 'e', 'x', 'a', 'm', 'p', 'l', 'e', 3, 'o', 'r', 'g', 0,
}
opt := OptDomainSearch(&rfc1035label.Labels{
Labels: []string{
"example.com",
"subnet.example.org",
},
})
require.Equal(t, opt.Value.ToBytes(), expected)
}
func TestParseOptClientArchType(t *testing.T) {
m, _ := New(WithGeneric(OptionClientSystemArchitectureType, []byte{
0, 6, // EFI_IA32
}))
archs := m.ClientArch()
require.NotNil(t, archs)
require.Equal(t, archs[0], iana.EFI_IA32)
}
func TestParseOptClientArchTypeMultiple(t *testing.T) {
m, _ := New(WithGeneric(OptionClientSystemArchitectureType, []byte{
0, 6, // EFI_IA32
0, 2, // EFI_ITANIUM
}))
archs := m.ClientArch()
require.NotNil(t, archs)
require.Equal(t, archs[0], iana.EFI_IA32)
require.Equal(t, archs[1], iana.EFI_ITANIUM)
}
func TestParseOptClientArchTypeInvalid(t *testing.T) {
m, _ := New(WithGeneric(OptionClientSystemArchitectureType, []byte{42}))
archs := m.ClientArch()
require.Nil(t, archs)
}
func TestGetClientArchEmpty(t *testing.T) {
m, _ := New()
require.Nil(t, m.ClientArch())
}
func TestOptClientArchTypeParseAndToBytesMultiple(t *testing.T) {
data := []byte{
0, 6, // EFI_IA32
0, 8, // EFI_XSCALE
}
opt := OptClientArch(iana.EFI_IA32, iana.EFI_XSCALE)
require.Equal(t, opt.Value.ToBytes(), data)
require.Equal(t, opt.Code, OptionClientSystemArchitectureType)
require.Equal(t, opt.String(), "Client System Architecture Type: EFI IA32, EFI Xscale")
}
|
