Mobilizon-Reshare-condividi-sui-social/tests/models/test_event.py

from datetime import datetime, timedelta, timezone
from uuid import UUID

import arrow
import pytest
import tortoise.timezone

from mobilizon_reshare.event.event import EventPublicationStatus, MobilizonEvent
from mobilizon_reshare.models.event import Event
from mobilizon_reshare.models.publication import PublicationStatus


@pytest.mark.asyncio
async def test_event_create(event_model_generator):
    event_model = event_model_generator()
    await event_model.save()
    event_db = await Event.filter(name="event_1").first()
    assert event_db.begin_datetime == datetime(
        year=2021,
        month=1,
        day=1,
        hour=11,
        minute=30,
        tzinfo=timezone(timedelta(hours=0)),
    )


@pytest.mark.asyncio
async def test_event_timezone_storage(event_model_generator):
    cet = timezone(timedelta(hours=1))
    event_cet = event_model_generator(
        begin_date=datetime(year=2021, month=6, day=6, hour=5, minute=0, tzinfo=cet),
    )
    await event_cet.save()
    cest = timezone(timedelta(hours=2))
    event_cest = event_model_generator(
        idx=2,
        begin_date=datetime(year=2021, month=6, day=6, hour=6, minute=0, tzinfo=cest),
    )
    await event_cest.save()
    events = await Event.all()
    assert len(events) == 2
    assert events[0].begin_datetime == events[1].begin_datetime


@pytest.mark.asyncio
async def test_event_sort_by_date(event_model_generator):
    today = datetime(
        2021,
        5,
        10,
        hour=0,
        minute=0,
        second=0,
        microsecond=0,
        tzinfo=timezone(timedelta(hours=1)),
    )
    today_utc = datetime(
        2021,
        5,
        9,
        hour=23,
        minute=0,
        second=0,
        microsecond=0,
        tzinfo=timezone(timedelta(hours=0)),
    )
    event_yesterday = event_model_generator(idx=1, begin_date=today - timedelta(days=1))
    event_today = event_model_generator(idx=2, begin_date=today)
    event_tomorrow = event_model_generator(idx=3, begin_date=today + timedelta(days=1))

    await event_yesterday.save()
    await event_today.save()
    await event_tomorrow.save()

    events = await Event.all().order_by("begin_datetime")

    assert len(events) == 3

    assert (
        events[0].begin_datetime < events[1].begin_datetime < events[2].begin_datetime
    )

    assert events[0].begin_datetime == today_utc - timedelta(days=1)
    assert events[1].begin_datetime == today_utc
    assert events[2].begin_datetime == today_utc + timedelta(days=1)


@pytest.mark.asyncio
async def test_mobilizon_event_to_model(event):
    event_model = event.to_model()
    await event_model.save()

    event_db = await Event.all().first()
    begin_date_utc = arrow.Arrow(year=2021, month=1, day=1, hour=10, minute=30)
    begin_date_utc = begin_date_utc.astimezone(tortoise.timezone.get_default_timezone())

    assert event_db.name == "test event"
    assert event_db.description == "description of the event"
    assert event_db.begin_datetime == begin_date_utc
    assert event_db.end_datetime == begin_date_utc + timedelta(hours=1)
    assert event_db.mobilizon_link == "http://some_link.com/123"
    assert event_db.mobilizon_id == UUID(int=12345)
    assert event_db.thumbnail_link == "http://some_link.com/123.jpg"
    assert event_db.location == "location"


@pytest.mark.asyncio
async def test_mobilizon_event_from_model(
    event_model_generator, publication_model_generator, publisher_model_generator
):
    event_model = event_model_generator()
    await event_model.save()

    publisher_model = publisher_model_generator()
    await publisher_model.save()
    publisher_model_2 = publisher_model_generator(idx=2)
    await publisher_model_2.save()

    publication = publication_model_generator(
        status=PublicationStatus.FAILED,
        event_id=event_model.id,
        publisher_id=publisher_model.id,
    )
    await publication.save()
    publication_2 = publication_model_generator(
        status=PublicationStatus.COMPLETED,
        event_id=event_model.id,
        publisher_id=publisher_model_2.id,
    )
    await publication_2.save()

    event_db = (
        await Event.all()
        .prefetch_related("publications")
        .prefetch_related("publications__publisher")
        .first()
    )
    event = MobilizonEvent.from_model(event=event_db)

    begin_date_utc = arrow.Arrow(year=2021, month=1, day=1, hour=11, minute=30)

    assert event.name == "event_1"
    assert event.description == "desc_1"
    assert event.begin_datetime == begin_date_utc
    assert event.end_datetime == begin_date_utc.shift(hours=2)
    assert event.mobilizon_link == "moblink_1"
    assert event.mobilizon_id == UUID(int=1)
    assert event.thumbnail_link == "thumblink_1"
    assert event.location == ", loc_1, "
    assert event.publication_time[publisher_model.name] == publication.timestamp
    assert event.status == EventPublicationStatus.PARTIAL


@pytest.mark.parametrize(
    "statuses, expected_result",
    [
        (
            [PublicationStatus.FAILED, PublicationStatus.COMPLETED],
            EventPublicationStatus.PARTIAL,
        ),
        (
            [PublicationStatus.FAILED, PublicationStatus.FAILED],
            EventPublicationStatus.FAILED,
        ),
        (
            [PublicationStatus.COMPLETED, PublicationStatus.COMPLETED],
            EventPublicationStatus.COMPLETED,
        ),
        ([PublicationStatus.FAILED], EventPublicationStatus.FAILED),
        ([], EventPublicationStatus.WAITING),
    ],
)
@pytest.mark.asyncio
async def test_mobilizon_event_compute_status_partial(
    event_model_generator,
    publication_model_generator,
    publisher_model_generator,
    statuses,
    expected_result,
):
    event_model = event_model_generator()
    await event_model.save()
    publications = []
    for status in statuses:
        publisher_model = publisher_model_generator()
        await publisher_model.save()

        publication = publication_model_generator(
            status=status, event_id=event_model.id, publisher_id=publisher_model.id,
        )
        await publication.save()
        publications.append(publication)
    assert MobilizonEvent._compute_event_status(publications) == expected_result