Test logs

squirrelbattle/tests/entities_test.py

@@ -42,9 +42,11 @@ class TestEntities(unittest.TestCase):
         self.assertEqual(entity.maxhealth, entity.health)
         self.assertEqual(entity.strength, 2)
         for _ in range(9):
-            self.assertIsNone(entity.hit(entity))
+            self.assertEqual(entity.hit(entity),
+                             "beaver hits beaver. beaver takes 2 damage.")
             self.assertFalse(entity.dead)
-        self.assertIsNone(entity.hit(entity))
+        self.assertEqual(entity.hit(entity), "beaver hits beaver. "
+                         + "beaver takes 2 damage. beaver dies.")
         self.assertTrue(entity.dead)
 
         entity = Rabbit()
@@ -64,6 +66,9 @@ class TestEntities(unittest.TestCase):
         self.map.tick()
         self.assertTrue(entity.y == 2 and entity.x == 6)
         self.assertEqual(old_health - entity.strength, self.player.health)
+        self.assertEqual(self.map.logs.messages[-1],
+                         f"{entity.name} hits {self.player.name}. \
+{self.player.name} takes {entity.strength} damage.")
 
         # Fight the rabbit
         old_health = entity.health

squirrelbattle/tests/game_test.py

@@ -256,6 +256,18 @@ class TestGame(unittest.TestCase):
         self.game.handle_key_pressed(KeyValues.ENTER)
         self.assertEqual(self.game.state, GameMode.MAINMENU)
 
+    def test_logs(self) -> None:
+        """
+        Tests the use of logs
+        """
+        self.game.logs.add_message("Hello World !")
+        self.assertEqual(self.game.logs.messages, ["Hello World !"])
+        self.game.logs.add_messages(["Hello", "World"])
+        self.assertEqual(self.game.logs.messages, ["Hello World !",
+                                                   "Hello", "World"])
+        self.game.logs.clear()
+        self.assertEqual(self.game.logs.messages, [])
+
     def test_dead_screen(self) -> None:
         """
         Kill player and render dead screen.