wehub-resource-sync
09e9f3545f
CodeQL / Analyze (push) Waiting to run
dependency-audit / pip-audit (push) Waiting to run
Test / Code Quality (push) Has been cancelled
Test / Test (macos-latest, Python 3.10) (push) Has been cancelled
Test / Test (macos-latest, Python 3.11) (push) Has been cancelled
Test / Test (macos-latest, Python 3.12) (push) Has been cancelled
Test / Test (macos-latest, Python 3.13) (push) Has been cancelled
Test / Test (macos-latest, Python 3.14) (push) Has been cancelled
Test / Test (ubuntu-latest, Python 3.10) (push) Has been cancelled
Test / Test (ubuntu-latest, Python 3.11) (push) Has been cancelled
Test / Test (ubuntu-latest, Python 3.12) (push) Has been cancelled
Test / Test (ubuntu-latest, Python 3.13) (push) Has been cancelled
Test / Test (ubuntu-latest, Python 3.14) (push) Has been cancelled
Test / Test (windows-latest, Python 3.10) (push) Has been cancelled
Test / Test (windows-latest, Python 3.11) (push) Has been cancelled
Test / Test (windows-latest, Python 3.12) (push) Has been cancelled
Test / Test (windows-latest, Python 3.13) (push) Has been cancelled
Test / Test (windows-latest, Python 3.14) (push) Has been cancelled
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
"""Tests for the layer-4 master-token re-mint recovery in _auth/session.py."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
|
|
import httpx
|
|
import pytest
|
|
|
|
import notebooklm._auth.cookies as cookies_mod
|
|
from notebooklm._auth import master_token as mt
|
|
from notebooklm._auth import session as session_mod
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reauth_declines_without_storage_path():
|
|
auth = MagicMock(storage_path=None)
|
|
assert await session_mod._try_master_token_reauth(auth=auth, kernel=MagicMock()) is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reauth_declines_without_token_file(tmp_path):
|
|
auth = MagicMock(storage_path=tmp_path / "storage_state.json")
|
|
assert await session_mod._try_master_token_reauth(auth=auth, kernel=MagicMock()) is False
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reauth_success_remints_and_reloads(tmp_path):
|
|
mt.write_master_token(
|
|
tmp_path / "master_token.json", email="e@x.com", master_token="aas_et/M", android_id="abc"
|
|
)
|
|
auth = MagicMock(storage_path=tmp_path / "storage_state.json")
|
|
jar = httpx.Cookies()
|
|
jar.set("SID", "v", domain=".google.com")
|
|
# _replace_cookie_jar runs for real against the MagicMock kernel (a no-op on a
|
|
# mock jar) — only the network mint + the recovery-aware reload are stubbed.
|
|
with (
|
|
patch.object(mt, "mint_cookies", new=AsyncMock(return_value=jar)),
|
|
patch.object(mt, "persist_minted_jar") as persist,
|
|
patch.object(cookies_mod, "build_httpx_cookies_from_storage", return_value=httpx.Cookies()),
|
|
):
|
|
ok = await session_mod._try_master_token_reauth(auth=auth, kernel=MagicMock())
|
|
assert ok is True
|
|
persist.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_reauth_returns_false_on_revoked_token(tmp_path):
|
|
mt.write_master_token(
|
|
tmp_path / "master_token.json", email="e@x.com", master_token="aas_et/M", android_id="abc"
|
|
)
|
|
auth = MagicMock(storage_path=tmp_path / "storage_state.json")
|
|
with patch.object(
|
|
mt, "mint_cookies", new=AsyncMock(side_effect=mt.MasterTokenError("revoked"))
|
|
):
|
|
ok = await session_mod._try_master_token_reauth(auth=auth, kernel=MagicMock())
|
|
assert ok is False
|