mirror of
https://github.com/danderson/netboot.git
synced 2025-08-18 21:21:20 +02:00
88 lines
3.2 KiB
Go
88 lines
3.2 KiB
Go
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) {
|
|
in := []byte{0, 6, 0, 3, 0, 1, 1}
|
|
if _, err := MakeOptions(in); err == nil {
|
|
t.Fatalf("Parsing options should fail: option request for options has odd length.")
|
|
}
|
|
}
|