Function test_relations

tests/test_relations.py:37–88

(db)

@pytest.mark.asyncio
async def test_relations(db):
    tournament = Tournament(name="New Tournament")
    await tournament.save()
    await Event(name="Without participants", tournament_id=tournament.id).save()
    event = Event(name="Test", tournament_id=tournament.id)
    await event.save()
    participants = []
    for i in range(2):
        team = Team(name=f"Team {(i + 1)}")
        await team.save()
        participants.append(team)
    await event.participants.add(participants[0], participants[1])
    await event.participants.add(participants[0], participants[1])

    with pytest.raises(NoValuesFetched):
        [team.id for team in event.participants]  # pylint: disable=W0104

    teamids = []
    async for team in event.participants:
        teamids.append(team.id)
    assert set(teamids) == {participants[0].id, participants[1].id}
    teamids = [team.id async for team in event.participants]
    assert set(teamids) == {participants[0].id, participants[1].id}

    assert {team.id for team in event.participants} == {participants[0].id, participants[1].id}

    assert event.participants[0].id in {participants[0].id, participants[1].id}

    selected_events = await Event.filter(participants=participants[0].id).prefetch_related(
        "participants", "tournament"
    )
    assert len(selected_events) == 1
    assert selected_events[0].tournament.id == tournament.id
    assert len(selected_events[0].participants) == 2
    await participants[0].fetch_related("events")
    assert participants[0].events[0] == event

    await Team.fetch_for_list(participants, "events")

    await Team.filter(events__tournament__id=tournament.id)

    await Event.filter(tournament=tournament)

    await Tournament.filter(events__name__in=["Test", "Prod"]).distinct()

    result = await Event.filter(pk=event.pk).values(
        "event_id", "name", tournament="tournament__name"
    )
    assert result[0]["tournament"] == tournament.name

    result = await Event.filter(pk=event.pk).values_list("event_id", "participants__name")
    assert len(result) == 2


@pytest.mark.asyncio