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