From 8e0b8d4fee2a2ce4989635a4006d854de59d5371 Mon Sep 17 00:00:00 2001
From: eichhornchen <eichhornchen@aspiredeb>
Date: Thu, 31 Dec 2020 15:16:40 +0100
Subject: [PATCH] Added a test for lines 106-107 of game.py

---
 squirrelbattle/tests/game_test.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/squirrelbattle/tests/game_test.py b/squirrelbattle/tests/game_test.py
index 40a2331..f8a6e93 100644
--- 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)