From b613208950453ce42c5c1e4e4e7ad16774d96e10 Mon Sep 17 00:00:00 2001
From: konrad <k@knt.li>
Date: Tue, 16 Jan 2018 16:07:21 +0100
Subject: [PATCH] Added unit tests for quantity itself

---
 models/quantity_test.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)
 create mode 100644 models/quantity_test.go

diff --git a/models/quantity_test.go b/models/quantity_test.go
new file mode 100644
index 0000000..5b8618f
--- /dev/null
+++ b/models/quantity_test.go
@@ -0,0 +1,29 @@
+package models
+
+import (
+	"testing"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestSetQuantity(t *testing.T) {
+	// Create test database
+	assert.NoError(t, PrepareTestDatabase())
+
+	// Set Quantity for a nonexistent item (should create)
+	err := SetQuantity(9999, 12)
+	assert.NoError(t, err)
+
+	// Check
+	qty, err := GetQuantity(9999)
+	assert.NoError(t, err)
+	assert.Equal(t, int64(12), qty)
+
+	// Update and check again
+	err = SetQuantity(9999, 120)
+	assert.NoError(t, err)
+
+	// Check
+	qty, err = GetQuantity(9999)
+	assert.NoError(t, err)
+	assert.Equal(t, int64(120), qty)
+}