def test_make_team_dict(self): t = MagicMock() t.name = 'Red' t.players = [self._make_lobby_player()] p_dict = { 'id': 0, 'name': 'Name', 'steam_id': '0', } t_info = { 'id': 0, 'name': 'Red', 'players': [{ 'class_id': 0, 'ready': True, 'player': p_dict, }] } from lobbypy.models.utils import make_team_dict rv = make_team_dict(0, t) self.assertEqual(rv, t_info)
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) return jsonify(200, team = make_team_dict(team_id, lobby.teams[team_id]))
def get(self, lobby_id): lobby = Lobby.query.get_or_404(lobby_id) team_dicts = [make_team_dict(i, t) for i, t in enumerate(lobby.teams)] return jsonify(200, teams = team_dicts)