From 6a42db93905a6b12be791c3b2cfff01bfb61451f Mon Sep 17 00:00:00 2001
From: Gahtan Nahdi <155860115+gahtan-syarif@users.noreply.github.com>
Date: Sat, 28 Sep 2024 18:17:40 +0700
Subject: [PATCH] test: add tests for timecontrol (#715)

---
 app/Makefile                   |  1 +
 app/tests/timecontrol_test.cpp | 60 ++++++++++++++++++++++++++++++++++
 2 files changed, 61 insertions(+)
 create mode 100644 app/tests/timecontrol_test.cpp

diff --git a/app/Makefile b/app/Makefile
index e5ef34447..513e5eab9 100644
--- a/app/Makefile
+++ b/app/Makefile
@@ -49,6 +49,7 @@ SRC_FILES_TEST   := \
 	$(TESTDIR)/player.cpp \
 	$(TESTDIR)/scoreboard_test.cpp \
 	$(TESTDIR)/sprt_test.cpp \
+	$(TESTDIR)/timecontrol_test.cpp \
 	$(TESTDIR)/uci_engine_test.cpp \
 	$(TESTDIR)/epd_reader_test.cpp \
 	$(TESTDIR)/match_test.cpp
diff --git a/app/tests/timecontrol_test.cpp b/app/tests/timecontrol_test.cpp
new file mode 100644
index 000000000..bbd14cf55
--- /dev/null
+++ b/app/tests/timecontrol_test.cpp
@@ -0,0 +1,60 @@
+#include <time/timecontrol.hpp>
+
+#include "doctest/doctest.hpp"
+
+using namespace fastchess;
+
+TEST_SUITE("TimeControl") {
+    TEST_CASE("moves/time+increment") {
+        TimeControl::Limits limits;
+        limits.moves = 3;
+        limits.time = 10000;
+        limits.increment = 100;
+        limits.timemargin = 100;
+
+        TimeControl tc(limits);
+
+        CHECK(tc.updateTime(5555) == true);
+        CHECK(tc.getTimeLeft() == limits.time + limits.increment - 5555 + limits.increment);
+        CHECK(tc.getMovesLeft() == 2);
+
+        CHECK(tc.updateTime(4745) == true);
+        CHECK(tc.getTimeLeft() == limits.increment);
+        CHECK(tc.getMovesLeft() == 1);
+
+        CHECK(tc.updateTime(50) == true);
+        CHECK(tc.getTimeLeft() == 100 + limits.time - 50 + limits.increment);
+        CHECK(tc.getMovesLeft() == 3);
+
+        CHECK(tc.updateTime(10251) == false);
+        CHECK(tc.getTimeLeft() == 10150 - 10251);
+        CHECK(tc.getMovesLeft() == 2);
+    }
+
+    TEST_CASE("Fixed time") {
+        TimeControl::Limits limits;
+        limits.fixed_time = 5000;
+        limits.timemargin = 200;
+
+        TimeControl tc(limits);
+
+
+        CHECK(tc.updateTime(limits.fixed_time + limits.timemargin - 1) == true);
+        CHECK(tc.getTimeLeft() == limits.fixed_time);
+        
+        CHECK(tc.updateTime(limits.fixed_time + limits.timemargin) == true);
+        CHECK(tc.getTimeLeft() == limits.fixed_time);
+
+        CHECK(tc.updateTime(limits.fixed_time + limits.timemargin + 1) == false);
+        CHECK(tc.getTimeLeft() == limits.fixed_time - (limits.fixed_time + limits.timemargin + 1));
+    }
+
+    TEST_CASE("Fixed depth/nodes") {
+        TimeControl::Limits limits;
+
+        TimeControl tc(limits);
+
+        CHECK(tc.updateTime(523199) == true);
+        CHECK(tc.getTimeLeft() == 0);
+    }
+}