From 3c5af741416d659f809d442d151f3b795c1271e4 Mon Sep 17 00:00:00 2001
From: konrad <k@knt.li>
Date: Tue, 16 Jan 2018 16:13:17 +0100
Subject: [PATCH] Added unit tests for status

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

diff --git a/models/status_test.go b/models/status_test.go
new file mode 100644
index 0000000..eb44802
--- /dev/null
+++ b/models/status_test.go
@@ -0,0 +1,31 @@
+package models
+
+import (
+	"testing"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGetStatusList(t *testing.T) {
+	// Create test database
+	assert.NoError(t, PrepareTestDatabase())
+
+	// Insert some dummy data
+	_, err := x.Insert(Status{Name: "new"})
+	assert.NoError(t, err)
+	_, err = x.Insert(Status{Name: "used"})
+	assert.NoError(t, err)
+	_, err = x.Insert(Status{Name: "other"})
+	assert.NoError(t, err)
+
+	// Get a status list
+	list, err := GetStatusList()
+	assert.NoError(t, err)
+	assert.Equal(t, "new", list[0].Name)
+	assert.Equal(t, "used", list[1].Name)
+	assert.Equal(t, "other", list[2].Name)
+
+	// Get a status by its ID
+	status, err := GetStatusByID(1)
+	assert.NoError(t, err)
+	assert.Equal(t, "new", status.Name)
+}