fix OptUserClass

1 files changed, 14 insertions, 0 deletions

diff --git a/dhcpv4/option_userclass_test.go b/dhcpv4/option_userclass_test.go
index 02b6f94..0e74ee1 100644
--- a/dhcpv4/option_userclass_test.go
+++ b/dhcpv4/option_userclass_test.go
@@ -9,6 +9,7 @@ import (
 func TestOptUserClassToBytes(t *testing.T) {
 	opt := OptUserClass{
 		UserClasses: [][]byte{[]byte("linuxboot")},
+		Rfc3004: true,
 	}
 	data := opt.ToBytes()
 	expected := []byte{
@@ -19,6 +20,19 @@ func TestOptUserClassToBytes(t *testing.T) {
 	require.Equal(t, expected, data)
 }
 
+func TestOptUserClassMicrosoftToBytes(t *testing.T) {
+	opt := OptUserClass{
+		UserClasses: [][]byte{[]byte("linuxboot")},
+	}
+	data := opt.ToBytes()
+	expected := []byte{
+		77, // OptionUserClass
+		9,  // length
+		'l', 'i', 'n', 'u', 'x', 'b', 'o', 'o', 't',
+	}
+	require.Equal(t, expected, data)
+}
+
 func TestParseOptUserClassMultiple(t *testing.T) {
 	expected := []byte{
 		77, 15,