diff --git a/logic/__tests__/knights_queue_test.py b/logic/__tests__/knights_queue_test.py index 782be15..99ba7da 100644 --- a/logic/__tests__/knights_queue_test.py +++ b/logic/__tests__/knights_queue_test.py @@ -5,6 +5,20 @@ from models.knight import Knight class KnightsQueueTest(unittest.TestCase): + + def test_should_raise_when_knight_died_and_whole_team_dead(self): + with self.assertRaises(Exception): + knight1 = Knight(None) + knight1.health = 222 + knight2 = Knight(None) + knight2.health = 1 + + knights_queue = KnightsQueue([knight1], [knight2]) + knights_queue.dequeue_knight() + knights_queue.dequeue_knight() + knight2.health = -2 + knights_queue.dequeue_knight() + def test_should_make_valid_next_turn(self): knight1 = Knight(None) knight1.health = 222