def get(self, lobby_id, team_id): lobby = Lobby.query.get_or_404(lobby_id) if team_id >= len(lobby.teams) or team_id < 0: abort(404) players_dict = [make_lobby_player_dict(lp) for lp in lobby.teams[team_id].players] return jsonify(200, lobby_players = players_dict)
def get(self, lobby_id, team_id, player_id): lobby = Lobby.query.get_or_404(lobby_id) if team_id >= len(lobby.teams) or team_id < 0: abort(404) team = lobby.teams[team_id] player = Player.query.get_or_404(player_id) if not player in team: abort(404) return jsonify(200, lobby_player = make_lobby_player_dict(team.get_lobby_player(player)))
def test_make_lobby_player_dict(self): lp = MagicMock() lp.class_id = 0 lp.ready = True lp.player = MagicMock() lp.player.id = 0 lp.player.name = 'Name' lp.player.steam_id = '0' lp_info = { 'class_id': 0, 'ready': True, 'player': { 'id': 0, 'name': 'Name', 'steam_id': '0', }, } from lobbypy.models.utils import make_lobby_player_dict rv = make_lobby_player_dict(lp) self.assertEqual(rv, lp_info)