From 9aa8a590b74e5260df649c303d9b03d52cbb32bb Mon Sep 17 00:00:00 2001
From: Vincent Demeester <vincent@sbr.pm>
Date: Fri, 29 May 2015 15:09:35 +0200
Subject: [PATCH] Add test coverage to pkg/timeutils

Signed-off-by: Vincent Demeester <vincent@sbr.pm>
---
 pkg/timeutils/json_test.go | 47 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 pkg/timeutils/json_test.go

diff --git a/pkg/timeutils/json_test.go b/pkg/timeutils/json_test.go
new file mode 100644
index 0000000000..1ff3331797
--- /dev/null
+++ b/pkg/timeutils/json_test.go
@@ -0,0 +1,47 @@
+package timeutils
+
+import (
+	"testing"
+	"time"
+)
+
+// Testing to ensure 'year' fields is between 0 and 9999
+func TestFastMarshalJSONWithInvalidDate(t *testing.T) {
+	aTime := time.Date(-1, 1, 1, 0, 0, 0, 0, time.Local)
+	json, err := FastMarshalJSON(aTime)
+	if err == nil {
+		t.Fatalf("FastMarshalJSON should throw an error, but was '%v'", json)
+	}
+	anotherTime := time.Date(10000, 1, 1, 0, 0, 0, 0, time.Local)
+	json, err = FastMarshalJSON(anotherTime)
+	if err == nil {
+		t.Fatalf("FastMarshalJSON should throw an error, but was '%v'", json)
+	}
+
+}
+
+func TestFastMarshalJSON(t *testing.T) {
+	aTime := time.Date(2015, 5, 29, 11, 1, 2, 3, time.UTC)
+	json, err := FastMarshalJSON(aTime)
+	if err != nil {
+		t.Fatal(err)
+	}
+	expected := "\"2015-05-29T11:01:02.000000003Z\""
+	if json != expected {
+		t.Fatalf("Expected %v, got %v", expected, json)
+	}
+
+	location, err := time.LoadLocation("Europe/Paris")
+	if err != nil {
+		t.Fatal(err)
+	}
+	aTime = time.Date(2015, 5, 29, 11, 1, 2, 3, location)
+	json, err = FastMarshalJSON(aTime)
+	if err != nil {
+		t.Fatal(err)
+	}
+	expected = "\"2015-05-29T11:01:02.000000003+02:00\""
+	if json != expected {
+		t.Fatalf("Expected %v, got %v", expected, json)
+	}
+}