1 files changed, 48 insertions, 0 deletions
diff --git a/Testing/include/simulation/SimulationHistoryTest.h b/Testing/include/simulation/SimulationHistoryTest.h
new file mode 100644
index 00000000..d519de89
--- /dev/null
+++ b/Testing/include/simulation/SimulationHistoryTest.h
@@ -0,0 +1,48 @@
+#pragma once
+#include "simulation\SimulationHistory.h"
+#include "simulation\workloads\WorkloadHistory.h"
+
+#include <gtest\gtest.h>
+
+TEST(SimulationHistoryTest, SetGetHistoryAtTick)
+{
+	Simulation::SimulationHistory simulationHistory;
+	Simulation::WorkloadHistory workloadHistory;
+	workloadHistory.setFlopsDone(1, 100);
+
+	simulationHistory.setHistoryAtTick(1, workloadHistory);
+
+	auto resultHistory = simulationHistory.getHistoryAtTick(1);
+	ASSERT_EQ(resultHistory.history.at(0).first, 1);
+	ASSERT_EQ(resultHistory.history.at(0).second, 100);
+}
+
+TEST(SimulationHistoryTest, ClearHistory)
+{
+	Simulation::SimulationHistory simulationHistory;
+	Simulation::WorkloadHistory workloadHistory;
+	simulationHistory.setHistoryAtTick(1, workloadHistory);
+
+	ASSERT_EQ(simulationHistory.workloadHistories.size(), 1);
+
+	simulationHistory.clearHistory();
+
+	ASSERT_EQ(simulationHistory.workloadHistories.size(), 0);
+}
+
+TEST(SimulationHistoryTest, GetHistorySize)
+{
+	Simulation::SimulationHistory simulationHistory;
+	Simulation::WorkloadHistory workloadHistory;
+	simulationHistory.setHistoryAtTick(1, workloadHistory);
+
+	ASSERT_EQ(simulationHistory.getHistorySize(), 1);
+
+	simulationHistory.setHistoryAtTick(2, workloadHistory);
+
+	ASSERT_EQ(simulationHistory.getHistorySize(), 2);
+
+	simulationHistory.clearHistory();
+
+	ASSERT_EQ(simulationHistory.getHistorySize(), 0);
+}
+\ No newline at end of file