From 3822e4a09e7d5ba4d3420bf39424da3cf4dced7c Mon Sep 17 00:00:00 2001
From: konrad <k@knt.li>
Date: Tue, 16 Jan 2018 16:01:42 +0100
Subject: [PATCH] Added unit test for item quantities

---
 models/item_test.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/models/item_test.go b/models/item_test.go
index b580243..20bdcb7 100644
--- a/models/item_test.go
+++ b/models/item_test.go
@@ -64,6 +64,19 @@ func TestAddOrUpdateItem(t *testing.T) {
 	assert.Equal(t, testitem.Description, item1updated.Description)
 	assert.Equal(t, testitem.Other, item1updated.Other)
 
+	// Test Quantity
+	qty1, err := item1.getQuantity()
+	assert.NoError(t, err)
+	assert.Equal(t, item1.Quantity, qty1)
+
+	// Update the quantity and check again
+	err = item1.setQuantity(int64(99))
+	assert.NoError(t, err)
+
+	qty2, err := item1.getQuantity()
+	assert.NoError(t, err)
+	assert.Equal(t, int64(99), qty2)
+
 	// Delete the item
 	err = DeleteItemByID(item1.ID)
 	assert.NoError(t, err)