mirrors/netboot
1
0
Fork 0
mirror of https://github.com/danderson/netboot.git synced 2026-05-05 20:36:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

added options tests

Browse Source
This commit is contained in:
Dmitri Dolguikh 2017-09-28 15:11:28 -07:00 committed by Dave Anderson
parent ffff3b81ba
commit b6741d78b8
1 changed files with 73 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

96
dhcp6/options_test.go
View File

@ -2,8 +2,81 @@ package dhcp6
import (
"testing"
"encoding/binary"
"net"
)
func TestMarshalOption(t *testing.T) {
expectedUrl := []byte("http://blah")
expectedLength := uint16(len(expectedUrl))
opt := &Option{Id: OptBootfileUrl, Length: expectedLength, Value: expectedUrl}
marshalled, err := opt.Marshal()
if err != nil {
t.Fatalf("Unexpected marshalling failure.")
}
if id := uint16(binary.BigEndian.Uint16(marshalled[0:2])); id != OptBootfileUrl {
t.Fatalf("Expected optionId %d, got: %d", OptBootfileUrl, id)
}
if l := uint16(binary.BigEndian.Uint16(marshalled[2:4])); l != expectedLength {
t.Fatalf("Expected length of %d, got: %d", expectedLength, l)
}
if url := marshalled[4:]; string(url) != string(expectedUrl) {
t.Fatalf("Expected %s, got %s", expectedUrl, url)
}
}
func TestMakeIaAddrOption(t *testing.T) {
expectedIp := net.ParseIP("2001:db8:f00f:cafe::99")
var expectedPreferredLifetime, expectedValidLifetime uint32 = 27000, 43200
iaAddrOption := MakeIaAddrOption(expectedIp, expectedPreferredLifetime, expectedValidLifetime)
if iaAddrOption.Id != OptIaAddr {
t.Fatalf("Expected option id %d, got %d", OptIaAddr, iaAddrOption.Id)
}
if iaAddrOption.Length != 24 {
t.Fatalf("Expected length 24 bytes, got %d", iaAddrOption.Length)
}
if string(iaAddrOption.Value[0:16]) != string(expectedIp) {
t.Fatalf("Expected address %v, got %v", expectedIp, iaAddrOption.Value[0:16])
}
if preferredLifetime := uint32(binary.BigEndian.Uint32(iaAddrOption.Value[16:20])); preferredLifetime != expectedPreferredLifetime {
t.Fatalf("Expected preferred lifetime of %d, got %d", expectedPreferredLifetime, preferredLifetime)
}
if validLifetime := uint32(binary.BigEndian.Uint32(iaAddrOption.Value[20:24])); validLifetime != expectedValidLifetime {
t.Fatalf("Expected valid lifetime of %d, got %d", expectedValidLifetime, validLifetime)
}
}
func TestMakeIaNaOption(t *testing.T) {
iaAddrOption := MakeIaAddrOption(net.ParseIP("2001:db8:f00f:cafe::100"), 100, 200)
expectedSerializedIaAddrOption, err := iaAddrOption.Marshal()
if err != nil {
t.Fatalf("Unexpected serialization error: %s", err)
}
expectedId := []byte("1234")
var expectedT1, expectedT2 uint32 = 100, 200
iaNaOption := MakeIaNaOption(expectedId, expectedT1, expectedT2, iaAddrOption)
if iaNaOption.Id != OptIaNa {
t.Fatalf("Expected optionId %d, got %d", OptIaNa, iaNaOption.Id)
}
if string(iaNaOption.Value[0:4]) != string(expectedId) {
t.Fatalf("Expected id %s, got %s", expectedId, string(iaNaOption.Value[0:4]))
}
if t1 := uint32(binary.BigEndian.Uint32(iaNaOption.Value[4:])); t1 != expectedT1 {
t.Fatalf("Expected t1 of %d, got %d", expectedT1, t1)
}
if t2 := uint32(binary.BigEndian.Uint32(iaNaOption.Value[8:])); t2 != expectedT2 {
t.Fatalf("Expected t2 of %d, got %d", expectedT2, t2)
}
if serializedIaAddrOption := iaNaOption.Value[12:]; string(serializedIaAddrOption) != string(expectedSerializedIaAddrOption) {
t.Fatalf("Expected serialized ia addr option %v, got %v", expectedSerializedIaAddrOption, serializedIaAddrOption)
}
}
func TestUnmarshalFailsIfOROLengthIsOdd(t *testing.T) {
@ -12,26 +85,3 @@ func TestUnmarshalFailsIfOROLengthIsOdd(t *testing.T) {
t.Fatalf("Parsing options should fail: option request for options has odd length.")
}
}
func TestUnmarshalORO(t *testing.T) {
options := Options{
6: [][]byte{{0, 1, 0, 5, 1, 0}},
}
parsed_options := options.UnmarshalOptionRequestOption()
if l := len(parsed_options); l != 3 {
t.Fatalf("Expected 3 options, got: %d", l)
}
if _, present := parsed_options[1]; !present {
t.Fatalf("Should contain option id 1")
}
if _, present := parsed_options[5]; !present {
t.Fatalf("Should contain option id 5")
}
if _, present := parsed_options[256]; !present {
t.Fatalf("Should contain option id 256")
}
}