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 122 123 124
|
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())
m, _ = New(WithGeneric(OptionHostName, []byte{'t', 'e', 's', 't', 0}))
require.Equal(t, "test", 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()))
}
|