diff --git a/mobilizon_reshare/main/retry.py b/mobilizon_reshare/main/retry.py new file mode 100644 index 0000000..1a210f5 --- /dev/null +++ b/mobilizon_reshare/main/retry.py @@ -0,0 +1,43 @@ +import logging +from uuid import UUID + +from mobilizon_reshare.publishers.coordinator import ( + PublisherCoordinator, + PublicationFailureNotifiersCoordinator, +) +from mobilizon_reshare.publishers.exceptions import EventNotFound +from mobilizon_reshare.storage.query.read import get_failed_publications_for_event +from mobilizon_reshare.storage.query.write import save_publication_report + +logger = logging.getLogger(__name__) + + +async def retry_event_publications(event_id): + + failed_publications = await get_failed_publications_for_event(event_id) + if not failed_publications: + logger.info("No failed publications found.") + return + + logger.info(f"Found {len(failed_publications)} publications.") + return PublisherCoordinator(failed_publications).run() + + +async def retry(mobilizon_event_id: UUID = None): + if mobilizon_event_id is None: + raise NotImplementedError( + "Autonomous retry not implemented yet, please specify an event_id" + ) + try: + reports = await retry_event_publications(mobilizon_event_id) + except EventNotFound as e: + logger.debug(e, exc_info=True) + logger.error(f"Event with id {mobilizon_event_id} not found") + return + + if not reports: + return + await save_publication_report(reports) + for report in reports.reports: + if not report.succesful: + PublicationFailureNotifiersCoordinator(report,).notify_failure() diff --git a/mobilizon_reshare/publishers/exceptions.py b/mobilizon_reshare/publishers/exceptions.py index 0420daa..2142470 100644 --- a/mobilizon_reshare/publishers/exceptions.py +++ b/mobilizon_reshare/publishers/exceptions.py @@ -38,3 +38,7 @@ class ZulipError(PublisherError): class HTTPResponseError(PublisherError): """ Publisher receives a HTTP error""" + + +class EventNotFound(Exception): + """ Event is not present in the database """ diff --git a/mobilizon_reshare/storage/query/read.py b/mobilizon_reshare/storage/query/read.py index 2991d60..91e4be9 100644 --- a/mobilizon_reshare/storage/query/read.py +++ b/mobilizon_reshare/storage/query/read.py @@ -1,3 +1,4 @@ +from functools import partial from typing import Iterable, Optional from uuid import UUID @@ -10,6 +11,7 @@ from mobilizon_reshare.models.event import Event from mobilizon_reshare.models.publication import Publication, PublicationStatus from mobilizon_reshare.publishers import get_active_publishers from mobilizon_reshare.publishers.abstract import EventPublication +from mobilizon_reshare.publishers.exceptions import EventNotFound from mobilizon_reshare.storage.query import CONNECTION_NAME @@ -82,7 +84,9 @@ async def prefetch_event_relations(queryset: QuerySet[Event]) -> list[Event]: ) -async def prefetch_publication_relations(queryset: QuerySet[Publication]) -> list[Publication]: +async def prefetch_publication_relations( + queryset: QuerySet[Publication], +) -> list[Publication]: return ( await queryset.prefetch_related("publisher", "event") .order_by("timestamp") @@ -164,3 +168,28 @@ async def build_publications(event: MobilizonEvent) -> list[EventPublication]: for name in get_active_publishers() ] return list(EventPublication.from_orm(m, event) for m in models) + + +@atomic(CONNECTION_NAME) +async def get_event(event_mobilizon_id) -> None: + event = await Event.filter(mobilizon_id=event_mobilizon_id).first() + if not event: + raise EventNotFound(f"No event with mobilizon_id {event_mobilizon_id} found.") + await event.fetch_related("publications") + return event + + +@atomic(CONNECTION_NAME) +async def get_failed_publications_for_event(event_mobilizon_id): + event = await get_event(event_mobilizon_id) + failed_publications = list( + filter( + lambda publications: publications.status == PublicationStatus.FAILED, + event.publications, + ) + ) + for p in failed_publications: + await p.fetch_related("publisher") + return list( + map(partial(EventPublication.from_orm, event=event), failed_publications) + ) diff --git a/mobilizon_reshare/storage/query/write.py b/mobilizon_reshare/storage/query/write.py index a300a09..b5427bb 100644 --- a/mobilizon_reshare/storage/query/write.py +++ b/mobilizon_reshare/storage/query/write.py @@ -13,6 +13,33 @@ from mobilizon_reshare.storage.query import CONNECTION_NAME from mobilizon_reshare.storage.query.read import events_without_publications +async def upsert_publication(publication_report, event): + + publisher = await get_publisher_by_name( + name=publication_report.publication.publisher.name + ) + old_publication = await Publication.filter( + id=publication_report.publication.id + ).first() + if old_publication: + # I update the existing publication with the new report + old_publication.timestamp = arrow.now().datetime + old_publication.status = publication_report.status + old_publication.reason = publication_report.reason + + await old_publication.save(force_update=True) + else: + # I create a new publication + await Publication.create( + id=publication_report.publication.id, + event_id=event.id, + publisher_id=publisher.id, + status=publication_report.status, + reason=publication_report.reason, + timestamp=arrow.now().datetime, + ) + + @atomic(CONNECTION_NAME) async def save_publication_report( coordinator_report: PublisherCoordinatorReport, @@ -24,17 +51,7 @@ async def save_publication_report( event = await Event.filter( mobilizon_id=publication_report.publication.event.mobilizon_id ).first() - publisher = await Publisher.filter( - name=publication_report.publication.publisher.name - ).first() - await Publication.create( - id=publication_report.publication.id, - event_id=event.id, - publisher_id=publisher.id, - status=publication_report.status, - reason=publication_report.reason, - timestamp=arrow.now().datetime, - ) + await upsert_publication(publication_report, event) @atomic(CONNECTION_NAME) @@ -69,11 +86,13 @@ async def create_publisher(name: str, account_ref: Optional[str] = None) -> None @atomic(CONNECTION_NAME) -async def update_publishers( - names: Iterable[str], -) -> None: +async def update_publishers(names: Iterable[str],) -> None: names = set(names) known_publisher_names = set(p.name for p in await Publisher.all()) for name in names.difference(known_publisher_names): logging.info(f"Creating {name} publisher") await create_publisher(name) + + +async def get_publisher_by_name(name): + return await Publisher.filter(name=name).first() diff --git a/poetry.lock b/poetry.lock index 2a930ca..5b9d97f 100644 --- a/poetry.lock +++ b/poetry.lock @@ -677,28 +677,12 @@ markdownify = [ {file = "markdownify-0.9.4.tar.gz", hash = "sha256:a32d7d6b0801f81d12a44533093a5681b402c6b7e9a7049512d26697517eb795"}, ] markupsafe = [ - {file = "MarkupSafe-2.0.1-cp310-cp310-macosx_10_9_universal2.whl", hash = "sha256:d8446c54dc28c01e5a2dbac5a25f071f6653e6e40f3a8818e8b45d790fe6ef53"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-macosx_10_9_x86_64.whl", hash = "sha256:36bc903cbb393720fad60fc28c10de6acf10dc6cc883f3e24ee4012371399a38"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:2d7d807855b419fc2ed3e631034685db6079889a1f01d5d9dac950f764da3dad"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-manylinux_2_5_i686.manylinux1_i686.manylinux_2_12_i686.manylinux2010_i686.whl", hash = "sha256:add36cb2dbb8b736611303cd3bfcee00afd96471b09cda130da3581cbdc56a6d"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:168cd0a3642de83558a5153c8bd34f175a9a6e7f6dc6384b9655d2697312a646"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-musllinux_1_1_aarch64.whl", hash = "sha256:4dc8f9fb58f7364b63fd9f85013b780ef83c11857ae79f2feda41e270468dd9b"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-musllinux_1_1_i686.whl", hash = "sha256:20dca64a3ef2d6e4d5d615a3fd418ad3bde77a47ec8a23d984a12b5b4c74491a"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-musllinux_1_1_x86_64.whl", hash = "sha256:cdfba22ea2f0029c9261a4bd07e830a8da012291fbe44dc794e488b6c9bb353a"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-win32.whl", hash = "sha256:99df47edb6bda1249d3e80fdabb1dab8c08ef3975f69aed437cb69d0a5de1e28"}, - {file = "MarkupSafe-2.0.1-cp310-cp310-win_amd64.whl", hash = "sha256:e0f138900af21926a02425cf736db95be9f4af72ba1bb21453432a07f6082134"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-macosx_10_9_x86_64.whl", hash = "sha256:f9081981fe268bd86831e5c75f7de206ef275defcb82bc70740ae6dc507aee51"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux1_i686.whl", hash = "sha256:0955295dd5eec6cb6cc2fe1698f4c6d84af2e92de33fbcac4111913cd100a6ff"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux1_x86_64.whl", hash = "sha256:0446679737af14f45767963a1a9ef7620189912317d095f2d9ffa183a4d25d2b"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux2010_i686.whl", hash = "sha256:f826e31d18b516f653fe296d967d700fddad5901ae07c622bb3705955e1faa94"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux2010_x86_64.whl", hash = "sha256:fa130dd50c57d53368c9d59395cb5526eda596d3ffe36666cd81a44d56e48872"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux2014_aarch64.whl", hash = "sha256:905fec760bd2fa1388bb5b489ee8ee5f7291d692638ea5f67982d968366bef9f"}, - {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:bf5d821ffabf0ef3533c39c518f3357b171a1651c1ff6827325e4489b0e46c3c"}, - {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux_2_5_i686.manylinux1_i686.manylinux_2_12_i686.manylinux2010_i686.whl", hash = "sha256:0d4b31cc67ab36e3392bbf3862cfbadac3db12bdd8b02a2731f509ed5b829724"}, - {file = "MarkupSafe-2.0.1-cp36-cp36m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:baa1a4e8f868845af802979fcdbf0bb11f94f1cb7ced4c4b8a351bb60d108145"}, - {file = "MarkupSafe-2.0.1-cp36-cp36m-musllinux_1_1_aarch64.whl", hash = "sha256:deb993cacb280823246a026e3b2d81c493c53de6acfd5e6bfe31ab3402bb37dd"}, - {file = "MarkupSafe-2.0.1-cp36-cp36m-musllinux_1_1_i686.whl", hash = "sha256:63f3268ba69ace99cab4e3e3b5840b03340efed0948ab8f78d2fd87ee5442a4f"}, - {file = "MarkupSafe-2.0.1-cp36-cp36m-musllinux_1_1_x86_64.whl", hash = "sha256:8d206346619592c6200148b01a2142798c989edcb9c896f9ac9722a99d4e77e6"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-win32.whl", hash = "sha256:6c4ca60fa24e85fe25b912b01e62cb969d69a23a5d5867682dd3e80b5b02581d"}, {file = "MarkupSafe-2.0.1-cp36-cp36m-win_amd64.whl", hash = "sha256:b2f4bf27480f5e5e8ce285a8c8fd176c0b03e93dcc6646477d4630e83440c6a9"}, {file = "MarkupSafe-2.0.1-cp37-cp37m-macosx_10_9_x86_64.whl", hash = "sha256:0717a7390a68be14b8c793ba258e075c6f4ca819f15edfc2a3a027c823718567"}, @@ -707,27 +691,14 @@ markupsafe = [ {file = "MarkupSafe-2.0.1-cp37-cp37m-manylinux2010_i686.whl", hash = "sha256:d7f9850398e85aba693bb640262d3611788b1f29a79f0c93c565694658f4071f"}, {file = "MarkupSafe-2.0.1-cp37-cp37m-manylinux2010_x86_64.whl", hash = "sha256:6a7fae0dd14cf60ad5ff42baa2e95727c3d81ded453457771d02b7d2b3f9c0c2"}, {file = "MarkupSafe-2.0.1-cp37-cp37m-manylinux2014_aarch64.whl", hash = "sha256:b7f2d075102dc8c794cbde1947378051c4e5180d52d276987b8d28a3bd58c17d"}, - {file = "MarkupSafe-2.0.1-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:e9936f0b261d4df76ad22f8fee3ae83b60d7c3e871292cd42f40b81b70afae85"}, - {file = "MarkupSafe-2.0.1-cp37-cp37m-manylinux_2_5_i686.manylinux1_i686.manylinux_2_12_i686.manylinux2010_i686.whl", hash = "sha256:2a7d351cbd8cfeb19ca00de495e224dea7e7d919659c2841bbb7f420ad03e2d6"}, - {file = "MarkupSafe-2.0.1-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:60bf42e36abfaf9aff1f50f52644b336d4f0a3fd6d8a60ca0d054ac9f713a864"}, - {file = "MarkupSafe-2.0.1-cp37-cp37m-musllinux_1_1_aarch64.whl", hash = "sha256:d6c7ebd4e944c85e2c3421e612a7057a2f48d478d79e61800d81468a8d842207"}, - {file = "MarkupSafe-2.0.1-cp37-cp37m-musllinux_1_1_i686.whl", hash = "sha256:f0567c4dc99f264f49fe27da5f735f414c4e7e7dd850cfd8e69f0862d7c74ea9"}, - {file = "MarkupSafe-2.0.1-cp37-cp37m-musllinux_1_1_x86_64.whl", hash = "sha256:89c687013cb1cd489a0f0ac24febe8c7a666e6e221b783e53ac50ebf68e45d86"}, {file = "MarkupSafe-2.0.1-cp37-cp37m-win32.whl", hash = "sha256:a30e67a65b53ea0a5e62fe23682cfe22712e01f453b95233b25502f7c61cb415"}, {file = "MarkupSafe-2.0.1-cp37-cp37m-win_amd64.whl", hash = "sha256:611d1ad9a4288cf3e3c16014564df047fe08410e628f89805e475368bd304914"}, - {file = "MarkupSafe-2.0.1-cp38-cp38-macosx_10_9_universal2.whl", hash = "sha256:5bb28c636d87e840583ee3adeb78172efc47c8b26127267f54a9c0ec251d41a9"}, {file = "MarkupSafe-2.0.1-cp38-cp38-macosx_10_9_x86_64.whl", hash = "sha256:be98f628055368795d818ebf93da628541e10b75b41c559fdf36d104c5787066"}, {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux1_i686.whl", hash = "sha256:1d609f577dc6e1aa17d746f8bd3c31aa4d258f4070d61b2aa5c4166c1539de35"}, {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux1_x86_64.whl", hash = "sha256:7d91275b0245b1da4d4cfa07e0faedd5b0812efc15b702576d103293e252af1b"}, {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux2010_i686.whl", hash = "sha256:01a9b8ea66f1658938f65b93a85ebe8bc016e6769611be228d797c9d998dd298"}, {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux2010_x86_64.whl", hash = "sha256:47ab1e7b91c098ab893b828deafa1203de86d0bc6ab587b160f78fe6c4011f75"}, {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux2014_aarch64.whl", hash = "sha256:97383d78eb34da7e1fa37dd273c20ad4320929af65d156e35a5e2d89566d9dfb"}, - {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:6fcf051089389abe060c9cd7caa212c707e58153afa2c649f00346ce6d260f1b"}, - {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux_2_5_i686.manylinux1_i686.manylinux_2_12_i686.manylinux2010_i686.whl", hash = "sha256:5855f8438a7d1d458206a2466bf82b0f104a3724bf96a1c781ab731e4201731a"}, - {file = "MarkupSafe-2.0.1-cp38-cp38-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:3dd007d54ee88b46be476e293f48c85048603f5f516008bee124ddd891398ed6"}, - {file = "MarkupSafe-2.0.1-cp38-cp38-musllinux_1_1_aarch64.whl", hash = "sha256:aca6377c0cb8a8253e493c6b451565ac77e98c2951c45f913e0b52facdcff83f"}, - {file = "MarkupSafe-2.0.1-cp38-cp38-musllinux_1_1_i686.whl", hash = "sha256:04635854b943835a6ea959e948d19dcd311762c5c0c6e1f0e16ee57022669194"}, - {file = "MarkupSafe-2.0.1-cp38-cp38-musllinux_1_1_x86_64.whl", hash = "sha256:6300b8454aa6930a24b9618fbb54b5a68135092bc666f7b06901f897fa5c2fee"}, {file = "MarkupSafe-2.0.1-cp38-cp38-win32.whl", hash = "sha256:023cb26ec21ece8dc3907c0e8320058b2e0cb3c55cf9564da612bc325bed5e64"}, {file = "MarkupSafe-2.0.1-cp38-cp38-win_amd64.whl", hash = "sha256:984d76483eb32f1bcb536dc27e4ad56bba4baa70be32fa87152832cdd9db0833"}, {file = "MarkupSafe-2.0.1-cp39-cp39-macosx_10_9_universal2.whl", hash = "sha256:2ef54abee730b502252bcdf31b10dacb0a416229b72c18b19e24a4509f273d26"}, @@ -737,12 +708,6 @@ markupsafe = [ {file = "MarkupSafe-2.0.1-cp39-cp39-manylinux2010_i686.whl", hash = "sha256:4efca8f86c54b22348a5467704e3fec767b2db12fc39c6d963168ab1d3fc9135"}, {file = "MarkupSafe-2.0.1-cp39-cp39-manylinux2010_x86_64.whl", hash = "sha256:ab3ef638ace319fa26553db0624c4699e31a28bb2a835c5faca8f8acf6a5a902"}, {file = "MarkupSafe-2.0.1-cp39-cp39-manylinux2014_aarch64.whl", hash = "sha256:f8ba0e8349a38d3001fae7eadded3f6606f0da5d748ee53cc1dab1d6527b9509"}, - {file = "MarkupSafe-2.0.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:c47adbc92fc1bb2b3274c4b3a43ae0e4573d9fbff4f54cd484555edbf030baf1"}, - {file = "MarkupSafe-2.0.1-cp39-cp39-manylinux_2_5_i686.manylinux1_i686.manylinux_2_12_i686.manylinux2010_i686.whl", hash = "sha256:37205cac2a79194e3750b0af2a5720d95f786a55ce7df90c3af697bfa100eaac"}, - {file = "MarkupSafe-2.0.1-cp39-cp39-manylinux_2_5_x86_64.manylinux1_x86_64.manylinux_2_12_x86_64.manylinux2010_x86_64.whl", hash = "sha256:1f2ade76b9903f39aa442b4aadd2177decb66525062db244b35d71d0ee8599b6"}, - {file = "MarkupSafe-2.0.1-cp39-cp39-musllinux_1_1_aarch64.whl", hash = "sha256:4296f2b1ce8c86a6aea78613c34bb1a672ea0e3de9c6ba08a960efe0b0a09047"}, - {file = "MarkupSafe-2.0.1-cp39-cp39-musllinux_1_1_i686.whl", hash = "sha256:9f02365d4e99430a12647f09b6cc8bab61a6564363f313126f775eb4f6ef798e"}, - {file = "MarkupSafe-2.0.1-cp39-cp39-musllinux_1_1_x86_64.whl", hash = "sha256:5b6d930f030f8ed98e3e6c98ffa0652bdb82601e7a016ec2ab5d7ff23baa78d1"}, {file = "MarkupSafe-2.0.1-cp39-cp39-win32.whl", hash = "sha256:10f82115e21dc0dfec9ab5c0223652f7197feb168c940f3ef61563fc2d6beb74"}, {file = "MarkupSafe-2.0.1-cp39-cp39-win_amd64.whl", hash = "sha256:693ce3f9e70a6cf7d2fb9e6c9d8b204b6b39897a2c4a1aa65728d5ac97dcc1d8"}, {file = "MarkupSafe-2.0.1.tar.gz", hash = "sha256:594c67807fb16238b30c44bdf74f36c02cdf22d1c8cda91ef8a0ed8dabf5620a"}, diff --git a/tests/commands/test_retry.py b/tests/commands/test_retry.py new file mode 100644 index 0000000..2203d2c --- /dev/null +++ b/tests/commands/test_retry.py @@ -0,0 +1,83 @@ +import uuid +from logging import INFO + +import pytest + +from mobilizon_reshare.main.retry import retry +from mobilizon_reshare.models.publication import PublicationStatus, Publication + + +@pytest.mark.asyncio +async def test_retry_decision(): + with pytest.raises(NotImplementedError): + await retry() + + +@pytest.mark.parametrize( + "publisher_class", [pytest.lazy_fixture("mock_publisher_class")] +) +@pytest.mark.asyncio +async def test_retry( + event_with_failed_publication, + mock_publisher_config, + message_collector, + failed_publication, +): + assert failed_publication.status == PublicationStatus.FAILED + await retry(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_no_publications( + stored_event, mock_publisher_config, message_collector, caplog +): + with caplog.at_level(INFO): + await retry(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_missing_event(mock_publisher_config, message_collector, caplog): + event_id = uuid.uuid4() + with caplog.at_level(INFO): + await retry(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_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_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" diff --git a/tests/conftest.py b/tests/conftest.py index 1f845a8..e9de599 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -22,6 +22,7 @@ from mobilizon_reshare.publishers.abstract import ( AbstractEventFormatter, ) from mobilizon_reshare.publishers.exceptions import PublisherError, InvalidResponse +from mobilizon_reshare.storage.query.write import get_publisher_by_name from tests import today @@ -102,6 +103,14 @@ def event() -> MobilizonEvent: ) +@pytest.fixture +async def stored_event(event): + model = event.to_model() + await model.save() + await model.fetch_related("publications") + return model + + @pytest.fixture(scope="function", autouse=True) def initialize_db_tests() -> None: db_url = os.environ.get("TORTOISE_TEST_DB", "sqlite://:memory:") @@ -357,3 +366,23 @@ def mock_publisher_invalid_class(message_collector): def mock_publisher_invalid(mock_publisher_invalid_class): return mock_publisher_invalid_class() + + +@pytest.fixture +async def event_with_failed_publication( + stored_event, mock_publisher_config, failed_publication +): + return stored_event + + +@pytest.fixture +async def failed_publication(stored_event): + + p = Publication( + event=stored_event, + status=PublicationStatus.FAILED, + timestamp=arrow.now().datetime, + publisher=await get_publisher_by_name("mock"), + ) + await p.save() + return p