From 68ec59ab7ce4d43a4854eb0a5047b7807e378e6a Mon Sep 17 00:00:00 2001
From: Philipp Matthes <philipp.matthes@tu-dresden.de>
Date: Mon, 29 Apr 2024 10:42:55 +0200
Subject: [PATCH] Add validator test

---
 histories/validator_test.go | 154 ++++++++++++++++++++++++++++++++++++
 1 file changed, 154 insertions(+)
 create mode 100644 histories/validator_test.go

diff --git a/histories/validator_test.go b/histories/validator_test.go
new file mode 100644
index 0000000..d9c40dc
--- /dev/null
+++ b/histories/validator_test.go
@@ -0,0 +1,154 @@
+package histories
+
+import (
+	"predictor/phases"
+	"strings"
+	"testing"
+	"time"
+)
+
+func TestValidateEmptyPhases(t *testing.T) {
+	err := validatePhases(
+		time.Unix(0, 0),
+		time.Unix(90, 0),
+		[]HistoryPhaseEvent{},
+	)
+	if err == nil || err.Error() != "no phases" {
+		t.Errorf("unexpected validation result: %v", err)
+		t.FailNow()
+	}
+}
+
+func TestValidateNoPhaseBefore(t *testing.T) {
+	err := validatePhases(
+		time.Unix(0, 0),
+		time.Unix(90, 0),
+		[]HistoryPhaseEvent{
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.Green, // Green
+			},
+		},
+	)
+	if err == nil || err.Error() != "no phase before start time" {
+		t.Errorf("unexpected validation result: %v", err)
+		t.FailNow()
+	}
+}
+
+func TestDisallowedOrders(t *testing.T) {
+	disallowedOrders := [][]HistoryPhaseEvent{
+		{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.Red,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.Amber,
+			},
+		},
+		{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.Amber,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.Green,
+			},
+		},
+		{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.Amber,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.RedAmber,
+			},
+		},
+		{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.Green,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.RedAmber,
+			},
+		},
+		{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.RedAmber,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.Red,
+			},
+		},
+		{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.RedAmber,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.Amber,
+			},
+		},
+	}
+
+	for _, order := range disallowedOrders {
+		err := validatePhases(time.Unix(0, 0), time.Unix(10, 0), order)
+		// Must throw an error
+		if err == nil {
+			t.Errorf("erroneous phase order was not detected")
+			t.FailNow()
+		}
+		if !strings.HasSuffix(err.Error(), "is disallowed") {
+			t.Errorf("unexpected error message: %s", err.Error())
+		}
+	}
+}
+
+func TestValidatePhaseDuration(t *testing.T) {
+	err := validatePhases(
+		time.Unix(0, 0),
+		time.Unix(90, 0),
+		[]HistoryPhaseEvent{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.Red,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.RedAmber,
+			},
+		},
+	)
+	if err == nil || err.Error() != "redamber phase is too long" {
+		t.Errorf("unexpected validation result: %v", err)
+		t.FailNow()
+	}
+
+	err = validatePhases(
+		time.Unix(0, 0),
+		time.Unix(90, 0),
+		[]HistoryPhaseEvent{
+			{
+				Time:  time.Unix(0, 0),
+				Color: phases.Green,
+			},
+			{
+				Time:  time.Unix(10, 0),
+				Color: phases.Amber,
+			},
+		},
+	)
+	if err == nil || err.Error() != "amber phase is too long" {
+		t.Errorf("unexpected validation result: %v", err)
+		t.FailNow()
+	}
+}