From 18b76622665f9e8c81f7ca3002dca0b45ebb1d00 Mon Sep 17 00:00:00 2001
From: Alex Willmer <alex@moreati.org.uk>
Date: Sun, 29 Mar 2020 17:42:47 +0100
Subject: [PATCH] utils: Add test for Packer.CheckSpace

---
 utils/wrappers/packing_test.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/utils/wrappers/packing_test.go b/utils/wrappers/packing_test.go
index 9671b76..8678f46 100644
--- a/utils/wrappers/packing_test.go
+++ b/utils/wrappers/packing_test.go
@@ -17,6 +17,32 @@ const (
 	BoolSentinal  = false
 )
 
+func TestPackerCheckSpace(t *testing.T) {
+	p := Packer{Offset: -1}
+	p.CheckSpace(1)
+	if !p.Errored() {
+		t.Fatal("Expected errNegativeOffset")
+	}
+
+	p = Packer{}
+	p.CheckSpace(-1)
+	if !p.Errored() {
+		t.Fatal("Expected errInvalidInput")
+	}
+
+	p = Packer{Bytes: []byte{0x01}, Offset: 1}
+	p.CheckSpace(1)
+	if !p.Errored() {
+		t.Fatal("Expected errBadLength")
+	}
+
+	p = Packer{Bytes: []byte{0x01}, Offset: 2}
+	p.CheckSpace(0)
+	if !p.Errored() {
+		t.Fatal("Expected errBadLength, due to out of bounds offset")
+	}
+}
+
 func TestPackerPackByte(t *testing.T) {
 	p := Packer{MaxSize: 1}