summaryrefslogtreecommitdiffhomepage
path: root/dhcpv4/option_string_test.go
blob: 9778134a34492e67980c15ad0423927018a0c4dc (plain)
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package dhcpv4

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func TestOptDomainName(t *testing.T) {
	o := OptDomainName("foo")
	require.Equal(t, OptionDomainName, o.Code, "Code")
	require.Equal(t, []byte{'f', 'o', 'o'}, o.Value.ToBytes(), "ToBytes")
	require.Equal(t, "Domain Name: foo", o.String())
}

func TestParseOptDomainName(t *testing.T) {
	m, _ := New(WithGeneric(OptionDomainName, []byte{'t', 'e', 's', 't'}))
	require.Equal(t, "test", m.DomainName())

	m, _ = New()
	require.Equal(t, "", m.DomainName())
}

func TestOptHostName(t *testing.T) {
	o := OptHostName("foo")
	require.Equal(t, OptionHostName, o.Code, "Code")
	require.Equal(t, []byte{'f', 'o', 'o'}, o.Value.ToBytes(), "ToBytes")
	require.Equal(t, "Host Name: foo", o.String())
}

func TestParseOptHostName(t *testing.T) {
	m, _ := New(WithGeneric(OptionHostName, []byte{'t', 'e', 's', 't'}))
	require.Equal(t, "test", m.HostName())

	m, _ = New()
	require.Equal(t, "", m.HostName())
}

func TestOptRootPath(t *testing.T) {
	o := OptRootPath("foo")
	require.Equal(t, OptionRootPath, o.Code, "Code")
	require.Equal(t, []byte{'f', 'o', 'o'}, o.Value.ToBytes(), "ToBytes")
	require.Equal(t, "Root Path: foo", o.String())
}

func TestParseOptRootPath(t *testing.T) {
	m, _ := New(WithGeneric(OptionRootPath, []byte{'t', 'e', 's', 't'}))
	require.Equal(t, "test", m.RootPath())

	m, _ = New()
	require.Equal(t, "", m.RootPath())
}

func TestOptBootFileName(t *testing.T) {
	o := OptBootFileName("foo")
	require.Equal(t, OptionBootfileName, o.Code, "Code")
	require.Equal(t, []byte{'f', 'o', 'o'}, o.Value.ToBytes(), "ToBytes")
	require.Equal(t, "Bootfile Name: foo", o.String())
}

func TestParseOptBootFileName(t *testing.T) {
	m, _ := New(WithGeneric(OptionBootfileName, []byte{'t', 'e', 's', 't'}))
	require.Equal(t, "test", m.BootFileNameOption())

	m, _ = New()
	require.Equal(t, "", m.BootFileNameOption())

	m, _ = New(WithGeneric(OptionBootfileName, []byte{'t', 'e', 's', 't', 0}))
	require.Equal(t, "test", m.BootFileNameOption())
}

func TestOptTFTPServerName(t *testing.T) {
	o := OptTFTPServerName("foo")
	require.Equal(t, OptionTFTPServerName, o.Code, "Code")
	require.Equal(t, []byte{'f', 'o', 'o'}, o.Value.ToBytes(), "ToBytes")
	require.Equal(t, "TFTP Server Name: foo", o.String())
}

func TestParseOptTFTPServerName(t *testing.T) {
	m, _ := New(WithGeneric(OptionTFTPServerName, []byte{'t', 'e', 's', 't'}))
	require.Equal(t, "test", m.TFTPServerName())

	m, _ = New()
	require.Equal(t, "", m.TFTPServerName())

	m, _ = New(WithGeneric(OptionTFTPServerName, []byte{'t', 'e', 's', 't', 0}))
	require.Equal(t, "test", m.TFTPServerName())
}

func TestOptClassIdentifier(t *testing.T) {
	o := OptClassIdentifier("foo")
	require.Equal(t, OptionClassIdentifier, o.Code, "Code")
	require.Equal(t, []byte{'f', 'o', 'o'}, o.Value.ToBytes(), "ToBytes")
	require.Equal(t, "Class Identifier: foo", o.String())
}

func TestParseOptClassIdentifier(t *testing.T) {
	m, _ := New(WithGeneric(OptionClassIdentifier, []byte{'t', 'e', 's', 't'}))
	require.Equal(t, "test", m.ClassIdentifier())

	m, _ = New()
	require.Equal(t, "", m.ClassIdentifier())
}

func TestOptUserClass(t *testing.T) {
	o := OptUserClass("linuxboot")
	require.Equal(t, OptionUserClassInformation, o.Code, "Code")
	expected := []byte{
		'l', 'i', 'n', 'u', 'x', 'b', 'o', 'o', 't',
	}
	require.Equal(t, expected, o.Value.ToBytes(), "ToBytes")
	require.Equal(t, "User Class Information: linuxboot", o.String())
}

func TestParseOptUserClass(t *testing.T) {
	m, _ := New(WithUserClass("linuxboot", false))
	require.Equal(t, []string{"linuxboot"}, m.UserClass())

	m, _ = New()
	require.Equal(t, 0, len(m.UserClass()))
}