first commit

tests/services/test_session_context.py

@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+from app.services.session_context import extract_protocol
+
+
+def test_extract_protocol_prefers_start_payload_data() -> None:
+    protocol = extract_protocol(
+        {
+            "data": {
+                "protocolo": "PRT-123",
+            }
+        },
+        {"protocolo": "PRT-999"},
+    )
+
+    assert protocol == "PRT-123"
+
+
+def test_extract_protocol_falls_back_to_session_data() -> None:
+    protocol = extract_protocol(
+        {"data": {}},
+        {"protocolo": "PRT-999"},
+    )
+
+    assert protocol == "PRT-999"
+
+
+def test_extract_protocol_falls_back_to_router_call_key_when_protocol_is_missing() -> None:
+    protocol = extract_protocol(
+        {
+            "data": {
+                "routerCallKey": "RCK-123",
+            }
+        },
+        {},
+    )
+
+    assert protocol == "RCK-123"
+
+
+def test_extract_protocol_accepts_protocol_id() -> None:
+    protocol = extract_protocol(
+        {
+            "data": {
+                "protocol_id": "PRT-777",
+                "routerCallKey": "RCK-123",
+            }
+        },
+        {},
+    )
+
+    assert protocol == "PRT-777"