Function test_aggregation

tests/test_aggregation.py:22–61

(db)

@pytest.mark.asyncio
async def test_aggregation(db):
    tournament = Tournament(name="New Tournament")
    await tournament.save()
    await Tournament.create(name="Second tournament")
    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])

    tournaments_with_count = (
        await Tournament.all().annotate(events_count=Count("events")).filter(events_count__gte=1)
    )
    assert len(tournaments_with_count) == 1
    assert tournaments_with_count[0].events_count == 2

    event_with_lowest_team_id = (
        await Event.filter(event_id=event.event_id)
        .first()
        .annotate(lowest_team_id=Min("participants__id"))
    )
    assert event_with_lowest_team_id.lowest_team_id == participants[0].id

    ordered_tournaments = (
        await Tournament.all().annotate(events_count=Count("events")).order_by("events_count")
    )
    assert len(ordered_tournaments) == 2
    assert ordered_tournaments[1].id == tournament.id
    event_with_annotation = (
        await Event.all().annotate(tournament_test_id=Sum("tournament__id")).first()
    )
    assert event_with_annotation.tournament_test_id == event_with_annotation.tournament_id

    with pytest.raises(FieldError, match="name__id not resolvable"):
        await Event.all().annotate(tournament_test_id=Sum("name__id")).first()


@pytest.mark.asyncio