Choose a search result to load its neighborhood.
@@ -828,14 +980,23 @@ _GRAPH_BROWSER_HTML = r"""
pointer: null,
suppressClick: false,
inspectedNode: null,
+ dialogDrag: null,
+ leaseTimer: null,
};
const $ = (id) => document.getElementById(id);
const api = async (path) => {
- const response = await fetch(`${base}api/${path}`, {cache: "no-store"});
+ let response;
+ try {
+ response = await fetch(`${base}api/${path}`, {cache: "no-store"});
+ } catch (_error) {
+ $("connection-state").hidden = false;
+ throw new Error("Visualization listener disconnected; invoke docforge_visualize again");
+ }
const body = await response.json();
if (!response.ok || body.status === "error") {
throw new Error(body.error?.message || "DocForge request failed");
}
+ $("connection-state").hidden = true;
return body;
};
const escapeText = (value) => String(value ?? "");
@@ -879,6 +1040,22 @@ _GRAPH_BROWSER_HTML = r"""
$("status").textContent = message;
$("status").classList.toggle("error", error);
}
+ async function renewViewerLease() {
+ try {
+ await api("heartbeat");
+ } catch (error) {
+ setStatus(error.message, true);
+ if (state.leaseTimer !== null) clearInterval(state.leaseTimer);
+ state.leaseTimer = null;
+ }
+ }
+ function startViewerLease() {
+ if (state.leaseTimer !== null) clearInterval(state.leaseTimer);
+ state.leaseTimer = setInterval(renewViewerLease, 15000);
+ document.addEventListener("visibilitychange", () => {
+ if (!document.hidden) renewViewerLease();
+ });
+ }
function renderOverview(data) {
state.overview = data;
state.searchLimit = Math.max(1, Number(data.max_results) || 1);
@@ -918,22 +1095,153 @@ _GRAPH_BROWSER_HTML = r"""
results.append(button);
}
}
- function layoutNodes(nodes, rootId) {
- const ordered = [...nodes].sort((a, b) => a.node_id.localeCompare(b.node_id));
+ function analyzeTopology(data) {
+ const nodeIds = new Set(data.nodes.map((node) => node.node_id));
+ const adjacency = new Map([...nodeIds].map((nodeId) => [nodeId, new Set()]));
+ const outgoing = new Map([...nodeIds].map((nodeId) => [nodeId, new Set()]));
+ for (const edge of data.edges) {
+ if (!nodeIds.has(edge.source_id) || !nodeIds.has(edge.target_id)) continue;
+ adjacency.get(edge.source_id).add(edge.target_id);
+ adjacency.get(edge.target_id).add(edge.source_id);
+ outgoing.get(edge.source_id).add(edge.target_id);
+ }
+ const hops = new Map([[data.root, 0]]);
+ let frontier = [data.root];
+ while (frontier.length) {
+ const next = [];
+ for (const nodeId of frontier) {
+ for (const neighbor of adjacency.get(nodeId) || []) {
+ if (hops.has(neighbor)) continue;
+ hops.set(neighbor, hops.get(nodeId) + 1);
+ next.push(neighbor);
+ }
+ }
+ frontier = next;
+ }
+ const children = new Set();
+ frontier = [...(outgoing.get(data.root) || [])];
+ for (const nodeId of frontier) children.add(nodeId);
+ while (frontier.length) {
+ const next = [];
+ for (const nodeId of frontier) {
+ for (const candidate of outgoing.get(nodeId) || []) {
+ if (candidate === data.root || children.has(candidate)) continue;
+ children.add(candidate);
+ next.push(candidate);
+ }
+ }
+ frontier = next;
+ }
+ return new Map(data.nodes.map((node) => [
+ node.node_id,
+ {
+ hop: hops.get(node.node_id) ?? data.depth + 1,
+ role: node.node_id === data.root
+ ? "primary"
+ : children.has(node.node_id) ? "child" : "edge",
+ },
+ ]));
+ }
+ function layoutNodes(nodes, rootId, topology) {
+ const ordered = [...nodes].sort((a, b) => {
+ const first = topology.get(a.node_id);
+ const second = topology.get(b.node_id);
+ return first.hop - second.hop
+ || first.role.localeCompare(second.role)
+ || a.node_id.localeCompare(b.node_id);
+ });
const root = ordered.find((node) => node.node_id === rootId);
- const others = ordered.filter((node) => node.node_id !== rootId);
const positions = new Map();
if (root) positions.set(root.node_id, {x: 0, y: 0});
- others.forEach((node, index) => {
- const ring = Math.floor(index / 18) + 1;
- const ringStart = (ring - 1) * 18;
- const ringCount = Math.min(18, others.length - ringStart);
- const angle = ((index - ringStart) / Math.max(1, ringCount)) * Math.PI * 2 - Math.PI / 2;
- const radius = 165 + (ring - 1) * 145;
- positions.set(node.node_id, {x: Math.cos(angle) * radius, y: Math.sin(angle) * radius});
- });
+ const rings = new Map();
+ for (const node of ordered) {
+ if (node.node_id === rootId) continue;
+ const hop = Math.max(1, topology.get(node.node_id).hop);
+ if (!rings.has(hop)) rings.set(hop, []);
+ rings.get(hop).push(node);
+ }
+ for (const [hop, ringNodes] of rings) {
+ ringNodes.forEach((node, index) => {
+ const angle = (index / Math.max(1, ringNodes.length)) * Math.PI * 2 - Math.PI / 2;
+ const radius = 165 + (hop - 1) * 145;
+ positions.set(node.node_id, {
+ x: Math.cos(angle) * radius,
+ y: Math.sin(angle) * radius,
+ });
+ });
+ }
return positions;
}
+ function darken(hex, amount) {
+ const value = Number.parseInt(hex.slice(1), 16);
+ const factor = 1 - Math.min(.5, Math.max(0, amount));
+ const channels = [
+ (value >> 16) & 255,
+ (value >> 8) & 255,
+ value & 255,
+ ].map((channel) => Math.round(channel * factor).toString(16).padStart(2, "0"));
+ return `#${channels.join("")}`;
+ }
+ function nodePalette(role, hop) {
+ const colors = {
+ primary: {fill: "#176b7d", stroke: "#83e8ff"},
+ child: {fill: "#216c51", stroke: "#91f2bd"},
+ edge: {fill: "#634580", stroke: "#d0a7ff"},
+ }[role];
+ const distanceShade = Math.min(.5, Math.max(0, hop - 1) * .17);
+ return {
+ fill: darken(colors.fill, distanceShade),
+ stroke: darken(colors.stroke, distanceShade),
+ };
+ }
+ function renderNeighborhood(data, topology) {
+ const sections = [
+ {role: "primary", label: "Primary focus"},
+ {role: "child", label: "Children"},
+ {role: "edge", label: "Edge & context"},
+ ];
+ const container = $("neighborhood-sections");
+ container.replaceChildren();
+ for (const section of sections) {
+ const nodes = data.nodes
+ .filter((node) => topology.get(node.node_id).role === section.role)
+ .sort((a, b) => topology.get(a.node_id).hop - topology.get(b.node_id).hop
+ || a.node_id.localeCompare(b.node_id));
+ if (!nodes.length) continue;
+ const heading = document.createElement("div");
+ heading.className = "section-label";
+ heading.append(document.createTextNode(section.label));
+ const count = document.createElement("span");
+ count.textContent = String(nodes.length);
+ heading.append(count);
+ const list = document.createElement("div");
+ list.className = "node-list";
+ for (const node of nodes) {
+ const topologyNode = topology.get(node.node_id);
+ const palette = nodePalette(topologyNode.role, topologyNode.hop);
+ const button = document.createElement("button");
+ button.type = "button";
+ button.className = "node-list-item";
+ button.style.setProperty("--item-color", palette.stroke);
+ button.title = `Focus ${node.title}`;
+ const swatch = document.createElement("i");
+ swatch.className = "node-swatch";
+ const copy = document.createElement("span");
+ copy.className = "node-list-copy";
+ const title = document.createElement("strong");
+ title.textContent = node.title;
+ const meta = document.createElement("span");
+ const hopLabel = `${topologyNode.hop} hop${topologyNode.hop === 1 ? "" : "s"}`;
+ meta.textContent = `${node.family} · ${hopLabel}`;
+ copy.append(title, meta);
+ button.append(swatch, copy);
+ button.addEventListener("click", () => loadNode(node.node_id));
+ list.append(button);
+ }
+ container.append(heading, list);
+ }
+ $("neighborhood").hidden = false;
+ }
function svgElement(name, attributes = {}) {
const element = document.createElementNS("http://www.w3.org/2000/svg", name);
for (const [key, value] of Object.entries(attributes)) element.setAttribute(key, value);
@@ -946,15 +1254,24 @@ _GRAPH_BROWSER_HTML = r"""
const svg = $("graph");
svg.replaceChildren();
$("empty").hidden = data.nodes.length > 0;
- const positions = layoutNodes(data.nodes, data.root);
+ const topology = analyzeTopology(data);
+ const positions = layoutNodes(data.nodes, data.root, topology);
+ renderNeighborhood(data, topology);
const edgeLayer = svgElement("g");
const nodeLayer = svgElement("g");
for (const edge of data.edges) {
const source = positions.get(edge.source_id);
const target = positions.get(edge.target_id);
if (!source || !target) continue;
+ const targetRole = topology.get(edge.target_id)?.role || "edge";
+ const edgeRole = edge.source_id === data.root && targetRole === "child"
+ ? "child-edge"
+ : edge.target_id === data.root || targetRole === "edge"
+ ? "context-edge"
+ : "boundary-edge";
edgeLayer.append(svgElement("line", {
- x1: source.x, y1: source.y, x2: target.x, y2: target.y, class: "edge"
+ x1: source.x, y1: source.y, x2: target.x, y2: target.y,
+ class: `edge ${edgeRole}`
}));
const label = svgElement("text", {
x: (source.x + target.x) / 2,
@@ -968,14 +1285,23 @@ _GRAPH_BROWSER_HTML = r"""
for (const node of data.nodes) {
const point = positions.get(node.node_id);
if (!point) continue;
+ const topologyNode = topology.get(node.node_id);
+ const palette = nodePalette(topologyNode.role, topologyNode.hop);
const group = svgElement("g", {
- class: `node${node.node_id === data.root ? " root" : ""}`,
+ class: `node ${topologyNode.role}${node.node_id === data.root ? " root" : ""}`,
transform: `translate(${point.x} ${point.y})`,
+ "data-hop": String(topologyNode.hop),
tabindex: "0",
role: "button",
- "aria-label": `${node.title}, ${node.family}`
+ "aria-label": [
+ node.title, node.family, topologyNode.role, `${topologyNode.hop} hops`
+ ].join(", ")
});
- group.append(svgElement("circle", {r: node.node_id === data.root ? 25 : 18}));
+ group.append(svgElement("circle", {
+ r: node.node_id === data.root ? 25 : 18,
+ fill: palette.fill,
+ stroke: palette.stroke,
+ }));
const title = svgElement("text", {y: 35, "text-anchor": "middle"});
title.textContent = short(node.title, 26);
const family = svgElement("text", {y: 48, "text-anchor": "middle", class: "family"});
@@ -1043,6 +1369,7 @@ _GRAPH_BROWSER_HTML = r"""
const data = await api(`node?${params}`);
state.inspectedNode = nodeId;
renderDetails($("node-dialog-details"), data.node, data);
+ $("node-dialog-label").textContent = short(data.node.title, 72);
const dialog = $("node-dialog");
if (!dialog.open) dialog.showModal();
$("close-node-dialog").focus();
@@ -1079,6 +1406,90 @@ _GRAPH_BROWSER_HTML = r"""
setStatus(error.message, true);
}
}
+ function clamp(value, minimum, maximum) {
+ return Math.min(maximum, Math.max(minimum, value));
+ }
+ function setPanelWidth(side, width) {
+ const maximum = Math.max(260, Math.floor(window.innerWidth * .46));
+ const bounded = clamp(width, side === "left" ? 220 : 240, maximum);
+ document.documentElement.style.setProperty(`--${side}-width`, `${bounded}px`);
+ $(`${side}-resizer`).setAttribute("aria-valuenow", String(Math.round(bounded)));
+ }
+ function setupPanelResizer(side) {
+ const handle = $(`${side}-resizer`);
+ handle.addEventListener("pointerdown", (event) => {
+ if (event.button !== 0) return;
+ const property = getComputedStyle(document.documentElement)
+ .getPropertyValue(`--${side}-width`);
+ const startWidth = Number.parseFloat(property) || (side === "left" ? 310 : 350);
+ const startX = event.clientX;
+ handle.classList.add("resizing");
+ handle.setPointerCapture(event.pointerId);
+ const move = (moveEvent) => {
+ const delta = moveEvent.clientX - startX;
+ setPanelWidth(side, startWidth + (side === "left" ? delta : -delta));
+ };
+ const end = (endEvent) => {
+ if (handle.hasPointerCapture(endEvent.pointerId)) {
+ handle.releasePointerCapture(endEvent.pointerId);
+ }
+ handle.classList.remove("resizing");
+ handle.removeEventListener("pointermove", move);
+ handle.removeEventListener("pointerup", end);
+ handle.removeEventListener("pointercancel", end);
+ };
+ handle.addEventListener("pointermove", move);
+ handle.addEventListener("pointerup", end);
+ handle.addEventListener("pointercancel", end);
+ });
+ handle.addEventListener("keydown", (event) => {
+ if (event.key !== "ArrowLeft" && event.key !== "ArrowRight") return;
+ event.preventDefault();
+ const property = getComputedStyle(document.documentElement)
+ .getPropertyValue(`--${side}-width`);
+ const width = Number.parseFloat(property) || (side === "left" ? 310 : 350);
+ const direction = event.key === "ArrowRight" ? 16 : -16;
+ setPanelWidth(side, width + (side === "left" ? direction : -direction));
+ });
+ handle.addEventListener("dblclick", () => setPanelWidth(side, side === "left" ? 310 : 350));
+ }
+ function beginDialogDrag(event) {
+ if (event.button !== 0 || event.target.closest("button")) return;
+ const dialog = $("node-dialog");
+ const header = event.currentTarget;
+ const bounds = dialog.getBoundingClientRect();
+ dialog.style.margin = "0";
+ dialog.style.right = "auto";
+ dialog.style.bottom = "auto";
+ dialog.style.left = `${bounds.left}px`;
+ dialog.style.top = `${bounds.top}px`;
+ dialog.style.width = `${bounds.width}px`;
+ dialog.style.height = `${bounds.height}px`;
+ state.dialogDrag = {
+ id: event.pointerId,
+ startX: event.clientX,
+ startY: event.clientY,
+ left: bounds.left,
+ top: bounds.top,
+ };
+ header.setPointerCapture(event.pointerId);
+ }
+ function moveDialog(event) {
+ const drag = state.dialogDrag;
+ if (!drag || drag.id !== event.pointerId) return;
+ const dialog = $("node-dialog");
+ const maximumLeft = Math.max(8, window.innerWidth - dialog.offsetWidth - 8);
+ const maximumTop = Math.max(8, window.innerHeight - dialog.offsetHeight - 8);
+ dialog.style.left = `${clamp(drag.left + event.clientX - drag.startX, 8, maximumLeft)}px`;
+ dialog.style.top = `${clamp(drag.top + event.clientY - drag.startY, 8, maximumTop)}px`;
+ }
+ function endDialogDrag(event) {
+ const drag = state.dialogDrag;
+ if (!drag || drag.id !== event.pointerId) return;
+ const header = $("node-dialog").querySelector(".dialog-head");
+ if (header.hasPointerCapture(event.pointerId)) header.releasePointerCapture(event.pointerId);
+ state.dialogDrag = null;
+ }
$("search-form").addEventListener("submit", (event) => { event.preventDefault(); search(); });
$("family").addEventListener("change", search);
$("zoom-in").addEventListener("click", () => zoomAt(.8));
@@ -1086,6 +1497,12 @@ _GRAPH_BROWSER_HTML = r"""
$("reset-view").addEventListener("click", resetViewport);
$("close-node-dialog").addEventListener("click", closeNodeDialog);
$("dismiss-node-dialog").addEventListener("click", closeNodeDialog);
+ setupPanelResizer("left");
+ setupPanelResizer("right");
+ $("node-dialog").querySelector(".dialog-head").addEventListener("pointerdown", beginDialogDrag);
+ $("node-dialog").querySelector(".dialog-head").addEventListener("pointermove", moveDialog);
+ $("node-dialog").querySelector(".dialog-head").addEventListener("pointerup", endDialogDrag);
+ $("node-dialog").querySelector(".dialog-head").addEventListener("pointercancel", endDialogDrag);
$("explore-node").addEventListener("click", async () => {
const nodeId = state.inspectedNode;
closeNodeDialog();
@@ -1162,6 +1579,7 @@ _GRAPH_BROWSER_HTML = r"""
state.depth = Math.max(1, Number(params.get("depth")) || 1);
const overview = await api("overview");
renderOverview(overview);
+ startViewerLease();
const nodeId = params.get("node");
const query = params.get("q");
if (nodeId) {
diff --git a/tests/test_mcp_server.py b/tests/test_mcp_server.py
index 922dc70..3bcc85f 100644
--- a/tests/test_mcp_server.py
+++ b/tests/test_mcp_server.py
@@ -11,7 +11,14 @@ from mcp.client.stdio import stdio_client
from mcp.shared.memory import create_connected_server_and_client_session
from docforge.index import ProjectIndex
-from docforge.mcp_server import ALL_TOOLS, CONTENT_WARNING, PROPOSAL_TOOLS, create_server
+from docforge.mcp_server import (
+ ALL_TOOLS,
+ CONTENT_WARNING,
+ PROPOSAL_TOOLS,
+ DocForgeService,
+ _create_bound_server,
+ create_server,
+)
from docforge.project import Project
ROOT = Path(__file__).resolve().parents[1]
@@ -62,10 +69,16 @@ class DocForgeMcpTests(unittest.IsolatedAsyncioTestCase):
("docforge_render_status", {}),
("docforge_visualize", {"node_id": "guide.workflow", "depth": 1}),
)
- async with create_connected_server_and_client_session(
- create_server(root), raise_exceptions=True
- ) as session:
- results = [await session.call_tool(name, arguments) for name, arguments in calls]
+ service = DocForgeService(Project.open(root))
+ try:
+ async with create_connected_server_and_client_session(
+ _create_bound_server(service, read_only=True), raise_exceptions=True
+ ) as session:
+ results = [
+ await session.call_tool(name, arguments) for name, arguments in calls
+ ]
+ finally:
+ service.visualization.stop()
for result in results:
self.assertFalse(result.isError)
@@ -88,7 +101,8 @@ class DocForgeMcpTests(unittest.IsolatedAsyncioTestCase):
visualization = results[11].structuredContent["visualization"]
self.assertTrue(visualization["read_only"])
self.assertTrue(visualization["project_bound"])
- self.assertEqual("graph-browser@4", visualization["template"])
+ self.assertEqual("graph-browser@5", visualization["template"])
+ self.assertEqual("browser_lease", visualization["lifetime"]["policy"])
self.assertTrue(visualization["url"].startswith("http://127.0.0.1:"))
context = results[8].structuredContent
self.assertLessEqual(context["estimated_tokens"], 180)
diff --git a/tests/test_visualization.py b/tests/test_visualization.py
index f2266d1..5b69683 100644
--- a/tests/test_visualization.py
+++ b/tests/test_visualization.py
@@ -4,6 +4,7 @@ import json
import shutil
import subprocess
import tempfile
+import time
import unittest
import urllib.error
import urllib.parse
@@ -76,6 +77,55 @@ class VisualizationTests(unittest.TestCase):
self.assertEqual("", result.stderr)
self.assertEqual(0, result.returncode)
+ @unittest.skipUnless(shutil.which("node"), "Node.js is required for topology validation")
+ def test_embedded_topology_roles_hops_and_shading_are_deterministic(self) -> None:
+ script = _GRAPH_BROWSER_HTML.split("", 1)[0]
+ topology_logic = script.split("function analyzeTopology", 1)[1].split(
+ "function renderNeighborhood", 1
+ )[0]
+ harness = (
+ "function analyzeTopology"
+ + topology_logic
+ + """
+const data = {
+ root: "primary",
+ depth: 2,
+ nodes: [
+ {node_id: "primary"},
+ {node_id: "child-one"},
+ {node_id: "child-two"},
+ {node_id: "incoming"},
+ ],
+ edges: [
+ {source_id: "primary", relation: "contains", target_id: "child-one"},
+ {source_id: "child-one", relation: "contains", target_id: "child-two"},
+ {source_id: "incoming", relation: "references", target_id: "primary"},
+ ],
+};
+const topology = analyzeTopology(data);
+const positions = layoutNodes(data.nodes, data.root, topology);
+const fail = (message) => { throw new Error(message); };
+if (topology.get("primary").role !== "primary") fail("root role");
+if (topology.get("child-one").role !== "child") fail("direct child role");
+if (topology.get("child-two").role !== "child") fail("descendant role");
+if (topology.get("incoming").role !== "edge") fail("incoming edge role");
+if (topology.get("child-two").hop !== 2) fail("descendant hop");
+if (Math.hypot(positions.get("child-two").x, positions.get("child-two").y)
+ <= Math.hypot(positions.get("child-one").x, positions.get("child-one").y)) {
+ fail("hop rings");
+}
+if (nodePalette("child", 2).fill === nodePalette("child", 1).fill) fail("hop shading");
+"""
+ )
+ result = subprocess.run(
+ ["node", "-e", harness],
+ check=False,
+ capture_output=True,
+ text=True,
+ )
+ self.assertEqual("", result.stderr)
+ self.assertEqual(0, result.returncode)
+
def test_runner_serves_only_token_bound_read_only_graph_endpoints(self) -> None:
with tempfile.TemporaryDirectory() as directory:
root = self.copy_fixture("alpha", Path(directory))
@@ -91,6 +141,8 @@ class VisualizationTests(unittest.TestCase):
self.assertEqual(VISUALIZATION_TEMPLATE, first["template"])
self.assertTrue(first["read_only"])
+ self.assertEqual("browser_lease", first["lifetime"]["policy"])
+ self.assertFalse(runner._thread.daemon)
self.assertEqual(first_url.netloc, second_url.netloc)
self.assertEqual(first_url.path, second_url.path)
@@ -103,12 +155,24 @@ class VisualizationTests(unittest.TestCase):
self.assertIn('id="reset-view"', html)
self.assertIn('id="node-dialog"', html)
self.assertIn('id="explore-node"', html)
+ self.assertIn('id="left-resizer"', html)
+ self.assertIn('id="right-resizer"', html)
+ self.assertIn('id="neighborhood-sections"', html)
self.assertIn(".empty[hidden] { display: none; }", html)
+ self.assertIn("resize: both", html)
+ self.assertNotIn("backdrop-filter", html)
self.assertIn('addEventListener("wheel"', html)
self.assertIn('addEventListener("pointermove"', html)
self.assertIn("inspectNode(node.node_id)", html)
self.assertIn("dialog.showModal()", html)
self.assertIn("await loadNode(nodeId)", html)
+ self.assertIn("beginDialogDrag", html)
+ self.assertIn('setupPanelResizer("left")', html)
+ self.assertIn('setupPanelResizer("right")', html)
+ self.assertIn("Primary focus", html)
+ self.assertIn("Edge & context", html)
+ self.assertIn("distanceShade", html)
+ self.assertIn("renewViewerLease", html)
pointerdown = html.split('$("graph").addEventListener("pointerdown"', 1)[1].split(
'$("graph").addEventListener("pointermove"', 1
)[0]
@@ -128,6 +192,11 @@ class VisualizationTests(unittest.TestCase):
self.assertEqual(3, overview["node_count"])
self.assertEqual(20, overview["max_results"])
+ with urllib.request.urlopen(f"{base}api/heartbeat", timeout=2) as response:
+ heartbeat = json.load(response)
+ self.assertEqual("alive", heartbeat["viewer"])
+ self.assertEqual(runner.lease_seconds, heartbeat["lease_seconds"])
+
search_query = urllib.parse.urlencode(
{"q": "canonical nodes", "family": "", "limit": overview["max_results"]}
)
@@ -170,6 +239,37 @@ class VisualizationTests(unittest.TestCase):
finally:
runner.stop()
+ def test_browser_lease_keeps_listener_alive_then_closes_it(self) -> None:
+ with tempfile.TemporaryDirectory() as directory:
+ root = self.copy_fixture("alpha", Path(directory))
+ index = ProjectIndex(Project.open(root))
+ index.build()
+ runner = VisualizationRunner(
+ index,
+ initial_grace_seconds=0.2,
+ lease_seconds=0.2,
+ monitor_interval_seconds=0.02,
+ )
+ try:
+ result = runner.start()
+ parsed = urllib.parse.urlparse(str(result["url"]))
+ base = f"{parsed.scheme}://{parsed.netloc}{parsed.path}"
+ heartbeat = f"{base}api/heartbeat"
+ with urllib.request.urlopen(heartbeat, timeout=2) as response:
+ self.assertEqual("alive", json.load(response)["viewer"])
+ time.sleep(0.12)
+ with urllib.request.urlopen(heartbeat, timeout=2) as response:
+ self.assertEqual("alive", json.load(response)["viewer"])
+
+ deadline = time.monotonic() + 2
+ while runner._server is not None and time.monotonic() < deadline:
+ time.sleep(0.02)
+ self.assertIsNone(runner._server)
+ with self.assertRaises(OSError):
+ urllib.request.urlopen(base, timeout=0.2)
+ finally:
+ runner.stop()
+
def test_runner_rejects_ambiguous_targets_and_changed_index_snapshot(self) -> None:
with tempfile.TemporaryDirectory() as directory:
root = self.copy_fixture("alpha", Path(directory))
diff --git a/uv.lock b/uv.lock
index cae5497..4613e0a 100644
--- a/uv.lock
+++ b/uv.lock
@@ -206,7 +206,7 @@ wheels = [
[[package]]
name = "docforge"
-version = "0.7.3"
+version = "0.8.0"
source = { editable = "." }
dependencies = [
{ name = "markdown-it-py" },