More tests

dungeonbattle/tests/game_test.py

@@ -1,5 +1,7 @@
 import unittest
 
+from dungeonbattle.display.display import Display
+from dungeonbattle.display.statsdisplay import StatsDisplay
 from dungeonbattle.game import Game, KeyValues, GameMode
 from dungeonbattle.menus import MainMenuValues
 
@@ -14,6 +16,7 @@ class TestGame(unittest.TestCase):
 
     def test_load_game(self) -> None:
         self.assertRaises(NotImplementedError, Game.load_game, "game.save")
+        self.assertRaises(NotImplementedError, Display(None).refresh)
 
     def test_key_translation(self) -> None:
         """
@@ -37,6 +40,7 @@ class TestGame(unittest.TestCase):
             self.game.settings.KEY_RIGHT_SECONDARY), KeyValues.RIGHT)
         self.assertEqual(self.game.translate_key(
             self.game.settings.KEY_ENTER), KeyValues.ENTER)
+        self.assertEqual(self.game.translate_key(' '), KeyValues.SPACE)
 
     def test_key_press(self) -> None:
         """
@@ -54,7 +58,8 @@ class TestGame(unittest.TestCase):
         self.game.handle_key_pressed(KeyValues.ENTER)
         self.assertEqual(self.game.state, GameMode.SETTINGS)
 
-        self.game.state = GameMode.MAINMENU
+        self.game.handle_key_pressed(KeyValues.SPACE)
+        self.assertEqual(self.game.state, GameMode.MAINMENU)
 
         self.game.handle_key_pressed(KeyValues.DOWN)
         self.assertEqual(self.game.main_menu.validate(),
@@ -95,3 +100,11 @@ class TestGame(unittest.TestCase):
         new_y, new_x = self.game.player.y, self.game.player.x
         self.assertEqual(new_y, y)
         self.assertEqual(new_x, x - 1)
+
+        self.game.handle_key_pressed(KeyValues.SPACE)
+        self.assertEqual(self.game.state, GameMode.MAINMENU)
+
+    def test_stats_display(self) -> None:
+        self.game.current_display = StatsDisplay(None, self.game.player,
+                                                 42, 42, 0, 0)
+        self.game.current_display.refresh()