Add unit tests for models

1 files changed, 57 insertions, 0 deletions

diff --git a/model/entry_test.go b/model/entry_test.go
new file mode 100644
index 0000000..3f2e196
--- /dev/null
+++ b/model/entry_test.go
@@ -0,0 +1,57 @@
+// Copyright 2017 Frédéric Guillot. All rights reserved.
+// Use of this source code is governed by the Apache 2.0
+// license that can be found in the LICENSE file.
+
+package model
+
+import "testing"
+
+func TestValidateEntryStatus(t *testing.T) {
+	for _, status := range []string{EntryStatusRead, EntryStatusUnread, EntryStatusRemoved} {
+		if err := ValidateEntryStatus(status); err != nil {
+			t.Error(`A valid status should not generate any error`)
+		}
+	}
+
+	if err := ValidateEntryStatus("invalid"); err == nil {
+		t.Error(`An invalid status should generate a error`)
+	}
+}
+
+func TestValidateEntryOrder(t *testing.T) {
+	for _, status := range []string{"id", "status", "published_at", "category_title", "category_id"} {
+		if err := ValidateEntryOrder(status); err != nil {
+			t.Error(`A valid order should not generate any error`)
+		}
+	}
+
+	if err := ValidateEntryOrder("invalid"); err == nil {
+		t.Error(`An invalid order should generate a error`)
+	}
+}
+
+func TestValidateEntryDirection(t *testing.T) {
+	for _, status := range []string{"asc", "desc"} {
+		if err := ValidateDirection(status); err != nil {
+			t.Error(`A valid direction should not generate any error`)
+		}
+	}
+
+	if err := ValidateDirection("invalid"); err == nil {
+		t.Error(`An invalid direction should generate a error`)
+	}
+}
+
+func TestGetOppositeDirection(t *testing.T) {
+	if GetOppositeDirection("asc") != "desc" {
+		t.Errorf(`The opposite direction of "asc" should be "desc"`)
+	}
+
+	if GetOppositeDirection("desc") != "asc" {
+		t.Errorf(`The opposite direction of "desc" should be "asc"`)
+	}
+
+	if GetOppositeDirection("invalid") != "asc" {
+		t.Errorf(`An invalid direction should return "asc"`)
+	}
+}