def test_is_over(self):
f1 = Faction('Goblins', [Goblin(), Goblin()])
f2 = Faction('Orcs', [Orc(), Orc()])
a = Arena([f1, f2])
f1.get_entities()[0].state = EntityState.DEAD
f1.get_entities()[1].state = EntityState.DEAD
self.assertTrue(a.is_over())
def test_get_winning_faction(self):
f1 = Faction('Goblins', [Goblin(), Goblin()])
f2 = Faction('Orcs', [Orc(), Orc()])
a = Arena([f1, f2])
f1.get_entities()[0].state = EntityState.DEAD
f1.get_entities()[1].state = EntityState.DEAD
self.assertEqual(a.get_winning_faction(), f2)
def test_get_opposing_factions(self):
f1 = Faction('Goblins', [Goblin()])
f2 = Faction('Orcs', [Orc()])
f3 = Faction('More Bad Orcs', [Orc()])
a = Arena([f1, f2, f3])
opposing = a.get_opposing_factions(f1)
self.assertNotIn(f1, opposing)
def test_all_factions_dead(self):
f1 = Faction('Goblins', [Goblin(), Goblin()])
f2 = Faction('Orcs', [Orc(), Orc()])
a = Arena([f1, f2])
f1.get_entities()[0].state = EntityState.DEAD
f1.get_entities()[1].state = EntityState.DEAD
f2.get_entities()[0].state = EntityState.DEAD
f2.get_entities()[1].state = EntityState.DEAD
self.assertIsNone(a.get_winning_faction())
def test_initiative(self):
f1 = Faction('Goblins', [Goblin()])
f2 = Faction('Orcs', [Orc()])
a = Arena([f1, f2])
with patch('game_engine.dice._random_int') as mock_roll:
mock_roll.return_value = 10
a.determine_initiative_order()
self.assertIsInstance(a.get_init_order()[0], Goblin)
self.assertIsInstance(a.get_init_order()[1], Orc)
def test_is_not_over(self):
f1 = Faction('Goblins', [Goblin(), Goblin()])
f2 = Faction('Orcs', [Orc(), Orc()])
f3 = Faction('More Bad Orcs', [Orc(), Orc()])
a = Arena([f1, f2, f3])
# kill the first faction and then one from the other two
f1.get_entities()[0].state = EntityState.DEAD
f1.get_entities()[1].state = EntityState.DEAD
f2.get_entities()[0].state = EntityState.DEAD
f3.get_entities()[1].state = EntityState.DEAD
self.assertFalse(a.is_over())
def test_random_target(self):
g = Goblin()
o1 = Orc()
o2 = Orc()
o3 = Orc()
goblins = Faction('Goblins', [g])
orcs = Faction('Orcs', [o1, o2, o3])
arena = Arena([goblins, orcs])
# this is only for testing, need to set the arena on the goblin
# normally this would happen via the take_turn method
g.arena = arena
g.select_target(TargetStrategy.RANDOM)
self.assertTrue(g.target is not None)
self.assertIn(g.target, [o1, o2, o3])
def test_no_winning_faction_yet(self):
f1 = Faction('Goblins', [Goblin(), Goblin()])
f2 = Faction('Orcs', [Orc(), Orc()])
a = Arena([f1, f2])
self.assertIsNone(a.get_winning_faction())
def test_create_arena(self):
f1 = Faction('Goblins', [Goblin()])
f2 = Faction('Orcs', [Orc()])
a = Arena([f1, f2])
self.assertIsNotNone(a)