refactor: split concrete menus into multiple files

Author: FlysonBot

src/mastermind/ui/menu/__init__.py

@@ -1,9 +1,7 @@
-from mastermind.ui.menu.concrete_menus import (
-    GameHistoryMenu,
-    MainMenu,
-    NewGameMenu,
-    ResumeGameMenu,
-)
+from mastermind.ui.menu.game_history_menu import GameHistoryMenu
+from mastermind.ui.menu.main_menu import MainMenu
+from mastermind.ui.menu.new_game_menu import NewGameMenu
+from mastermind.ui.menu.resume_game_menu import ResumeGameMenu
 
 __all__ = [
     "MainMenu",

src/mastermind/ui/menu/game_history_menu.py

@@ -0,0 +1,41 @@
+from typing import Optional
+
+import pandas as pd
+
+from mastermind.main.game_history import GameHistoryManager
+from mastermind.ui.menu.data_menu import DataDisplayMenu
+from mastermind.utils.render_dataframe import render_dataframe
+
+
+class GameHistoryMenu(DataDisplayMenu):
+    """
+    The menu for displaying the game history.
+    """
+
+    name = "Game History"
+    width = 25
+
+    def _fetch_data(self) -> Optional[pd.DataFrame]:
+        """
+        Retrieves the game history data.
+        """
+        return GameHistoryManager.retrieve_game_history()
+
+    def _render_data(self, data: pd.DataFrame) -> None:
+        """
+        Renders the game history data.
+        """
+        render_dataframe(data)
+
+    def _empty_message(self) -> str:
+        """
+        Returns the message to display when there is no game history.
+        """
+        return "No game history found."
+
+    def display(self) -> None:
+        """
+        Displays the game history menu and waits for user input to continue.
+        """
+        super().display()
+        input("\nPress Enter to continue...")

src/mastermind/ui/menu/main_menu.py

@@ -0,0 +1,15 @@
+from mastermind.ui.menu.option_menu import OptionMenu
+
+
+class MainMenu(OptionMenu):
+    """
+    The main menu of the application.
+    """
+
+    name = "Main Menu"
+    menu = {
+        "1": "Start New Game",
+        "2": "Load Saved Game",
+        "3": "Game History",
+        "0": "Save and Exit",
+    }

src/mastermind/ui/menu/new_game_menu.py

@@ -0,0 +1,16 @@
+from mastermind.ui.menu.option_menu import OptionMenu
+
+
+class NewGameMenu(OptionMenu):
+    """
+    The menu for starting a new game.
+    """
+
+    name = "New Game Menu"
+    menu = {
+        "1": "You vs Someone Else",
+        "2": "You vs AI",
+        "3": "AI vs You",
+        "4": "Solve External Game",
+        "0": "Return to Main Menu",
+    }

src/mastermind/ui/menu/concrete_menus.py …c/mastermind/ui/menu/resume_game_menu.pysrc/mastermind/ui/menu/concrete_menus.py renamed to src/mastermind/ui/menu/resume_game_menu.py src/mastermind/ui/menu/concrete_menus.py renamed to src/mastermind/ui/menu/resume_game_menu.py

@@ -1,74 +1,9 @@
 from typing import Optional, Union
 
 import pandas as pd
-
 from mastermind.main.game_history import GameHistoryManager
 from mastermind.ui.menu.data_menu import DataDisplayMenu
-from mastermind.ui.menu.option_menu import OptionMenu
-from mastermind.utils import render_dataframe
-
-
-class MainMenu(OptionMenu):
-    """
-    The main menu of the application.
-    """
-
-    name = "Main Menu"
-    menu = {
-        "1": "Start New Game",
-        "2": "Load Saved Game",
-        "3": "Game History",
-        "0": "Save and Exit",
-    }
-
-
-class NewGameMenu(OptionMenu):
-    """
-    The menu for starting a new game.
-    """
-
-    name = "New Game Menu"
-    menu = {
-        "1": "You vs Someone Else",
-        "2": "You vs AI",
-        "3": "AI vs You",
-        "4": "Solve External Game",
-        "0": "Return to Main Menu",
-    }
-
-
-class GameHistoryMenu(DataDisplayMenu):
-    """
-    The menu for displaying the game history.
-    """
-
-    name = "Game History"
-    width = 25
-
-    def _fetch_data(self) -> Optional[pd.DataFrame]:
-        """
-        Retrieves the game history data.
-        """
-        return GameHistoryManager.retrieve_game_history()
-
-    def _render_data(self, data: pd.DataFrame) -> None:
-        """
-        Renders the game history data.
-        """
-        render_dataframe(data)
-
-    def _empty_message(self) -> str:
-        """
-        Returns the message to display when there is no game history.
-        """
-        return "No game history found."
-
-    def display(self) -> None:
-        """
-        Displays the game history menu and waits for user input to continue.
-        """
-        super().display()
-        input("\nPress Enter to continue...")
+from mastermind.utils.render_dataframe import render_dataframe
 
 
 class ResumeGameMenu(DataDisplayMenu):

tests/ui/menu/test_concrete_menus.py

@@ -0,0 +1,52 @@
+import unittest
+from unittest.mock import MagicMock, patch
+
+import pandas as pd
+
+from mastermind.ui.menu.game_history_menu import GameHistoryMenu
+
+
+class TestGameHistoryMenu(unittest.TestCase):
+    def setUp(self):
+        self.menu = GameHistoryMenu()
+
+    @patch("mastermind.main.game_history.GameHistoryManager.retrieve_game_history")
+    def test_print_content_with_data(self, mock_retrieve_game_history):
+        mock_retrieve_game_history.return_value = pd.DataFrame(
+            {"Game": ["Game 1", "Game 2"]}
+        )
+        self.menu._render_data = MagicMock()
+        with patch("builtins.print") as mock_print:
+            self.menu._print_content()
+            mock_retrieve_game_history.assert_called()
+            self.menu._render_data.assert_called_with(
+                mock_retrieve_game_history.return_value
+            )
+            mock_print.assert_not_called()
+
+    @patch("mastermind.main.game_history.GameHistoryManager.retrieve_game_history")
+    @patch("builtins.print")
+    def test_print_content_without_data(self, mock_print, mock_retrieve_game_history):
+        mock_retrieve_game_history.return_value = None
+        self.menu._print_content()
+        mock_retrieve_game_history.assert_called()
+        mock_print.assert_called_with("No game history found.")
+
+    def test_fetch_data(self):
+        with patch(
+            "mastermind.main.game_history.GameHistoryManager.retrieve_game_history"
+        ) as mock_retrieve_game_history:
+            mock_retrieve_game_history.return_value = pd.DataFrame(
+                {"Game": ["Game 1", "Game 2"]}
+            )
+            self.assertEqual(
+                self.menu._fetch_data().to_dict(),
+                pd.DataFrame({"Game": ["Game 1", "Game 2"]}).to_dict(),
+            )
+
+    @patch("builtins.input", return_value="")
+    def test_display(self, mock_input):
+        with patch.object(GameHistoryMenu, "_print_content") as mock_print_content:
+            self.menu.display()
+            mock_print_content.assert_called()
+            mock_input.assert_called_once_with("\nPress Enter to continue...")