Added tests for OptClientId and using net.HardwareAddr for DUID (#58)

1 files changed, 67 insertions, 0 deletions

diff --git a/dhcpv6/option_clientid_test.go b/dhcpv6/option_clientid_test.go
new file mode 100644
index 0000000..dcaffca
--- /dev/null
+++ b/dhcpv6/option_clientid_test.go
@@ -0,0 +1,67 @@
+package dhcpv6
+
+import (
+	"net"
+	"testing"
+
+	"github.com/insomniacslk/dhcp/iana"
+	"github.com/stretchr/testify/require"
+)
+
+func TestParseOptClientId(t *testing.T) {
+	data := []byte{
+		0, 3, // DUID_LL
+		0, 1, // hwtype ethernet
+		0, 1, 2, 3, 4, 5, // hw addr
+	}
+	opt, err := ParseOptClientId(data)
+	require.NoError(t, err)
+	require.Equal(t, opt.Cid.Type, DUID_LL)
+	require.Equal(t, opt.Cid.HwType, iana.HwTypeEthernet)
+	require.Equal(t, opt.Cid.LinkLayerAddr, net.HardwareAddr([]byte{0, 1, 2, 3, 4, 5}))
+}
+
+func TestOptClientIdToBytes(t *testing.T) {
+	opt := OptClientId{
+		Cid: Duid{
+			Type:          DUID_LL,
+			HwType:        iana.HwTypeEthernet,
+			LinkLayerAddr: net.HardwareAddr([]byte{5, 4, 3, 2, 1, 0}),
+		},
+	}
+	expected := []byte{
+		0, 1, // OPTION_CLIENTID
+		0, 10, // length
+		0, 3, // DUID_LL
+		0, 1, // hwtype ethernet
+		5, 4, 3, 2, 1, 0, // hw addr
+	}
+	require.Equal(t, expected, opt.ToBytes())
+}
+
+func TestOptClientIdDecodeEncode(t *testing.T) {
+	data := []byte{
+		0, 3, // DUID_LL
+		0, 1, // hwtype ethernet
+		5, 4, 3, 2, 1, 0, // hw addr
+	}
+	expected := append([]byte{
+		0, 1, // OPTION_CLIENTID
+		0, 10, // length
+	}, data...)
+	opt, err := ParseOptClientId(data)
+	require.NoError(t, err)
+	require.Equal(t, expected, opt.ToBytes())
+}
+
+func TestOptionClientId(t *testing.T) {
+	opt := OptClientId{
+		Cid: Duid{
+			Type:          DUID_LL,
+			HwType:        iana.HwTypeEthernet,
+			LinkLayerAddr: net.HardwareAddr([]byte{0xde, 0xad, 0, 0, 0xbe, 0xef}),
+		},
+	}
+	require.Equal(t, opt.Length(), 10)
+	require.Equal(t, opt.Code(), OPTION_CLIENTID)
+}