ChessCode/backend/tests/unit/api/play/test_game_queue.py at a5ae0d556afe1b53fd0f5f4ffc9e3876ee7e8201 · Patai5/ChessCode · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import pytest
from api.play.chess_board import CustomOutcome
from api.play.game import ALL_ACTIVE_GAMES_MANAGER, Game
from api.play.game_modes import GameMode, TimeControl
from api.play.game_queue import GameQueueManager
from chess import Termination
from users.models import AnonymousSessionUser


@pytest.mark.django_db
def test_anonymous_user_game() -> None:
    user1 = AnonymousSessionUser.objects.create(session_key="session1")
    user2 = AnonymousSessionUser.objects.create(session_key="session2")

    gameMode = GameMode("Blitz", [TimeControl(120)])
    queueManager = GameQueueManager([gameMode])
    queue = queueManager.get_game_queue("Blitz", 120)

    assert queue is not None

    addedUser1 = False
    addedUser2 = False

    def onAddUser1Callback(_: Game) -> None:
        nonlocal addedUser1
        addedUser1 = True

    def onAddUser2Callback(_: Game) -> None:
        nonlocal addedUser2
        addedUser2 = True

    queueManager.add_user(user1, queue, onAddUser1Callback)
    queueManager.add_user(user2, queue, onAddUser2Callback)

    first_game = next(iter(ALL_ACTIVE_GAMES_MANAGER.games.values()))

    outcome = CustomOutcome(Termination.CHECKMATE, None)
    first_game.finish(outcome)
    gameId = first_game.game_id
    assert gameId is not None

    ALL_ACTIVE_GAMES_MANAGER.remove_game(gameId)