161 lines
5.0 KiB
Python
161 lines
5.0 KiB
Python
import uuid
|
|
from logging import INFO, ERROR
|
|
|
|
import pytest
|
|
|
|
from mobilizon_reshare.main.retry import retry_event, retry_publication
|
|
from mobilizon_reshare.models.publication import PublicationStatus, Publication
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_retry_decision():
|
|
with pytest.raises(NotImplementedError):
|
|
await retry_event()
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_retry_event(
|
|
event_with_failed_publication,
|
|
mock_publisher_config,
|
|
message_collector,
|
|
failed_publication,
|
|
):
|
|
await retry_event(event_with_failed_publication.mobilizon_id)
|
|
p = await Publication.filter(id=failed_publication.id).first()
|
|
assert p.status == PublicationStatus.COMPLETED, p.id
|
|
assert len(message_collector) == 1
|
|
assert message_collector[0] == "test event|description of the event"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_retry_event_no_publications(
|
|
stored_event, mock_publisher_config, message_collector, caplog
|
|
):
|
|
with caplog.at_level(INFO):
|
|
await retry_event(stored_event.mobilizon_id)
|
|
assert "No failed publications found" in caplog.text
|
|
assert len(message_collector) == 0
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_retry_event_missing_event(
|
|
mock_publisher_config, message_collector, caplog
|
|
):
|
|
event_id = uuid.uuid4()
|
|
with caplog.at_level(INFO):
|
|
await retry_event(event_id)
|
|
assert f"Event with id {event_id} not found" in caplog.text
|
|
|
|
assert len(message_collector) == 0
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_retry_event_mixed_publications(
|
|
event_with_failed_publication,
|
|
mock_publisher_config,
|
|
message_collector,
|
|
failed_publication,
|
|
publication_model_generator,
|
|
):
|
|
p = publication_model_generator(
|
|
event_id=event_with_failed_publication.id,
|
|
status=PublicationStatus.COMPLETED,
|
|
publisher_id=mock_publisher_config.id,
|
|
)
|
|
await p.save()
|
|
|
|
assert failed_publication.status == PublicationStatus.FAILED
|
|
await retry_event(event_with_failed_publication.mobilizon_id)
|
|
p = await Publication.filter(id=failed_publication.id).first()
|
|
assert p.status == PublicationStatus.COMPLETED, p.id
|
|
assert len(message_collector) == 1
|
|
assert message_collector[0] == "test event|description of the event"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_retry_publication(
|
|
event_with_failed_publication,
|
|
mock_publisher_config,
|
|
message_collector,
|
|
failed_publication: Publication,
|
|
):
|
|
await retry_publication(failed_publication.id)
|
|
p = await Publication.filter(id=failed_publication.id).first()
|
|
assert p.status == PublicationStatus.COMPLETED, p.id
|
|
assert len(message_collector) == 1
|
|
assert message_collector[0] == "test event|description of the event"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_retry_publication_missing(
|
|
mock_publisher_config, message_collector, caplog
|
|
):
|
|
publication_id = uuid.uuid4()
|
|
with caplog.at_level(INFO):
|
|
await retry_publication(publication_id)
|
|
assert f"Publication {publication_id} not found.\n" in caplog.text
|
|
assert len(message_collector) == 0
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_invalid_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_event_retry_failure(
|
|
event_with_failed_publication,
|
|
mock_publisher_config,
|
|
mock_notifier_config,
|
|
failed_publication: Publication,
|
|
):
|
|
|
|
report = await retry_event(event_with_failed_publication.mobilizon_id)
|
|
assert len(report.reports) == 1
|
|
assert (
|
|
f"Publication {failed_publication.id} failed with status: FAILED.\nReason: credentials error"
|
|
in report.reports[0].get_failure_message()
|
|
)
|
|
|
|
p = await Publication.filter(id=failed_publication.id).first()
|
|
assert p.status == PublicationStatus.FAILED, p.id
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"publisher_class", [pytest.lazy_fixture("mock_publisher_invalid_class")]
|
|
)
|
|
@pytest.mark.asyncio
|
|
async def test_publication_retry_failure(
|
|
event_with_failed_publication,
|
|
mock_publisher_config,
|
|
mock_notifier_config,
|
|
failed_publication: Publication,
|
|
caplog,
|
|
):
|
|
|
|
with caplog.at_level(ERROR):
|
|
report = await retry_publication(failed_publication.id)
|
|
assert len(report.reports) == 1
|
|
assert (
|
|
f"Publication {failed_publication.id} failed with status: FAILED.\nReason: credentials error"
|
|
in report.reports[0].get_failure_message()
|
|
)
|
|
p = await Publication.filter(id=failed_publication.id).first()
|
|
assert p.status == PublicationStatus.FAILED, p.id
|