chore: import upstream snapshot with attribution

tests/unit/test_cli_source_delete.py

@@ -0,0 +1,143 @@
+"""Integration tests for source delete CLI flows."""
+
+import json
+from pathlib import Path
+from unittest.mock import AsyncMock, patch
+
+import pytest
+from click.testing import CliRunner
+from pytest_httpx import HTTPXMock
+
+import notebooklm.auth as auth_module
+import notebooklm.cli.context as context_module
+import notebooklm.cli.helpers as helpers_module
+import notebooklm.cli.resolve as resolve_module
+from notebooklm.notebooklm_cli import cli
+from notebooklm.rpc import RPCMethod
+
+
+@pytest.fixture
+def runner() -> CliRunner:
+    """Create a Click test runner."""
+    return CliRunner()
+
+
+@pytest.fixture
+def mock_auth():
+    """Mock authentication for CLI integration tests."""
+    import httpx
+
+    mock_jar = httpx.Cookies()
+    mock_jar.set("SID", "test", domain=".google.com")
+
+    with (
+        patch.object(
+            helpers_module,
+            "load_auth_from_storage",
+            return_value={
+                "SID": "test",
+                "HSID": "test",
+                "SSID": "test",
+                "APISID": "test",
+                "SAPISID": "test",
+            },
+        ),
+        patch.object(
+            auth_module,
+            "fetch_tokens_with_domains",
+            new_callable=AsyncMock,
+        ) as mock_fetch,
+        patch.object(
+            helpers_module,
+            "build_cookie_jar",
+            return_value=mock_jar,
+        ),
+    ):
+        mock_fetch.return_value = ("csrf_token", "session_id")
+        yield
+
+
+@pytest.fixture
+def mock_context(tmp_path: Path):
+    """Provide a canonical notebook UUID so CLI skips notebook-list resolution."""
+    context_file = tmp_path / "context.json"
+    context_file.write_text(
+        json.dumps({"notebook_id": "06f0c5bd-108f-4c8b-8911-34b2acc656de"}),
+        encoding="utf-8",
+    )
+
+    with (
+        patch.object(helpers_module, "get_context_path", return_value=context_file),
+        patch.object(context_module, "get_context_path", return_value=context_file),
+        patch.object(resolve_module, "get_context_path", return_value=context_file),
+    ):
+        yield context_file
+
+
+def _build_source_list_response(build_rpc_response, source_id: str, title: str) -> str:
+    """Build a GET_NOTEBOOK response containing a single source."""
+    return build_rpc_response(
+        RPCMethod.GET_NOTEBOOK,
+        [
+            [
+                "Test Notebook",
+                [
+                    [
+                        [source_id],
+                        title,
+                        [None, 11, [1704067200, 0], None, 5, None, None, None],
+                        [None, 2],
+                    ]
+                ],
+                "06f0c5bd-108f-4c8b-8911-34b2acc656de",
+                "📘",
+                None,
+                [None, None, None, None, None, [1704067200, 0]],
+            ]
+        ],
+    )
+
+
+class TestCliSourceDeleteIntegration:
+    """Integration coverage for CLI source delete flows."""
+
+    def test_source_delete_by_title(
+        self, runner, mock_auth, mock_context, httpx_mock: HTTPXMock, build_rpc_response
+    ):
+        httpx_mock.add_response(
+            content=_build_source_list_response(
+                build_rpc_response,
+                "ff503bfa-5e39-4281-a1d8-2a66c7b86724",
+                "VCR Delete Test Source",
+            ).encode()
+        )
+        httpx_mock.add_response(
+            content=build_rpc_response(RPCMethod.DELETE_SOURCE, [True]).encode()
+        )
+
+        result = runner.invoke(
+            cli,
+            ["source", "delete-by-title", "VCR Delete Test Source", "-y"],
+        )
+
+        assert result.exit_code == 0
+        assert "Deleted source" in result.output
+
+    def test_source_delete_title_suggests_delete_by_title(
+        self, runner, mock_auth, mock_context, httpx_mock: HTTPXMock, build_rpc_response
+    ):
+        httpx_mock.add_response(
+            content=_build_source_list_response(
+                build_rpc_response,
+                "ff503bfa-5e39-4281-a1d8-2a66c7b86724",
+                "VCR Delete Test Source",
+            ).encode()
+        )
+
+        result = runner.invoke(
+            cli,
+            ["source", "delete", "VCR Delete Test Source", "-y"],
+        )
+
+        assert result.exit_code == 1
+        assert "delete-by-title" in result.output