def test_restart_round(self):
# wrong player trying to restart
self.assertIsNone(restart_round(153, self.hcriat))
# real case
player1, _, _, pw_map = create_player(Team.BLUE, self.hcriat)
player2, _, _, pw_map = create_player(Team.RED, self.hcriat)
# move both once place
teleport_player(player1.uid, player1.x+1, player1.y+1, self.hcriat)
teleport_player(player2.uid, player2.x+1, player2.y+1, self.hcriat)
# change life
player1.vida = 1
player2.vida = 1
# change score
self.score.murio_azul()
self.score.murio_rojo()
# restart
players, new_score = restart_round(player1.uid, self.hcriat)
# base positions
new_blue = players.pop(0) if players[0].team == Team.BLUE else players.pop()
new_red = players.pop()
# blue
self.assertEqual(new_blue.x, pw_map.x_azul)
self.assertEqual(new_blue.y, pw_map.y_azul)
self.assertEqual(new_blue.vida, self.hcriat.VIDA_MAX)
# red
self.assertEqual(new_red.x, pw_map.x_rojo)
self.assertEqual(new_red.y, pw_map.y_rojo)
self.assertEqual(new_red.vida, self.hcriat.VIDA_MAX)
# score
self.assertEqual(new_score[0], 0)
self.assertEqual(new_score[1], 0)
def test_teleport_player(self):
player, _, _, _ = create_player(Team.BLUE, self.hcriat)
teleport_player(player.uid, 1, 1, self.hcriat)
self.assertEqual((player.x, player.y), (1, 1))
