110 lines
3.8 KiB
Python
110 lines
3.8 KiB
Python
"""Issue #1160 — route-level regression for clearing the active-document pointer.
|
|
|
|
Exercises the REAL ``PATCH /api/document/{id}`` (session_id="") and
|
|
``DELETE /api/document/{id}`` handlers, proving that closing a document's tab
|
|
(detach or delete) clears the in-memory active-document pointer under the actual
|
|
owner/session routing — not just the helper in isolation.
|
|
|
|
Calls the route handler callables DIRECTLY (extracted from the router) instead of
|
|
through Starlette's TestClient. The TestClient path spun up a middleware app +
|
|
threadpool that could hang in some environments; calling the async handler with a
|
|
minimal fake request keeps the same real coverage (handler + DB + owner routing)
|
|
while completing reliably everywhere.
|
|
"""
|
|
|
|
import tempfile
|
|
import sys
|
|
import uuid
|
|
from types import SimpleNamespace
|
|
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.pool import NullPool
|
|
from unittest.mock import MagicMock
|
|
|
|
|
|
def _drop_fake_core_database():
|
|
parent = sys.modules.get("core")
|
|
attr = getattr(parent, "database", None) if parent is not None else None
|
|
mod = sys.modules.get("core.database") or attr
|
|
if mod is None or isinstance(getattr(mod, "__file__", None), str):
|
|
return
|
|
sys.modules.pop("core.database", None)
|
|
sys.modules.pop("src.database", None)
|
|
if parent is not None and attr is mod:
|
|
delattr(parent, "database")
|
|
|
|
|
|
_drop_fake_core_database()
|
|
|
|
import core.database as cdb
|
|
import routes.document_routes as droutes
|
|
from core.database import Document
|
|
from core.database import Session as DbSession
|
|
from routes.document_helpers import DocumentPatch
|
|
from src.tool_implementations import set_active_document, get_active_document
|
|
|
|
_TMPDB = tempfile.NamedTemporaryFile(suffix=".db", delete=False)
|
|
_ENGINE = create_engine(
|
|
f"sqlite:///{_TMPDB.name}",
|
|
connect_args={"check_same_thread": False},
|
|
poolclass=NullPool,
|
|
)
|
|
cdb.Base.metadata.create_all(_ENGINE)
|
|
_TS = sessionmaker(bind=_ENGINE, autoflush=False, autocommit=False)
|
|
droutes.SessionLocal = _TS # route handlers resolve SessionLocal at call time
|
|
|
|
|
|
def _req():
|
|
return SimpleNamespace(state=SimpleNamespace(current_user="tester"))
|
|
|
|
|
|
def _endpoint(method, path):
|
|
router = droutes.setup_document_routes(MagicMock(), None)
|
|
for r in router.routes:
|
|
if getattr(r, "path", None) == path and method in getattr(r, "methods", set()):
|
|
return r.endpoint
|
|
raise RuntimeError(f"{method} {path} not found")
|
|
|
|
|
|
def _make_doc():
|
|
sid = "s-" + uuid.uuid4().hex[:8]
|
|
db = _TS()
|
|
try:
|
|
db.add(DbSession(id=sid, owner="tester", name="s", model="m", endpoint_url="http://x"))
|
|
doc = Document(
|
|
id=str(uuid.uuid4()), session_id=sid, title="t",
|
|
language="markdown", current_content="hi", version_count=1,
|
|
is_active=True, owner="tester",
|
|
)
|
|
db.add(doc)
|
|
db.commit()
|
|
return doc.id
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
async def test_patch_unlink_clears_active_document():
|
|
patch_document = _endpoint("PATCH", "/api/document/{doc_id}")
|
|
doc_id = _make_doc()
|
|
set_active_document(doc_id)
|
|
await patch_document(_req(), doc_id, DocumentPatch(session_id=""))
|
|
assert get_active_document() is None
|
|
|
|
|
|
async def test_delete_clears_active_document():
|
|
delete_document = _endpoint("DELETE", "/api/document/{doc_id}")
|
|
doc_id = _make_doc()
|
|
set_active_document(doc_id)
|
|
await delete_document(_req(), doc_id)
|
|
assert get_active_document() is None
|
|
|
|
|
|
async def test_unlinking_a_different_doc_leaves_pointer():
|
|
patch_document = _endpoint("PATCH", "/api/document/{doc_id}")
|
|
active_id = _make_doc()
|
|
other_id = _make_doc()
|
|
set_active_document(active_id)
|
|
await patch_document(_req(), other_id, DocumentPatch(session_id=""))
|
|
assert get_active_document() == active_id
|