Added test for encapsulated DHCPv6 message

1 files changed, 71 insertions, 0 deletions

diff --git a/dhcpv6/option_relaymsg_test.go b/dhcpv6/option_relaymsg_test.go
index 95fb9fb..93bab2f 100644
--- a/dhcpv6/option_relaymsg_test.go
+++ b/dhcpv6/option_relaymsg_test.go
@@ -1,6 +1,7 @@
 package dhcpv6
 
 import (
+	"reflect"
 	"testing"
 )
 
@@ -45,3 +46,73 @@ func TestRelayMsgOptionsFromBytes(t *testing.T) {
 		)
 	}
 }
+
+func TestRelayMsgParseOptRelayMsgSingleEncapsulation(t *testing.T) {
+	d, err := FromBytes([]byte{
+		12,                                             // RELAY-FORW
+		0,                                              // hop count
+		0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // linkAddr
+		0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, // peerAddr
+		0, 9, // option: relay message
+		0, 10, // relayed message length
+		1,                // SOLICIT
+		0xaa, 0xbb, 0xcc, // transaction ID
+		0, 8, // option: elapsed time
+		0, 2, // option length
+		0x11, 0x22, // option value
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	r, ok := d.(*DHCPv6Relay)
+	if !ok {
+		t.Fatalf("Invalid DHCPv6 type. Expected DHCPv6Relay, got %v",
+			reflect.TypeOf(d),
+		)
+	}
+	if mType := r.Type(); mType != RELAY_FORW {
+		t.Fatalf("Invalid messge type for relay. Expected %v, got %v", RELAY_FORW, mType)
+	}
+	if len(r.options) != 1 {
+		t.Fatalf("Invalid number of options. Expected 1, got %v", len(r.options))
+	}
+	if code := r.options[0].Code(); code != OPTION_RELAY_MSG {
+		t.Fatalf("Invalid option code. Expected OPTION_RELAY_MSG (%v), got %v",
+			OPTION_RELAY_MSG, code,
+		)
+	}
+	opt := r.options[0]
+	ro, ok := opt.(*OptRelayMsg)
+	if !ok {
+		t.Fatalf("Invalid option type. Expected OptRelayMsg, got %v",
+			reflect.TypeOf(ro),
+		)
+	}
+	innerDHCP, ok := ro.RelayMessage().(*DHCPv6Message)
+	if !ok {
+		t.Fatalf("Invalid relay message type. Expected DHCPv6Message, got %v",
+			reflect.TypeOf(innerDHCP),
+		)
+	}
+	if dType := innerDHCP.Type(); dType != SOLICIT {
+		t.Fatal("Invalid inner DHCP type. Expected SOLICIT (%v), got %v",
+			SOLICIT, dType,
+		)
+	}
+	if tID := innerDHCP.TransactionID(); tID != 0xaabbcc {
+		t.Fatal("Invalid inner DHCP transaction ID. Expected 0xaabbcc, got %v", tID)
+	}
+	if len(innerDHCP.options) != 1 {
+		t.Fatal("Invalid inner DHCP options length. Expected 1, got %v", len(innerDHCP.options))
+	}
+	innerOpt := innerDHCP.options[0]
+	eto, ok := innerOpt.(*OptElapsedTime)
+	if !ok {
+		t.Fatal("Invalid inner option type. Expected OptElapsedTime, got %v",
+			reflect.TypeOf(innerOpt),
+		)
+	}
+	if eTime := eto.ElapsedTime(); eTime != 0x1122 {
+		t.Fatal("Invalid elapsed time. Expected 0x1122, got 0x%04x", eTime)
+	}
+}