retry command (#119)
* added failed publication retrieval * completed retry command * added more error handling * added atomic to query
This commit is contained in:
Simone Robutti
GitHub
parent
43d653f37c
commit
5bce70b4a1
|
|
@ -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()
|
||||
|
|
@ -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 """
|
||||
|
|
|
|||
|
|
@ -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)
|
||||
)
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
|
|||
|
|
@ -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"},
|
||||
|
|
|
|||
|
|
@ -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"
|
||||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue