Cover snapshot CLI branches

Author: zzstoatzz

tests/cli/test_snapshot.py

@@ -22,6 +22,44 @@ async def empty_docket(docket: Docket, aiolib: str):
     await docket.cancel("initial")
 
 
+class RecordingConsole:
+    def __init__(self) -> None:
+        self.calls: list[tuple[tuple[object, ...], dict[str, object]]] = []
+
+    def print(self, *args: object, **kwargs: object) -> None:
+        self.calls.append((args, kwargs))
+
+
+def _stub_snapshot_docket(
+    monkeypatch: MonkeyPatch,
+    snapshot_obj: DocketSnapshot,
+    registered: list[str],
+) -> None:
+    class FakeDocket:
+        def __init__(self, name: str, url: str) -> None:
+            self.name = name
+            self.url = url
+
+        async def __aenter__(self) -> "FakeDocket":
+            return self
+
+        async def __aexit__(
+            self,
+            exc_type: type[BaseException] | None,
+            exc: BaseException | None,
+            tb: TracebackType | None,
+        ) -> bool:
+            return False
+
+        def register_collection(self, task_path: str) -> None:
+            registered.append(task_path)
+
+        async def snapshot(self) -> DocketSnapshot:
+            return snapshot_obj
+
+    monkeypatch.setattr("docket.cli.Docket", FakeDocket)
+
+
 async def test_snapshot_empty_docket(docket: Docket, runner: CliRunner):
     """Should show an empty snapshot when no tasks are scheduled"""
     result = await asyncio.get_running_loop().run_in_executor(
@@ -294,37 +332,7 @@ def test_snapshot_cli_renders_stats_without_backend(monkeypatch: MonkeyPatch):
     )
 
     registered: list[str] = []
-
-    class FakeDocket:
-        def __init__(self, name: str, url: str) -> None:
-            self.name = name
-            self.url = url
-
-        async def __aenter__(self) -> "FakeDocket":
-            return self
-
-        async def __aexit__(
-            self,
-            exc_type: type[BaseException] | None,
-            exc: BaseException | None,
-            tb: TracebackType | None,
-        ) -> bool:
-            return False
-
-        def register_collection(self, task_path: str) -> None:
-            registered.append(task_path)
-
-        async def snapshot(self) -> DocketSnapshot:
-            return snapshot_obj
-
-    monkeypatch.setattr("docket.cli.Docket", FakeDocket)
-
-    class RecordingConsole:
-        def __init__(self) -> None:
-            self.calls: list[tuple[tuple[object, ...], dict[str, object]]] = []
-
-        def print(self, *args: object, **kwargs: object) -> None:
-            self.calls.append((args, kwargs))
+    _stub_snapshot_docket(monkeypatch, snapshot_obj, registered)
 
     recorder = RecordingConsole()
     monkeypatch.setattr("docket.cli.Console", lambda: recorder)
@@ -343,6 +351,61 @@ def print(self, *args: object, **kwargs: object) -> None:
     )
 
 
+def test_snapshot_cli_without_stats_skips_table(monkeypatch: MonkeyPatch):
+    """Calling snapshot CLI without --stats should omit the second table."""
+    now = datetime(2023, 1, 1, 12, 0, tzinfo=timezone.utc)
+    snapshot_obj = DocketSnapshot(
+        taken=now,
+        total_tasks=0,
+        future=[],
+        running=[],
+        workers=[
+            WorkerInfo(
+                name="worker-1",
+                last_seen=now,
+                tasks=set(),
+            )
+        ],
+    )
+
+    registered: list[str] = []
+    _stub_snapshot_docket(monkeypatch, snapshot_obj, registered)
+
+    recorder = RecordingConsole()
+    monkeypatch.setattr("docket.cli.Console", lambda: recorder)
+
+    snapshot_command()
+
+    assert registered == ["docket.tasks:standard_tasks"]
+    tables = [args[0] for args, _ in recorder.calls if args]
+    # Only the summary table should be printed.
+    assert len([t for t in tables if isinstance(t, Table)]) == 1
+
+
+def test_snapshot_cli_stats_with_no_tasks(monkeypatch: MonkeyPatch):
+    """When stats are requested but no tasks exist, the stats table should be skipped."""
+    now = datetime(2023, 1, 1, 12, 0, tzinfo=timezone.utc)
+    snapshot_obj = DocketSnapshot(
+        taken=now,
+        total_tasks=0,
+        future=[],
+        running=[],
+        workers=[],
+    )
+
+    registered: list[str] = []
+    _stub_snapshot_docket(monkeypatch, snapshot_obj, registered)
+
+    recorder = RecordingConsole()
+    monkeypatch.setattr("docket.cli.Console", lambda: recorder)
+
+    snapshot_command(stats=True)
+
+    tables = [args[0] for args, _ in recorder.calls if args]
+    # Only the summary table should be printed since task_stats is empty.
+    assert len([t for t in tables if isinstance(t, Table)]) == 1
+
+
 async def test_snapshot_with_stats_shows_timestamp_columns(
     docket: Docket, runner: CliRunner
 ):