1 files changed, 64 insertions, 0 deletions
diff --git a/dhcpv4/option_vivc_test.go b/dhcpv4/option_vivc_test.go
new file mode 100644
index 0000000..14ded61
--- /dev/null
+++ b/dhcpv4/option_vivc_test.go
@@ -0,0 +1,64 @@
+package dhcpv4
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+var (
+	sampleVIVCOpt = OptVIVC{
+		Identifiers: []VIVCIdentifier{
+			{EntID: 9, Data: []byte("CiscoIdentifier")},
+			{EntID: 18, Data: []byte("WellfleetIdentifier")},
+		},
+	}
+	sampleVIVCOptRaw = []byte{
+		byte(OptionVendorIdentifyingVendorClass), 44, // option header
+		0x0, 0x0, 0x0, 0x9, // enterprise id 9
+		0xf, // length
+		'C', 'i', 's', 'c', 'o', 'I', 'd', 'e', 'n', 't', 'i', 'f', 'i', 'e', 'r',
+		0x0, 0x0, 0x0, 0x12, // enterprise id 18
+		0x13, // length
+		'W', 'e', 'l', 'l', 'f', 'l', 'e', 'e', 't', 'I', 'd', 'e', 'n', 't', 'i', 'f', 'i', 'e', 'r',
+	}
+)
+
+func TestOptVIVCInterfaceMethods(t *testing.T) {
+	require.Equal(t, OptionVendorIdentifyingVendorClass, sampleVIVCOpt.Code(), "Code")
+	require.Equal(t, 44, sampleVIVCOpt.Length(), "Length")
+	require.Equal(t, sampleVIVCOptRaw, sampleVIVCOpt.ToBytes(), "ToBytes")
+}
+
+func TestParseOptVICO(t *testing.T) {
+	o, err := ParseOptVIVC(sampleVIVCOptRaw)
+	require.NoError(t, err)
+	require.Equal(t, &sampleVIVCOpt, o)
+
+	// Short byte stream
+	data := []byte{byte(OptionVendorIdentifyingVendorClass)}
+	_, err = ParseOptVIVC(data)
+	require.Error(t, err, "should get error from short byte stream")
+
+	// Wrong code
+	data = []byte{54, 2, 1, 1}
+	_, err = ParseOptVIVC(data)
+	require.Error(t, err, "should get error from wrong code")
+
+	// Bad length
+	data = []byte{byte(OptionVendorIdentifyingVendorClass), 6, 1, 1, 1}
+	_, err = ParseOptVIVC(data)
+	require.Error(t, err, "should get error from bad length")
+
+	// Identifier len too long
+	data = make([]byte, len(sampleVIVCOptRaw))
+	copy(data, sampleVIVCOptRaw)
+	data[6] = 40
+	_, err = ParseOptVIVC(data)
+	require.Error(t, err, "should get error from bad length")
+}
+
+func TestOptVIVCString(t *testing.T) {
+	require.Equal(t, "Vendor-Identifying Vendor Class -> 9:'CiscoIdentifier', 18:'WellfleetIdentifier'",
+		sampleVIVCOpt.String())
+}