mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-11-11 00:27:01 +01:00
Krzysztof kuhy Rudnicki
cf44e54215
cups_queue.py 424 -> 245. display_cups_queue_status, _offer_queue_fix, the _dwj_* choices and the _handle_* dispatchers move to _queue_fix.py. The _cups_* action primitives stay put: test_cups_queue.py patches subprocess and time on cups_queue, and those primitives are their only consumers, so moving them would have left subprocess.run unpatched and free to restart the real CUPS service mid-suite. test_cups_queue.py therefore needs no edit; only part2/part3 repoint their MOD. Pass count is 402 before and after. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_011w2Pj4Ek3ZNMQVYcFHgXDY
279 lines
11 KiB
Python
279 lines
11 KiB
Python
"""Tests for brother_printer._queue_fix module - part 2 (interactive fix)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from io import StringIO
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from python_pkg.brother_printer._queue_fix import (
|
|
_dwj_cancel_and_enable,
|
|
_dwj_cancel_only,
|
|
_dwj_enable_only,
|
|
_dwj_restart_and_enable,
|
|
_dwj_restart_only,
|
|
_handle_backend_errors_only,
|
|
_handle_disabled_no_jobs,
|
|
_handle_disabled_with_jobs,
|
|
_handle_enabled_with_jobs,
|
|
_offer_queue_fix,
|
|
)
|
|
from python_pkg.brother_printer.data_classes import CUPSJob, CUPSQueueStatus
|
|
|
|
MOD = "python_pkg.brother_printer._queue_fix"
|
|
|
|
|
|
# ── _offer_queue_fix ─────────────────────────────────────────────────
|
|
|
|
|
|
class TestOfferQueueFix:
|
|
"""Tests for _offer_queue_fix menu routing."""
|
|
|
|
@patch(f"{MOD}._handle_disabled_with_jobs")
|
|
@patch(f"{MOD}._prompt", return_value="1")
|
|
def test_disabled_with_jobs(self, p: MagicMock, mock_handler: MagicMock) -> None:
|
|
queue = CUPSQueueStatus(
|
|
printer_name="B",
|
|
enabled=False,
|
|
jobs=[CUPSJob("j1", "alice", "1024", "Mon")],
|
|
)
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_offer_queue_fix(queue)
|
|
mock_handler.assert_called_once_with(queue, "1")
|
|
|
|
@patch(f"{MOD}._handle_disabled_no_jobs")
|
|
@patch(f"{MOD}._prompt", return_value="2")
|
|
def test_disabled_no_jobs(self, p: MagicMock, mock_handler: MagicMock) -> None:
|
|
queue = CUPSQueueStatus(printer_name="B", enabled=False)
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_offer_queue_fix(queue)
|
|
mock_handler.assert_called_once_with(queue, "2")
|
|
|
|
@patch(f"{MOD}._handle_enabled_with_jobs")
|
|
@patch(f"{MOD}._prompt", return_value="1")
|
|
def test_enabled_with_jobs(self, p: MagicMock, mock_handler: MagicMock) -> None:
|
|
queue = CUPSQueueStatus(
|
|
printer_name="B",
|
|
enabled=True,
|
|
jobs=[CUPSJob("j1", "alice", "1024", "Mon")],
|
|
)
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_offer_queue_fix(queue)
|
|
mock_handler.assert_called_once_with(queue, "1")
|
|
|
|
@patch(f"{MOD}._handle_backend_errors_only")
|
|
@patch(f"{MOD}._prompt", return_value="1")
|
|
def test_backend_errors_only(self, p: MagicMock, mock_handler: MagicMock) -> None:
|
|
queue = CUPSQueueStatus(printer_name="B", enabled=True)
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_offer_queue_fix(queue)
|
|
mock_handler.assert_called_once_with("1")
|
|
|
|
|
|
# ── _dwj_* action functions ─────────────────────────────────────────
|
|
|
|
|
|
class TestDwjEnableOnly:
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
def test_success(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_enable_only("B")
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=False)
|
|
def test_failure(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_enable_only("B")
|
|
|
|
|
|
class TestDwjCancelAndEnable:
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=True)
|
|
def test_success(self, c: MagicMock, e: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_cancel_and_enable("B")
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=False)
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=True)
|
|
def test_enable_fails(self, c: MagicMock, e: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_cancel_and_enable("B")
|
|
|
|
|
|
class TestDwjCancelOnly:
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=True)
|
|
def test_success(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_cancel_only("B")
|
|
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=False)
|
|
def test_failure(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_cancel_only("B")
|
|
|
|
|
|
class TestDwjRestartOnly:
|
|
@patch(f"{MOD}._cups_restart_service", return_value=True)
|
|
def test_success(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_restart_only("B")
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=False)
|
|
def test_failure(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_restart_only("B")
|
|
|
|
|
|
class TestDwjRestartAndEnable:
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
@patch(f"{MOD}._cups_restart_service", return_value=True)
|
|
def test_success(self, r: MagicMock, e: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_restart_and_enable("B")
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=False)
|
|
def test_restart_fails(self, r: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_dwj_restart_and_enable("B")
|
|
|
|
|
|
# ── _handle_disabled_with_jobs ───────────────────────────────────────
|
|
|
|
|
|
class TestHandleDisabledWithJobs:
|
|
"""Tests for _handle_disabled_with_jobs dispatch."""
|
|
|
|
def _make_queue(self) -> CUPSQueueStatus:
|
|
return CUPSQueueStatus(
|
|
printer_name="B",
|
|
enabled=False,
|
|
jobs=[CUPSJob("j1", "alice", "1024", "Mon")],
|
|
)
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
def test_choice_1(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_with_jobs(self._make_queue(), "1")
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=True)
|
|
def test_choice_2(self, c: MagicMock, e: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_with_jobs(self._make_queue(), "2")
|
|
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=True)
|
|
def test_choice_3(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_with_jobs(self._make_queue(), "3")
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=True)
|
|
def test_choice_4(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_with_jobs(self._make_queue(), "4")
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
@patch(f"{MOD}._cups_restart_service", return_value=True)
|
|
def test_choice_5(self, r: MagicMock, e: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_with_jobs(self._make_queue(), "5")
|
|
|
|
def test_choice_6_no_action(self) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_with_jobs(self._make_queue(), "6")
|
|
|
|
def test_invalid_choice(self) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_with_jobs(self._make_queue(), "99")
|
|
|
|
|
|
# ── _handle_disabled_no_jobs ─────────────────────────────────────────
|
|
|
|
|
|
class TestHandleDisabledNoJobs:
|
|
"""Tests for _handle_disabled_no_jobs."""
|
|
|
|
def _make_queue(self) -> CUPSQueueStatus:
|
|
return CUPSQueueStatus(printer_name="B", enabled=False)
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
def test_choice_1_enable(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_no_jobs(self._make_queue(), "1")
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=False)
|
|
def test_choice_1_enable_fails(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_no_jobs(self._make_queue(), "1")
|
|
|
|
@patch(f"{MOD}._cups_enable_printer", return_value=True)
|
|
@patch(f"{MOD}._cups_restart_service", return_value=True)
|
|
def test_choice_2_restart(self, r: MagicMock, e: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_no_jobs(self._make_queue(), "2")
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=False)
|
|
def test_choice_2_restart_fails(self, r: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_no_jobs(self._make_queue(), "2")
|
|
|
|
def test_choice_3_no_action(self) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_disabled_no_jobs(self._make_queue(), "3")
|
|
|
|
|
|
# ── _handle_enabled_with_jobs ────────────────────────────────────────
|
|
|
|
|
|
class TestHandleEnabledWithJobs:
|
|
"""Tests for _handle_enabled_with_jobs."""
|
|
|
|
def _make_queue(self) -> CUPSQueueStatus:
|
|
return CUPSQueueStatus(
|
|
printer_name="B",
|
|
enabled=True,
|
|
jobs=[CUPSJob("j1", "alice", "1024", "Mon")],
|
|
)
|
|
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=True)
|
|
def test_choice_1_cancel(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_enabled_with_jobs(self._make_queue(), "1")
|
|
|
|
@patch(f"{MOD}._cups_cancel_all_jobs", return_value=False)
|
|
def test_choice_1_cancel_fails(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_enabled_with_jobs(self._make_queue(), "1")
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=True)
|
|
def test_choice_2_restart(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_enabled_with_jobs(self._make_queue(), "2")
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=False)
|
|
def test_choice_2_restart_fails(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_enabled_with_jobs(self._make_queue(), "2")
|
|
|
|
def test_choice_3_no_action(self) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_enabled_with_jobs(self._make_queue(), "3")
|
|
|
|
|
|
# ── _handle_backend_errors_only ──────────────────────────────────────
|
|
|
|
|
|
class TestHandleBackendErrorsOnly:
|
|
"""Tests for _handle_backend_errors_only."""
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=True)
|
|
def test_choice_1_restart(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_backend_errors_only("1")
|
|
|
|
@patch(f"{MOD}._cups_restart_service", return_value=False)
|
|
def test_choice_1_restart_fails(self, m: MagicMock) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_backend_errors_only("1")
|
|
|
|
def test_choice_2_no_action(self) -> None:
|
|
with patch("sys.stdout", new_callable=StringIO):
|
|
_handle_backend_errors_only("2")
|