async def test_start_search_multiqueue(ladder_service: LadderService,
player_factory, queue_factory,
event_loop):
ladder_service.queues["tmm2v2"] = queue_factory("tmm2v2")
p1 = player_factory("Dostya",
ladder_rating=(1000, 10),
with_lobby_connection=True)
ladder_service.start_search([p1], "ladder1v1")
await exhaust_callbacks(event_loop)
assert "ladder1v1" in ladder_service._searches[p1]
ladder_service.start_search([p1], "tmm2v2")
await exhaust_callbacks(event_loop)
assert "ladder1v1" in ladder_service._searches[p1]
assert "tmm2v2" in ladder_service._searches[p1]
ladder_service.cancel_search(p1, "tmm2v2")
await exhaust_callbacks(event_loop)
assert "ladder1v1" in ladder_service._searches[p1]
assert "tmm2v2" not in ladder_service._searches[p1]
async def test_game_start_cancels_search(ladder_service: LadderService,
player_factory, queue_factory,
event_loop):
ladder_service.queues["tmm2v2"] = queue_factory("tmm2v2")
p1 = player_factory("Dostya",
player_id=1,
ladder_rating=(1000, 10),
with_lobby_connection=True)
p2 = player_factory("Brackman",
player_id=2,
ladder_rating=(1000, 10),
with_lobby_connection=True)
ladder_service.start_search([p1], "ladder1v1")
ladder_service.start_search([p2], "ladder1v1")
ladder_service.start_search([p1], "tmm2v2")
ladder_service.start_search([p2], "tmm2v2")
await exhaust_callbacks(event_loop)
assert "ladder1v1" in ladder_service._searches[p1]
assert "tmm2v2" in ladder_service._searches[p1]
assert "ladder1v1" in ladder_service._searches[p2]
assert "tmm2v2" in ladder_service._searches[p2]
ladder_service.on_match_found(ladder_service._searches[p1]["ladder1v1"],
ladder_service._searches[p2]["ladder1v1"],
ladder_service.queues["ladder1v1"])
assert "ladder1v1" not in ladder_service._searches[p1]
assert "tmm2v2" not in ladder_service._searches[p1]
assert "ladder1v1" not in ladder_service._searches[p2]
assert "tmm2v2" not in ladder_service._searches[p2]
async def test_cancel_twice(ladder_service: LadderService):
p1 = mock.create_autospec(Player('Dostya', id=1))
p1.ladder_rating = (1500, 500)
p1.numGames = 0
p2 = mock.create_autospec(Player('Brackman', id=1))
p2.ladder_rating = (2000, 50)
p2.numGames = 0
search = Search([p1])
search2 = Search([p2])
ladder_service.start_search(p1, search, 'ladder1v1')
ladder_service.start_search(p2, search2, 'ladder1v1')
searches = ladder_service._cancel_existing_searches(p1)
assert search.is_cancelled
assert searches == [search]
assert not search2.is_cancelled
searches = ladder_service._cancel_existing_searches(p1)
assert searches == []
searches = ladder_service._cancel_existing_searches(p2)
assert search2.is_cancelled
assert searches == [search2]
async def test_start_game_called_on_match(ladder_service: LadderService, player_factory):
p1 = player_factory(
"Dostya",
player_id=1,
ladder_rating=(2300, 64),
ladder_games=0,
with_lobby_connection=True
)
p2 = player_factory(
"QAI",
player_id=2,
ladder_rating=(2350, 125),
ladder_games=0,
with_lobby_connection=True
)
ladder_service.start_game = CoroutineMock()
ladder_service.write_rating_progress = CoroutineMock()
ladder_service.start_search([p1], "ladder1v1")
ladder_service.start_search([p2], "ladder1v1")
await asyncio.sleep(2)
ladder_service.write_rating_progress.assert_called()
ladder_service.start_game.assert_called_once()
async def test_start_search_cancels_previous_search(
ladder_service: LadderService,
player_factory,
event_loop
):
p1 = player_factory(
"Dostya",
player_id=1,
ladder_rating=(1500, 500),
ladder_games=0,
with_lobby_connection=True
)
ladder_service.start_search([p1], "ladder1v1")
await exhaust_callbacks(event_loop)
search1 = ladder_service._searches[p1]["ladder1v1"]
assert p1.state == PlayerState.SEARCHING_LADDER
assert search1 in ladder_service.queues["ladder1v1"]._queue
ladder_service.start_search([p1], "ladder1v1")
await exhaust_callbacks(event_loop)
search2 = ladder_service._searches[p1]["ladder1v1"]
assert p1.state == PlayerState.SEARCHING_LADDER
assert search1.is_cancelled
assert search1 not in ladder_service.queues["ladder1v1"]._queue
assert search2 in ladder_service.queues["ladder1v1"]._queue
async def test_search_info_message(ladder_service: LadderService,
player_factory, queue_factory, event_loop):
ladder_service.queues["tmm2v2"] = queue_factory("tmm2v2")
p1 = player_factory("Dostya",
player_id=1,
ladder_rating=(1000, 10),
with_lobby_connection=True)
p1.write_message = CoroutineMock()
p2 = player_factory("Rhiza",
player_id=2,
ladder_rating=(1000, 10),
with_lobby_connection=True)
p2.write_message = CoroutineMock()
ladder_service.start_search([p1, p2], "ladder1v1")
await exhaust_callbacks(event_loop)
msg = {
"command": "search_info",
"queue_name": "ladder1v1",
"state": "start"
}
p1.write_message.assert_called_once_with(msg)
p2.write_message.assert_called_once_with(msg)
p1.write_message.reset_mock()
p2.write_message.reset_mock()
ladder_service.start_search([p1, p2], "tmm2v2")
await exhaust_callbacks(event_loop)
msg = {"command": "search_info", "queue_name": "tmm2v2", "state": "start"}
p1.write_message.assert_called_once_with(msg)
p2.write_message.assert_called_once_with(msg)
p1.write_message.reset_mock()
p2.write_message.reset_mock()
ladder_service.cancel_search(p1)
await exhaust_callbacks(event_loop)
call_args = [
mock.call({
"command": "search_info",
"queue_name": "ladder1v1",
"state": "stop"
}),
mock.call({
"command": "search_info",
"queue_name": "tmm2v2",
"state": "stop"
}),
]
assert p1.write_message.call_args_list == call_args
assert p2.write_message.call_args_list == call_args
async def test_cancel_twice(ladder_service: LadderService, player_factory):
p1 = player_factory(login="******", player_id=1, ladder_rating=(1500, 500), ladder_games=0)
p2 = player_factory(login="******", player_id=2, ladder_rating=(2000, 500), ladder_games=0)
ladder_service.start_search([p1], "ladder1v1")
search = ladder_service._searches[p1]["ladder1v1"]
ladder_service.start_search([p2], "ladder1v1")
search2 = ladder_service._searches[p2]["ladder1v1"]
ladder_service.cancel_search(p1)
assert search.is_cancelled
assert not search2.is_cancelled
ladder_service.cancel_search(p1)
ladder_service.cancel_search(p2)
assert search2.is_cancelled
async def test_cancel_all_searches(ladder_service: LadderService,
player_factory, event_loop):
p1 = player_factory(login="******", player_id=1, ladder_rating=(1500, 500), ladder_games=0)
ladder_service.start_search([p1], "ladder1v1")
await exhaust_callbacks(event_loop)
search = ladder_service._searches[p1]["ladder1v1"]
assert p1.state == PlayerState.SEARCHING_LADDER
assert search in ladder_service.queues["ladder1v1"]._queue
assert not search.is_cancelled
ladder_service.cancel_search(p1)
assert p1.state == PlayerState.IDLE
assert search.is_cancelled
assert "ladder1v1" not in ladder_service._searches[p1]
async def test_start_and_cancel_search(ladder_service: LadderService):
p1 = mock.create_autospec(Player('Dostya', id=1))
p1.ladder_rating = (1500, 500)
p1.numGames = 0
search = Search([p1])
ladder_service.start_search(p1, search, 'ladder1v1')
await asyncio.sleep(0) # Give the other coro a chance to run
assert p1.state == PlayerState.SEARCHING_LADDER
assert ladder_service.queues['ladder1v1'].queue[search]
assert not search.is_cancelled
ladder_service.cancel_search(p1)
assert p1.state == PlayerState.IDLE
assert search.is_cancelled
async def test_start_game_called_on_match(ladder_service: LadderService):
p1 = mock.create_autospec(Player('Dostya', id=1))
p1.ladder_rating = (2300, 64)
p1.numGames = 0
p2 = mock.create_autospec(Player('QAI', id=4))
p2.ladder_rating = (2350, 125)
p2.numGames = 0
ladder_service.start_game = CoroMock()
ladder_service.inform_player = mock.Mock()
ladder_service.start_search(p1, Search([p1]), 'ladder1v1')
ladder_service.start_search(p2, Search([p2]), 'ladder1v1')
await asyncio.sleep(1)
ladder_service.inform_player.assert_called()
ladder_service.start_game.assert_called_once()
async def test_start_search_cancels_previous_search(
ladder_service: LadderService):
p1 = mock.create_autospec(Player('Dostya', id=1))
p1.ladder_rating = (1500, 500)
p1.numGames = 0
search1 = Search([p1])
ladder_service.start_search(p1, search1, 'ladder1v1')
await asyncio.sleep(0) # Give the other coro a chance to run
assert p1.state == PlayerState.SEARCHING_LADDER
assert ladder_service.queues['ladder1v1'].queue[search1]
search2 = Search([p1])
ladder_service.start_search(p1, search2, 'ladder1v1')
await asyncio.sleep(0) # Give the other coro a chance to run
assert p1.state == PlayerState.SEARCHING_LADDER
assert search1.is_cancelled
assert not ladder_service.queues['ladder1v1'].queue.get(search1)
assert ladder_service.queues['ladder1v1'].queue[search2]
async def test_on_match_found_sets_player_state(ladder_service: LadderService,
player_factory, event_loop):
p1 = player_factory("Dostya",
player_id=1,
ladder_rating=(1000, 10),
with_lobby_connection=True)
p2 = player_factory("Brackman",
player_id=2,
ladder_rating=(1000, 10),
with_lobby_connection=True)
ladder_service.start_search([p1], "ladder1v1")
ladder_service.start_search([p2], "ladder1v1")
await exhaust_callbacks(event_loop)
assert p1.state is PlayerState.SEARCHING_LADDER
assert p2.state is PlayerState.SEARCHING_LADDER
ladder_service.on_match_found(ladder_service._searches[p1]["ladder1v1"],
ladder_service._searches[p2]["ladder1v1"],
ladder_service.queues["ladder1v1"])
assert p1.state is PlayerState.STARTING_AUTOMATCH
assert p2.state is PlayerState.STARTING_AUTOMATCH