Add specific BSDP options

1 files changed, 130 insertions, 0 deletions

diff --git a/dhcpv4/bsdp/boot_image_test.go b/dhcpv4/bsdp/boot_image_test.go
new file mode 100644
index 0000000..ebf02c3
--- /dev/null
+++ b/dhcpv4/bsdp/boot_image_test.go
@@ -0,0 +1,130 @@
+// +build darwin
+
+package bsdp
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+/*
+ * BootImageID
+ */
+func TestBootImageIDToBytes(t *testing.T) {
+	b := BootImageID{
+		IsInstall: true,
+		ImageType: BootImageTypeMacOSX,
+		Index:     0x1000,
+	}
+	actual := b.ToBytes()
+	expected := []byte{0x81, 0, 0x10, 0}
+	require.Equal(t, expected, actual)
+
+	b.IsInstall = false
+	actual = b.ToBytes()
+	expected = []byte{0x01, 0, 0x10, 0}
+	require.Equal(t, expected, actual)
+}
+
+func TestBootImageIDFromBytes(t *testing.T) {
+	b := BootImageID{
+		IsInstall: false,
+		ImageType: BootImageTypeMacOSX,
+		Index:     0x1000,
+	}
+	newBootImage, err := BootImageIDFromBytes(b.ToBytes())
+	require.NoError(t, err)
+	require.Equal(t, b, *newBootImage)
+
+	b = BootImageID{
+		IsInstall: true,
+		ImageType: BootImageTypeMacOSX,
+		Index:     0x1011,
+	}
+	newBootImage, err = BootImageIDFromBytes(b.ToBytes())
+	require.NoError(t, err)
+	require.Equal(t, b, *newBootImage)
+}
+
+func TestBootImageIDFromBytesFail(t *testing.T) {
+	serialized := []byte{0x81, 0, 0x10} // intentionally left short
+	deserialized, err := BootImageIDFromBytes(serialized)
+	require.Nil(t, deserialized)
+	require.Error(t, err)
+}
+
+/*
+ * BootImage
+ */
+func TestBootImageToBytes(t *testing.T) {
+	b := BootImage{
+		ID: BootImageID{
+			IsInstall: true,
+			ImageType: BootImageTypeMacOSX,
+			Index:     0x1000,
+		},
+		Name: "bsdp-1",
+	}
+	expected := []byte{
+		0x81, 0, 0x10, 0, // boot image ID
+		6,                         // len(Name)
+		98, 115, 100, 112, 45, 49, // byte-encoding of Name
+	}
+	actual := b.ToBytes()
+	require.Equal(t, expected, actual)
+
+	b = BootImage{
+		ID: BootImageID{
+			IsInstall: false,
+			ImageType: BootImageTypeMacOSX,
+			Index:     0x1010,
+		},
+		Name: "bsdp-21",
+	}
+	expected = []byte{
+		0x1, 0, 0x10, 0x10, // boot image ID
+		7,                             // len(Name)
+		98, 115, 100, 112, 45, 50, 49, // byte-encoding of Name
+	}
+	actual = b.ToBytes()
+	require.Equal(t, expected, actual)
+}
+
+func TestBootImageFromBytes(t *testing.T) {
+	input := []byte{
+		0x1, 0, 0x10, 0x10, // boot image ID
+		7,                             // len(Name)
+		98, 115, 100, 112, 45, 50, 49, // byte-encoding of Name
+	}
+	b, err := BootImageFromBytes(input)
+	require.NoError(t, err)
+	expectedBootImage := BootImage{
+		ID: BootImageID{
+			IsInstall: false,
+			ImageType: BootImageTypeMacOSX,
+			Index:     0x1010,
+		},
+		Name: "bsdp-21",
+	}
+	require.Equal(t, expectedBootImage, *b)
+}
+
+func TestBootImageFromBytesOnlyBootImageID(t *testing.T) {
+	// Only a BootImageID, nothing else.
+	input := []byte{0x1, 0, 0x10, 0x10}
+	b, err := BootImageFromBytes(input)
+	require.Nil(t, b)
+	require.Error(t, err)
+}
+
+func TestBootImageFromBytesShortBootImage(t *testing.T) {
+	input := []byte{
+		0x1, 0, 0x10, 0x10, // boot image ID
+		7,                         // len(Name)
+		98, 115, 100, 112, 45, 50, // Name bytes (intentionally off-by-one)
+	}
+	b, err := BootImageFromBytes(input)
+	require.Nil(t, b)
+	require.Error(t, err)
+}