Mobilizon-Reshare-condividi-sui-social/tests/event/test_strategies.py

from datetime import datetime, timedelta

import pytest

from tests.event.event_selector import SelectNextEventStrategy


@pytest.mark.parametrize(
    "desired_break_window,days_passed_from_publication", [[2, 1], [3, 2]]
)
def test_window_simple_no_event(
    event_generator, desired_break_window, days_passed_from_publication
):
    "Testing that the break between events is respected"
    unpublished_events = [
        event_generator(
            published=False,
            begin_date=datetime(year=2021, month=1, day=5, hour=11, minute=30),
        )
    ]
    published_events = [
        event_generator(
            published=True,
            publication_time=datetime.now()
            - timedelta(days=days_passed_from_publication),
        )
    ]

    selected_event = SelectNextEventStrategy(
        minimum_break_between_events=timedelta(days=desired_break_window)
    ).select(published_events, unpublished_events)
    assert selected_event is None


@pytest.mark.parametrize(
    "desired_break_window,days_passed_from_publication", [[1, 2], [2, 10], [4, 4]]
)
def test_window_simple_event_found(
    event_generator, desired_break_window, days_passed_from_publication
):
    "Testing that the break between events is respected"
    unpublished_events = [
        event_generator(
            published=False,
            begin_date=datetime(year=2021, month=1, day=5, hour=11, minute=30),
        )
    ]
    published_events = [
        event_generator(
            published=True,
            publication_time=datetime.now()
            - timedelta(days=days_passed_from_publication),
        )
    ]

    selected_event = SelectNextEventStrategy(
        minimum_break_between_events=timedelta(days=desired_break_window)
    ).select(published_events, unpublished_events)
    assert selected_event is unpublished_events[0]