Новый модуль trace

tests/v2/test_trace_manager.py

@@ -0,0 +1,79 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+
+from config_manager.v2.trace import TraceManager
+
+
+@dataclass
+class MemoryTraceTransport:
+    contexts: list = field(default_factory=list)
+    messages: list = field(default_factory=list)
+
+    def write_context(self, record) -> None:
+        self.contexts.append(record)
+
+    def write_message(self, record) -> None:
+        self.messages.append(record)
+
+
+def test_bind_context_writes_context_and_returns_trace_id():
+    transport = MemoryTraceTransport()
+    trace = TraceManager(transport=transport)
+
+    trace_id = trace.bind_context(alias="email-1", type="email", attrs={"message_id": "abc"})
+
+    assert trace_id
+    assert transport.contexts[0].trace_id == trace_id
+    assert transport.contexts[0].alias == "email-1"
+    assert transport.contexts[0].type == "email"
+    assert transport.contexts[0].attrs == {"message_id": "abc"}
+
+
+def test_open_context_restores_parent_after_exit():
+    transport = MemoryTraceTransport()
+    trace = TraceManager(transport=transport)
+
+    parent_id = trace.bind_context(alias="email-1", type="email")
+    with trace.open_context(alias="order.xlsx", type="attachment") as child_id:
+        assert trace.current_trace_id() == child_id
+
+    assert trace.current_trace_id() == parent_id
+
+
+def test_messages_use_current_step_and_attrs():
+    transport = MemoryTraceTransport()
+    trace = TraceManager(transport=transport)
+
+    trace.bind_context(alias="email-1", type="email")
+    trace.step("parse_email")
+    trace.info("Письмо распарсено", status="completed", attrs={"attachments_count": 2})
+
+    message = transport.messages[0]
+    assert message.step == "parse_email"
+    assert message.status == "completed"
+    assert message.level == "INFO"
+    assert message.message == "Письмо распарсено"
+    assert message.attrs == {"attachments_count": 2}
+
+
+def test_bind_context_keeps_parent_empty_by_default():
+    transport = MemoryTraceTransport()
+    trace = TraceManager(transport=transport)
+
+    parent_id = trace.bind_context(alias="email-1", type="email")
+    child_id = trace.bind_context(alias="order.xlsx", type="attachment")
+
+    assert child_id != parent_id
+    assert transport.contexts[-1].parent_id is None
+
+
+def test_bind_context_uses_explicit_parent_id():
+    transport = MemoryTraceTransport()
+    trace = TraceManager(transport=transport)
+
+    parent_id = trace.bind_context(alias="email-1", type="email")
+    child_id = trace.bind_context(alias="order.xlsx", type="attachment", parent_id=parent_id)
+
+    assert child_id != parent_id
+    assert transport.contexts[-1].parent_id == parent_id