def lobbyconnection( event_loop, database, mock_protocol, mock_games, mock_players, mock_player, mock_geoip, mock_nts_client ): lc = LobbyConnection( database=database, geoip=mock_geoip, game_service=mock_games, players=mock_players, nts_client=mock_nts_client, ladder_service=asynctest.create_autospec(LadderService) ) lc.player = mock_player lc.protocol = mock_protocol lc.player_service.fetch_player_data = CoroutineMock() lc.peer_address = Address("127.0.0.1", 1234) lc._authenticated = True return lc
async def test_command_avatar_select(mocker, db_engine, lobbyconnection: LobbyConnection, mock_player: Player): lobbyconnection.player = mock_player lobbyconnection.player.id = 2 # Dostya test user lobbyconnection._authenticated = True await lobbyconnection.on_message_received({ 'command': 'avatar', 'action': 'select', 'avatar': "http://content.faforever.com/faf/avatars/qai2.png" }) async with db_engine.acquire() as conn: result = await conn.execute( "SELECT selected from avatars where idUser=2") row = await result.fetchone() assert row[0] == 1