Add tests for the isLevelPicked method.

przemek83 · Sep 30, 2024 · 414f51f · 414f51f
1 parent f9c539b
commit 414f51f
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/test/MenuTest.cpp b/test/MenuTest.cpp
@@ -0,0 +1,21 @@
+#include <catch2/catch_test_macros.hpp>
+
+#include <src/Menu.h>
+
+#include "FakeDisplay.h"
+
+TEST_CASE("Menu usage", "[Menu]")
+{
+    FakeDisplay display;
+    Menu menu(display);
+
+    SECTION("isLevelPicked")
+    {
+        REQUIRE(menu.isLevelPicked(UserChoice::LEVEL_1));
+        REQUIRE(menu.isLevelPicked(UserChoice::LEVEL_2));
+        REQUIRE(menu.isLevelPicked(UserChoice::LEVEL_3));
+        REQUIRE(menu.isLevelPicked(UserChoice::LEVEL_4));
+        REQUIRE(menu.isLevelPicked(UserChoice::LEVEL_MENU) == false);
+        REQUIRE(menu.isLevelPicked(UserChoice::EXIT) == false);
+    }
+}