53 lines
1.1 KiB
Python
53 lines
1.1 KiB
Python
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"
|