Agent/tests/test_trace_id.py

import unittest

from cyber_agent.trace.trace_id import extract_mode, parse_parent_trace_id


class NestedTraceIdTest(unittest.TestCase):
    def test_nested_trace_returns_direct_parent_and_current_mode(self):
        parent = "root@delegate-20260716120000-001"
        child = f"{parent}@explore-20260716120100-001"

        self.assertEqual(parent, parse_parent_trace_id(child))
        self.assertEqual("explore", extract_mode(child))

    def test_root_has_no_parent_or_mode(self):
        self.assertIsNone(parse_parent_trace_id("root"))
        self.assertIsNone(extract_mode("root"))


if __name__ == "__main__":
    unittest.main()