mirror of
https://github.com/calibrain/shelfmark.git
synced 2026-11-14 09:58:26 +00:00
splitsec2
e27fed1d40
#1429 made the first terminal status win, but the guard only refused terminal to terminal changes. A worker's poll that reports DOWNLOADING after a cancel still overwrote CANCELLED (the poll loop calls the status callback before it waits on the cancel flag), and the COMPLETE or ERROR that followed was then accepted as a first terminal status, so the terminal hook ran a second time. This refuses any status change away from a terminal status. `add()` and `enqueue_existing()` reset the status themselves, so retries are unaffected. The new test cancels a downloading task, sends DOWNLOADING and then COMPLETE, and expects CANCELLED and one hook call. It fails on main and passes here. Full non-e2e suite passes, ruff and basedpyright are clean. Found in a Claude pass over 1.5.0, same as my other small fixes from it.
126 lines
4.3 KiB
Python
126 lines
4.3 KiB
Python
"""Tests for queue hook failure handling and cancellation."""
|
|
|
|
from unittest.mock import patch
|
|
|
|
from shelfmark.core.models import DownloadTask, QueueStatus
|
|
from shelfmark.core.queue import BookQueue
|
|
|
|
|
|
def _make_task(task_id: str = "task-1") -> DownloadTask:
|
|
return DownloadTask(
|
|
task_id=task_id,
|
|
source="direct_download",
|
|
title="Example Title",
|
|
user_id=1,
|
|
username="alice",
|
|
)
|
|
|
|
|
|
def test_add_logs_queue_hook_failures():
|
|
queue = BookQueue()
|
|
|
|
def broken_hook(task_id: str, task: DownloadTask) -> None:
|
|
raise RuntimeError("boom")
|
|
|
|
queue.set_queue_hook(broken_hook)
|
|
|
|
with patch("shelfmark.core.queue.logger.warning") as mock_warning:
|
|
assert queue.add(_make_task()) is True
|
|
|
|
mock_warning.assert_called_once()
|
|
args = mock_warning.call_args.args
|
|
assert args[0] == "Queue hook failed while adding task %s: %s"
|
|
assert args[1] == "task-1"
|
|
assert str(args[2]) == "boom"
|
|
|
|
|
|
def test_enqueue_existing_logs_queue_hook_failures():
|
|
queue = BookQueue()
|
|
assert queue.add(_make_task("task-2")) is True
|
|
|
|
def broken_hook(task_id: str, task: DownloadTask) -> None:
|
|
raise RuntimeError("boom")
|
|
|
|
queue.set_queue_hook(broken_hook)
|
|
|
|
with patch("shelfmark.core.queue.logger.warning") as mock_warning:
|
|
assert queue.enqueue_existing("task-2") is True
|
|
|
|
mock_warning.assert_called_once()
|
|
args = mock_warning.call_args.args
|
|
assert args[0] == "Queue hook failed while requeueing task %s: %s"
|
|
assert args[1] == "task-2"
|
|
assert str(args[2]) == "boom"
|
|
|
|
|
|
def _downloading_task(queue: BookQueue, task_id: str) -> list[QueueStatus]:
|
|
"""Put a task in DOWNLOADING and return the list its terminal hook appends to."""
|
|
assert queue.add(_make_task(task_id)) is True
|
|
assert queue.get_next() is not None
|
|
queue.update_status(task_id, QueueStatus.DOWNLOADING)
|
|
|
|
terminal_events: list[QueueStatus] = []
|
|
queue.set_terminal_status_hook(lambda _task_id, status, _task: terminal_events.append(status))
|
|
return terminal_events
|
|
|
|
|
|
def test_cancel_does_not_overwrite_a_download_that_finished_first():
|
|
"""A cancel that arrives after the download completed is refused and changes nothing."""
|
|
queue = BookQueue()
|
|
terminal_events = _downloading_task(queue, "race-task")
|
|
|
|
queue.update_status("race-task", QueueStatus.COMPLETE)
|
|
|
|
assert queue.cancel_download("race-task") is False
|
|
assert queue.get_task_status("race-task") == QueueStatus.COMPLETE
|
|
assert terminal_events == [QueueStatus.COMPLETE]
|
|
|
|
|
|
def test_a_late_complete_does_not_overwrite_a_cancel():
|
|
"""The worker checks its cancel flag and then writes COMPLETE; a cancel in between wins."""
|
|
queue = BookQueue()
|
|
terminal_events = _downloading_task(queue, "race-task")
|
|
|
|
assert queue.cancel_download("race-task") is True
|
|
queue.update_status("race-task", QueueStatus.COMPLETE)
|
|
|
|
assert queue.get_task_status("race-task") == QueueStatus.CANCELLED
|
|
assert terminal_events == [QueueStatus.CANCELLED]
|
|
|
|
|
|
def test_a_late_progress_update_does_not_reopen_a_cancelled_task():
|
|
"""A poll that reports DOWNLOADING after the cancel must not reopen the task.
|
|
|
|
Without this the next COMPLETE or ERROR is accepted as a first terminal status and
|
|
the terminal hook fires a second time.
|
|
"""
|
|
queue = BookQueue()
|
|
terminal_events = _downloading_task(queue, "race-task")
|
|
|
|
assert queue.cancel_download("race-task") is True
|
|
queue.update_status("race-task", QueueStatus.DOWNLOADING)
|
|
queue.update_status("race-task", QueueStatus.COMPLETE)
|
|
|
|
assert queue.get_task_status("race-task") == QueueStatus.CANCELLED
|
|
assert terminal_events == [QueueStatus.CANCELLED]
|
|
|
|
|
|
def test_a_task_that_ended_in_error_is_not_completed_afterwards():
|
|
queue = BookQueue()
|
|
terminal_events = _downloading_task(queue, "race-task")
|
|
|
|
queue.update_status("race-task", QueueStatus.ERROR)
|
|
queue.update_status("race-task", QueueStatus.COMPLETE)
|
|
|
|
assert queue.get_task_status("race-task") == QueueStatus.ERROR
|
|
assert terminal_events == [QueueStatus.ERROR]
|
|
|
|
|
|
def test_a_terminal_task_can_still_be_requeued():
|
|
queue = BookQueue()
|
|
_downloading_task(queue, "race-task")
|
|
queue.update_status("race-task", QueueStatus.ERROR)
|
|
|
|
assert queue.enqueue_existing("race-task") is True
|
|
assert queue.get_task_status("race-task") == QueueStatus.QUEUED
|