wehub-resource-sync
09e9f3545f
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
CodeQL / Analyze (push) Has been cancelled
dependency-audit / pip-audit (push) Has been cancelled
67 lines
2.8 KiB
Python
67 lines
2.8 KiB
Python
"""E2E: the interactive (studio-artifact) mind map lifecycle.
|
|
|
|
Exercises the real NotebookLM API end-to-end through the public
|
|
``client.mind_maps`` surface for the new interactive mind map (type 4 /
|
|
variant 4, created via CREATE_ARTIFACT): generate -> poll -> read tree
|
|
(GET_INTERACTIVE_HTML) -> rename (RENAME_ARTIFACT) -> delete (DELETE_ARTIFACT).
|
|
Marked ``e2e``, so it only runs with real auth and ``-m e2e``. The wire
|
|
lifecycle was validated live while authoring #1256 Phase 2.
|
|
|
|
Run: ``uv run pytest tests/e2e/test_interactive_mind_map.py -m e2e``
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
from notebooklm.types import MindMapKind
|
|
|
|
# Live CREATE_ARTIFACT coverage — monitored by the nightly generation coverage
|
|
# floor so a fully-throttled run (every generation skipped) reds the nightly
|
|
# instead of passing hollow-green. See tests/e2e/conftest.py (#1819).
|
|
pytestmark = pytest.mark.live_generation
|
|
|
|
|
|
@pytest.mark.e2e
|
|
@pytest.mark.asyncio
|
|
async def test_interactive_mind_map_full_lifecycle(client, generation_notebook_id):
|
|
nb_id = generation_notebook_id
|
|
source_ids = await client.notebooks.get_source_ids(nb_id)
|
|
assert source_ids, "generation notebook must have at least one source"
|
|
|
|
# --- generate (CREATE_ARTIFACT, type 4 / variant 4) + poll to completion ---
|
|
mind_map = await client.mind_maps.generate(
|
|
nb_id, source_ids, kind=MindMapKind.INTERACTIVE, wait=True
|
|
)
|
|
# Enter the cleanup guard immediately after create so a failed assertion
|
|
# below never leaks the real interactive artifact in the live notebook.
|
|
try:
|
|
assert mind_map.kind == MindMapKind.INTERACTIVE
|
|
assert mind_map.id, "generate() must return a non-empty interactive artifact id"
|
|
|
|
# --- recognition (Phase 1) ---
|
|
listed = {m.id: m for m in await client.mind_maps.list(nb_id)}
|
|
assert mind_map.id in listed
|
|
assert listed[mind_map.id].kind == MindMapKind.INTERACTIVE
|
|
|
|
# --- read tree (GET_INTERACTIVE_HTML returns it at [0][9][3]) ---
|
|
tree = await client.mind_maps.get_tree(nb_id, mind_map.id, kind=MindMapKind.INTERACTIVE)
|
|
assert isinstance(tree, dict)
|
|
assert "name" in tree and "children" in tree
|
|
|
|
# --- rename (RENAME_ARTIFACT) ---
|
|
await client.mind_maps.rename(
|
|
nb_id, mind_map.id, "E2E Interactive Mind Map", kind=MindMapKind.INTERACTIVE
|
|
)
|
|
renamed = next(m for m in await client.mind_maps.list(nb_id) if m.id == mind_map.id)
|
|
assert renamed.title == "E2E Interactive Mind Map"
|
|
finally:
|
|
# --- delete (DELETE_ARTIFACT) ---
|
|
if mind_map.id:
|
|
await client.mind_maps.delete(nb_id, mind_map.id, kind=MindMapKind.INTERACTIVE)
|
|
|
|
remaining = [
|
|
m.id for m in await client.mind_maps.list(nb_id) if m.kind == MindMapKind.INTERACTIVE
|
|
]
|
|
assert mind_map.id not in remaining
|