From 414f51fa2e00ea73e4c242c4da3472b282bf0470 Mon Sep 17 00:00:00 2001
From: przemek83 <4788832+przemek83@users.noreply.github.com>
Date: Mon, 30 Sep 2024 17:21:04 +0200
Subject: [PATCH] Add tests for the isLevelPicked method.

---
 test/MenuTest.cpp | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)
 create mode 100644 test/MenuTest.cpp

diff --git a/test/MenuTest.cpp b/test/MenuTest.cpp
new file mode 100644
index 0000000..2ca70a1
--- /dev/null
+++ 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);
+    }
+}