Test logs

squirrelbattle/display/logsdisplay.py

@@ -17,6 +17,7 @@ class LogsDisplay(Display):
         messages = messages[::-1]
         self.pad.clear()
         for i in range(min(self.height, len(messages))):
-            self.pad.addstr(self.height - i - 1, self.x, messages[i][:self.width])
+            self.pad.addstr(self.height - i - 1, self.x,
+                            messages[i][:self.width])
         self.pad.refresh(0, 0, self.y, self.x, self.y + self.height,
                          self.x + self.width)

squirrelbattle/interfaces.py

@@ -52,6 +52,7 @@ class Map:
         self.start_x = start_x
         self.tiles = tiles
         self.entities = []
+        self.logs = Logs()
 
     def add_entity(self, entity: "Entity") -> None:
         """

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.