tests: added edge case
This commit is contained in:
parent
f06ea9e5b0
commit
4caba91c7e
@ -5,6 +5,20 @@ from models.knight import Knight
|
|||||||
|
|
||||||
|
|
||||||
class KnightsQueueTest(unittest.TestCase):
|
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):
|
def test_should_make_valid_next_turn(self):
|
||||||
knight1 = Knight(None)
|
knight1 = Knight(None)
|
||||||
knight1.health = 222
|
knight1.health = 222
|
||||||
|
Loading…
Reference in New Issue
Block a user