New test cases for testing the Storage Handlers

rrborja · Jun 5, 2018 · dbf7c81 · dbf7c81
1 parent b8d1a9a
commit dbf7c81
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/storage_test.go b/storage_test.go
@@ -0,0 +1,21 @@
+package schedule
+
+import (
+	"os"
+	"testing"
+)
+
+func TestSavedShift(t *testing.T) {
+	testRecord := DayRecord{6, 6, 2006}
+	testRecord.ClockIn(Employee{"Tester", 666}, 0, 48)
+
+	if _, err := os.Stat(testRecord.String()); os.IsNotExist(err) {
+		t.Error("Test record was not checked in, thus not saved")
+	}
+
+	defer os.RemoveAll(testRecord.String())
+
+	if Construct(testRecord) == nil {
+		t.Error("Test record was not checked in, thus not saved")
+	}
+}