mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-07-04 16:23:04 +02:00
279 lines
11 KiB
Python
279 lines
11 KiB
Python
|
"""Tests for brother_printer.cups_queue module - part 2 (interactive fix)."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from io import StringIO
|
||
|
|
from unittest.mock import MagicMock, patch
|
||
|
|
|
||
|
|
from python_pkg.brother_printer.cups_queue 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.cups_queue"
|
||
|
|
|
||
|
|
|
||
|
|
# ── _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")
|