def test_team_add_remove(self):
common.add_player_to_team(user=self.player1, player=self.player1, season=self.season_open, team=self.team)
self.assertEqual(
1,
common.TeamPlayerSeason.objects.filter(
player=self.player1, season=self.season_open, team=self.team
).count(),
)
common.add_player_to_team(
user=self.player2, player=self.player2, season=self.season_open, team=self.team, is_captain=True
)
self.assertTrue(self.player2.is_captain(self.team, self.season_open))
self.assertEqual(2, len(common.find_players_on(self.team, self.season_open)))
common.remove_player_from_team(user=self.player1, player=self.player1, season=self.season_open, team=self.team)
self.assertEqual(1, len(common.find_players_on(self.team, self.season_open)))
common.add_player_to_team(user=self.player1, player=self.player1, season=self.season_open, team=self.team)
self.assertEqual(2, len(common.find_players_on(self.team, self.season_open)))
common.remove_player_from_team(user=self.player2, player=self.player1, season=self.season_open, team=self.team)
self.assertEqual(1, len(common.find_players_on(self.team, self.season_open)))
def test_remove_player(self):
self.assertFalse(self.player1.is_superuser())
self.assertFalse(self.player1.is_captain(self.team, self.season_open))
common.add_player_to_team(user=self.player1, player=self.player1, season=self.season_open, team=self.team)
common.add_player_to_team(
user=self.player2, player=self.player2, season=self.season_open, team=self.team, is_captain=True
)
# cannot remove another if you are not a captain
self.assertRaisesMessage(
common.PermissionsException,
common.PermissionsException.MSG_CAPTAIN_REMOVE,
common.remove_player_from_team,
user=self.player1,
player=self.player2,
season=self.season_open,
team=self.team,
)
self.assertEqual(2, len(common.find_players_on(self.team, self.season_open)))
common.remove_player_from_team(user=self.player2, player=self.player1, season=self.season_open, team=self.team)
self.assertEqual(1, len(common.find_players_on(self.team, self.season_open)))
