Add unit tests for models

1 files changed, 19 insertions, 0 deletions

diff --git a/model/theme_test.go b/model/theme_test.go
new file mode 100644
index 0000000..b2a8cdb
--- /dev/null
+++ b/model/theme_test.go
@@ -0,0 +1,19 @@
+// 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 TestValidateTheme(t *testing.T) {
+	for _, status := range []string{"default", "black"} {
+		if err := ValidateTheme(status); err != nil {
+			t.Error(`A valid theme should not generate any error`)
+		}
+	}
+
+	if err := ValidateTheme("invalid"); err == nil {
+		t.Error(`An invalid theme should generate a error`)
+	}
+}