Function patch_session_login_dual

tests/_fixtures/cli_session.py:89–165

Patch session-side, package, and submodule bindings of ``name``. ``name`` is a helper symbol that lives in :mod:`notebooklm.cli.services.login` (or one of its submodules, post-T4) and is re-imported by :mod:`notebooklm.cli.session_cmd`. Tests that need a helper intercepted regardles

(name: str, **patch_kwargs: Any)

@contextmanager
def patch_session_login_dual(name: str, **patch_kwargs: Any) -> Iterator[Any]:
    """Patch session-side, package, and submodule bindings of ``name``.

    ``name`` is a helper symbol that lives in
    :mod:`notebooklm.cli.services.login` (or one of its submodules,
    post-T4) and is re-imported by :mod:`notebooklm.cli.session_cmd`.
    Tests that need a helper intercepted regardless of which entry point
    reaches it use this to avoid hand-wiring N ``patch(...)`` calls.

    The patches share the same mock object, so call assertions made
    against the returned mock count *every* invocation across all
    surfaces — matching the historical pre-D1 PR-3 behavior of the
    forwarding wrappers in ``cli.session``.

    Args:
        name: Bare symbol name (e.g. ``"_login_with_browser_cookies"``).
        **patch_kwargs: Forwarded to :func:`unittest.mock.patch` for the
            primary services-side surface. Typical: ``new=...``,
            ``side_effect=...``, ``return_value=...``,
            ``new_callable=AsyncMock``.

    Yields:
        The shared mock used for every surface.
    """
    services_target = f"notebooklm.cli.services.login.{name}"
    session_target = f"notebooklm.cli.session_cmd.{name}"
    submodule_targets = _services_login_submodule_targets(name)
    # The session-side patch only applies to names still re-exported by
    # ``session_cmd`` (the retired patch-surface bridge, #1367, dropped the
    # pure re-exports but kept every body-used name like ``get_storage_path``
    # and ``_refresh_from_browser_cookies``). Resolve the module object so the
    # patch can be guarded on the name still existing; a blind patch on a
    # removed name would ``AttributeError`` at setup.
    session_cmd = importlib.import_module("notebooklm.cli.session_cmd")

    # P3.T3 service modules also re-import external helpers (notably
    # ``get_storage_path`` / ``get_browser_profile_dir`` in
    # ``services.playwright_login`` and ``services.session_context``).
    # Fan the patch out to those bindings too so a single
    # ``patch_session_login_dual("get_storage_path", ...)`` call covers
    # every call site without requiring per-test patches.
    p3t3_modules = (
        "notebooklm.cli.services.playwright_login",
        "notebooklm.cli.services.session_context",
        "notebooklm.cli.services.auth_diagnostics",
        "notebooklm.cli.services.auth_source",
    )
    p3t3_targets: list[str] = []
    for mod_name in p3t3_modules:
        try:
            mod = importlib.import_module(mod_name)
        except ImportError:
            continue
        if name in vars(mod):
            p3t3_targets.append(f"{mod_name}.{name}")

    with ExitStack() as stack:
        primary = stack.enter_context(patch(services_target, **patch_kwargs))