Added a test for lines 106-107 of game.py

2020-12-31 15:16:40 +01:00
parent 7cd4721daa
commit 8e0b8d4fee
1 changed files with 5 additions and 0 deletions
--- a/squirrelbattle/tests/game_test.py
+++ b/squirrelbattle/tests/game_test.py
@@ -596,3 +596,8 @@ class TestGame(unittest.TestCase):
        self.assertEqual(self.game.state, GameMode.CREDITS)
        self.game.display_actions(DisplayActions.MOUSE, 21, 21)
        self.game.display_actions(DisplayActions.REFRESH)
+
+        self.game.state = GameMode.CREDITS
+        self.game.handle_key_pressed(KeyValues.ENTER)
+        
+        self.assertEqual(self.game.state, GameMode.MAINMENU)