INF-3 Добавить параметр для вывода дочерних и родительских трейсов

tests/test_trace_endpoint.py

@@ -72,7 +72,17 @@ def test_trace_endpoint_returns_html_by_default() -> None:
     assert "trace_id:" in response.text
     assert "first error" in response.text
     assert "second warning" in response.text
-    assert captured == [("trace-1", TraceQueryRequest(levels=("ERROR", "WARNING", "INFO"), include_attrs_json=False, response_format="html"))]
+    assert captured == [
+        (
+            "trace-1",
+            TraceQueryRequest(
+                levels=("ERROR", "WARNING", "INFO"),
+                include_attrs_json=False,
+                response_format="html",
+                ancestor_depth=0,
+            ),
+        )
+    ]
 
 
 def test_trace_endpoint_returns_text_when_requested() -> None:
@@ -203,9 +213,58 @@ def test_trace_endpoint_returns_json_payload() -> None:
                 "attrs_json": {"batch": 7},
             }
         ],
+        "ancestors": [],
     }
 
 
+def test_trace_endpoint_returns_json_payload_with_ancestors() -> None:
+    async def trace_provider(_trace_id: str, _request: TraceQueryRequest) -> TraceLogView:
+        return TraceLogView(
+            trace_id="trace-1",
+            parent_id="parent-1",
+            child_ids=("child-1",),
+            records=(
+                _trace_record(row_id=3, level="INFO", message="done", attrs_json={"batch": 7}),
+            ),
+            ancestors=(
+                TraceLogView(
+                    trace_id="parent-1",
+                    parent_id="root-1",
+                    child_ids=("trace-1", "sibling-1"),
+                    records=(
+                        _trace_record(row_id=4, level="WARNING", message="parent warning"),
+                    ),
+                ),
+            ),
+        )
+
+    client = _build_client(trace_provider)
+    try:
+        response = client.get("/traces/trace-1?format=json&ancestor_depth=1")
+    finally:
+        client.close()
+
+    assert response.status_code == 200
+    assert response.json()["ancestors"] == [
+        {
+            "trace_id": "parent-1",
+            "parent_id": "root-1",
+            "child_ids": ["trace-1", "sibling-1"],
+            "messages": [
+                {
+                    "id": 4,
+                    "trace_id": "trace-1",
+                    "event_time": "2026-04-28T10:11:12+00:00",
+                    "step": "process",
+                    "status": "failed",
+                    "level": "WARNING",
+                    "message": "parent warning",
+                }
+            ],
+        }
+    ]
+
+
 def test_trace_endpoint_returns_html_page_with_related_links() -> None:
     async def trace_provider(_trace_id: str, _request: TraceQueryRequest) -> TraceLogView:
         return TraceLogView(
@@ -250,11 +309,86 @@ def test_trace_endpoint_returns_html_page_with_related_links() -> None:
     assert "Related Traces" not in response.text
 
 
+def test_trace_endpoint_renders_ancestors_in_text_mode() -> None:
+    async def trace_provider(_trace_id: str, _request: TraceQueryRequest) -> TraceLogView:
+        return TraceLogView(
+            trace_id="trace-1",
+            parent_id="parent-1",
+            child_ids=(),
+            records=(_trace_record(row_id=1, level="INFO", message="child message"),),
+            ancestors=(
+                TraceLogView(
+                    trace_id="parent-1",
+                    parent_id="root-1",
+                    child_ids=("trace-1",),
+                    records=(_trace_record(row_id=2, level="WARNING", message="parent message"),),
+                ),
+            ),
+        )
+
+    client = _build_client(trace_provider)
+    try:
+        response = client.get("/traces/trace-1?format=text&ancestor_depth=1")
+    finally:
+        client.close()
+
+    assert response.status_code == 200
+    assert response.text == (
+        "trace_id: trace-1\n"
+        "parent_id: parent-1\n"
+        "child_ids:\n"
+        "------------------------------\n"
+        "step: process\n"
+        "child message\n"
+        "\n"
+        "ancestor[1]:\n"
+        "trace_id: parent-1\n"
+        "parent_id: root-1\n"
+        "child_ids:\n"
+        " - trace-1\n"
+        "------------------------------\n"
+        "step: process\n"
+        "parent message"
+    )
+
+
+def test_trace_endpoint_preserves_ancestor_depth_in_html_links() -> None:
+    async def trace_provider(_trace_id: str, _request: TraceQueryRequest) -> TraceLogView:
+        return TraceLogView(
+            trace_id="trace-1",
+            parent_id="parent-1",
+            child_ids=("child-1",),
+            records=(_trace_record(row_id=1, level="INFO", message="loaded prices"),),
+            ancestors=(
+                TraceLogView(
+                    trace_id="parent-1",
+                    parent_id="root-1",
+                    child_ids=("trace-1",),
+                    records=(_trace_record(row_id=2, level="WARNING", message="parent warning"),),
+                ),
+            ),
+        )
+
+    client = _build_client(trace_provider)
+    try:
+        response = client.get("/traces/trace-1?format=html&attrs_json=true&ancestor_depth=all")
+    finally:
+        client.close()
+
+    assert response.status_code == 200
+    assert 'href="/traces/trace-1?format=html&levels=ERROR%2CWARNING%2CINFO&attrs_json=true&ancestor_depth=all"' in response.text
+    assert 'href="/traces/parent-1?format=html&levels=ERROR%2CWARNING%2CINFO&attrs_json=true&ancestor_depth=all"' in response.text
+    assert '<div class="line">ancestor[1]:</div>' in response.text
+    assert "parent warning" in response.text
+
+
 def test_trace_endpoint_validates_query_params() -> None:
     client = _build_client(lambda _trace_id, _request: None)
     try:
         invalid_level = client.get("/traces/trace-1?levels=error,fatal")
         invalid_format = client.get("/traces/trace-1?format=xml")
+        invalid_ancestor_depth = client.get("/traces/trace-1?ancestor_depth=-1")
+        invalid_ancestor_type = client.get("/traces/trace-1?ancestor_depth=up")
     finally:
         client.close()
 
@@ -262,6 +396,16 @@ def test_trace_endpoint_validates_query_params() -> None:
     assert invalid_level.json() == {"status": "error", "detail": "unsupported trace levels: FATAL"}
     assert invalid_format.status_code == 400
     assert invalid_format.json() == {"status": "error", "detail": "unsupported trace format: xml"}
+    assert invalid_ancestor_depth.status_code == 400
+    assert invalid_ancestor_depth.json() == {
+        "status": "error",
+        "detail": "query parameter must be >= 0: ancestor_depth=-1",
+    }
+    assert invalid_ancestor_type.status_code == 400
+    assert invalid_ancestor_type.json() == {
+        "status": "error",
+        "detail": "invalid ancestor depth query parameter: ancestor_depth=up",
+    }
 
 
 def test_runtime_trace_logs_uses_configured_reader(monkeypatch) -> None:
@@ -273,15 +417,21 @@ def test_runtime_trace_logs_uses_configured_reader(monkeypatch) -> None:
     )
 
     class StubReader:
-        def read_trace(self, trace_id: str, levels: tuple[str, ...]) -> TraceLogView | None:
+        def read_trace(
+            self,
+            trace_id: str,
+            levels: tuple[str, ...],
+            ancestor_depth: int | None = 0,
+        ) -> TraceLogView | None:
             assert trace_id == "trace-1"
             assert levels == ("ERROR",)
+            assert ancestor_depth is None
             return expected
 
     monkeypatch.setattr(runtime_module, "build_trace_log_reader", lambda _transport: StubReader())
     runtime = RuntimeManager()
 
-    result = asyncio.run(runtime.trace_logs("trace-1", TraceQueryRequest(levels=("ERROR",))))
+    result = asyncio.run(runtime.trace_logs("trace-1", TraceQueryRequest(levels=("ERROR",), ancestor_depth=None)))
 
     assert result == expected
 
@@ -297,7 +447,11 @@ def test_mysql_trace_log_reader_maps_db_rows() -> None:
             self._current_query = query
 
         def fetchone(self) -> dict[str, object] | None:
-            return {"parent_id": "root-77"}
+            if self.executed[-1][1] == ("trace-1",):
+                return {"parent_id": "root-77"}
+            if self.executed[-1][1] == ("root-77",):
+                return {"parent_id": None}
+            return None
 
         def fetchall(self) -> list[dict[str, object]]:
             if "WHERE parent_id = %s" in self._current_query:
@@ -362,7 +516,91 @@ def test_mysql_trace_log_reader_maps_db_rows() -> None:
                 attrs_json={"attempt": 1},
             ),
         ),
+        ancestors=(),
     )
     assert len(factory.cursor.executed) == 3
     assert factory.cursor.executed[1][1] == ("trace-1",)
     assert factory.cursor.executed[2][1] == ("trace-1", "ERROR", "WARNING")
+
+
+def test_mysql_trace_log_reader_loads_requested_ancestors() -> None:
+    class FakeCursor:
+        def __init__(self) -> None:
+            self.executed: list[tuple[str, tuple[object, ...]]] = []
+            self._current_query = ""
+
+        def execute(self, query: str, params: tuple[object, ...]) -> None:
+            self.executed.append((query, params))
+            self._current_query = query
+
+        def fetchone(self) -> dict[str, object] | None:
+            if self.executed[-1][1] == ("trace-1",):
+                return {"parent_id": "parent-1"}
+            if self.executed[-1][1] == ("parent-1",):
+                return {"parent_id": "root-1"}
+            if self.executed[-1][1] == ("root-1",):
+                return {"parent_id": None}
+            return None
+
+        def fetchall(self) -> list[dict[str, object]]:
+            if "WHERE parent_id = %s" in self._current_query:
+                parent_id = self.executed[-1][1][0]
+                if parent_id == "trace-1":
+                    return []
+                if parent_id == "parent-1":
+                    return [{"trace_id": "trace-1"}]
+                if parent_id == "root-1":
+                    return [{"trace_id": "parent-1"}]
+                return []
+            trace_id = self.executed[-1][1][0]
+            return [
+                {
+                    "id": 8 if trace_id == "trace-1" else 9,
+                    "trace_id": trace_id,
+                    "event_time": datetime(2026, 4, 28, 10, 11, 12, tzinfo=timezone.utc),
+                    "step": "parse",
+                    "status": "failed",
+                    "level": "ERROR",
+                    "message": f"broken:{trace_id}",
+                    "attrs_json": '{"attempt":1}',
+                }
+            ]
+
+        def __enter__(self) -> FakeCursor:
+            return self
+
+        def __exit__(self, exc_type, exc, tb) -> None:
+            return None
+
+    class FakeConnection:
+        def __init__(self, cursor: FakeCursor) -> None:
+            self._cursor = cursor
+
+        def cursor(self) -> FakeCursor:
+            return self._cursor
+
+        def __enter__(self) -> FakeConnection:
+            return self
+
+        def __exit__(self, exc_type, exc, tb) -> None:
+            return None
+
+    class FakeConnectionFactory:
+        def __init__(self) -> None:
+            self.cursor = FakeCursor()
+
+        def connect(self) -> FakeConnection:
+            return FakeConnection(self.cursor)
+
+    factory = FakeConnectionFactory()
+    reader = MySqlTraceLogReader(factory)  # type: ignore[arg-type]
+
+    view = reader.read_trace("trace-1", ("ERROR",), 1)
+
+    assert view is not None
+    assert view.trace_id == "trace-1"
+    assert view.parent_id == "parent-1"
+    assert len(view.ancestors) == 1
+    assert view.ancestors[0].trace_id == "parent-1"
+    assert view.ancestors[0].parent_id == "root-1"
+    assert view.ancestors[0].child_ids == ("trace-1",)