administrator/DocForge2
1
0
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity Actions Pages

Keep visualization worker alive until explicit stop

Browse source
This commit is contained in:
Andraxion 2026-07-24 23:55:55 -04:00
parent 440ca7510f
commit eb48ba1a51
12 changed files with 410 additions and 150 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test_mcp_server.py
View file

@ -68,6 +68,7 @@ class DocForgeMcpTests(unittest.IsolatedAsyncioTestCase):
("docforge_validate_project", {}),
("docforge_render_status", {}),
("docforge_visualize", {"node_id": "guide.workflow", "depth": 1}),
("docforge_stop_visualization", {}),
)
service = DocForgeService(Project.open(root))
try:
@ -102,8 +103,10 @@ class DocForgeMcpTests(unittest.IsolatedAsyncioTestCase):
self.assertTrue(visualization["read_only"])
self.assertTrue(visualization["project_bound"])
self.assertEqual("graph-browser@8", visualization["template"])
self.assertEqual("browser_lease", visualization["lifetime"]["policy"])
self.assertEqual("explicit_stop", visualization["lifetime"]["policy"])
self.assertEqual("docforge_stop_visualization", visualization["lifetime"]["stop_tool"])
self.assertTrue(visualization["url"].startswith("http://127.0.0.1:"))
self.assertEqual("stopped", results[12].structuredContent["state"])
context = results[8].structuredContent
self.assertLessEqual(context["estimated_tokens"], 180)
self.assertTrue(context["omissions"])

28
tests/test_visualization.py
View file

@ -18,6 +18,7 @@ from docforge.project import Project
from docforge.visualization import (
_GRAPH_BROWSER_HTML,
VISUALIZATION_TEMPLATE,
PersistentVisualizationRunner,
VisualizationIndexSnapshot,
VisualizationRunner,
)
@ -379,7 +380,7 @@ if (dependencyEdge.source_id !== "dependency" || dependencyEdge.target_id !== "p
finally:
runner.stop()
def test_detached_worker_survives_the_launching_transport_process(self) -> None:
def test_persistent_worker_survives_launcher_and_stops_only_explicitly(self) -> None:
with tempfile.TemporaryDirectory() as directory:
root = self.copy_fixture("alpha", Path(directory))
ProjectIndex(Project.open(root)).build()
@ -388,14 +389,9 @@ import sys
from pathlib import Path
from docforge.index import ProjectIndex
from docforge.project import Project
from docforge.visualization import DetachedVisualizationRunner
from docforge.visualization import PersistentVisualizationRunner
runner = DetachedVisualizationRunner(
ProjectIndex(Project.open(Path(sys.argv[1]))),
initial_grace_seconds=1.0,
lease_seconds=0.3,
monitor_interval_seconds=0.02,
)
runner = PersistentVisualizationRunner(ProjectIndex(Project.open(Path(sys.argv[1]))))
print(runner.start()["url"], flush=True)
time.sleep(60)
"""
@ -416,8 +412,20 @@ time.sleep(60)
self.assertEqual(200, response.status)
time.sleep(0.6)
with self.assertRaises(OSError):
urllib.request.urlopen(url, timeout=0.2)
with urllib.request.urlopen(url, timeout=2) as response:
self.assertEqual(200, response.status)
runner = PersistentVisualizationRunner(ProjectIndex(Project.open(root)))
try:
reused = runner.start()
self.assertTrue(reused["reused"])
self.assertEqual(url.split("?", 1)[0], str(reused["url"]).split("?", 1)[0])
stopped = runner.stop()
self.assertEqual("stopped", stopped["state"])
with self.assertRaises(OSError):
urllib.request.urlopen(url, timeout=0.2)
finally:
runner.stop()
def test_runner_rejects_ambiguous_targets_and_changed_index_snapshot(self) -> None:
with tempfile.TemporaryDirectory() as directory: