фикс поллинга накладной

app/adapters/delivery_providers/cdek/order_mapper.py

@@ -173,6 +173,13 @@ def _latest_status_code(statuses: object) -> str | None:
 
 
 def _extract_waybill(payload: dict[str, Any]) -> tuple[str | None, str | None]:
+    entity = payload.get("entity")
+    related_entities = (
+        entity.get("related_entities")
+        if isinstance(entity, dict)
+        else None
+    )
+    if not isinstance(related_entities, list):
         related_entities = payload.get("related_entities")
     if not isinstance(related_entities, list):
         return None, None

tests/adapters/delivery_providers/cdek/test_order_info_client.py

@@ -64,11 +64,11 @@ def test_get_order_parses_status_and_waybill_uuid() -> None:
                 "statuses": [
                     {"code": "ACCEPTED", "date_time": "2026-05-24T10:00:00+0000"}
                 ],
-            },
                 "related_entities": [
                     {"type": "waybill", "uuid": "waybill-uuid-1"}
                 ],
             },
+        },
         request=httpx.Request("GET", "https://api.cdek.test/v2/orders/cdek-order-uuid"),
     )
     http_client = SequenceHTTPClient([response])

tests/adapters/delivery_providers/cdek/test_order_mapper.py

@@ -42,7 +42,8 @@ def test_map_cdek_order_response_extracts_order_uuid_without_waybill() -> None:
 def test_map_cdek_order_response_extracts_waybill_from_related_entities() -> None:
     result = map_cdek_order_response(
         {
-            "entity": {"uuid": "cdek-order-uuid"},
+            "entity": {
+                "uuid": "cdek-order-uuid",
                 "related_entities": [
                     {"type": "delivery", "uuid": "ignored"},
                     {
@@ -51,6 +52,7 @@ def test_map_cdek_order_response_extracts_waybill_from_related_entities() -> Non
                         "url": "https://cdek.test/waybill/1.pdf",
                     },
                 ],
+            },
         }
     )
 
@@ -61,6 +63,26 @@ def test_map_cdek_order_response_extracts_waybill_from_related_entities() -> Non
     )
 
 
+def test_map_cdek_order_response_supports_root_related_entities() -> None:
+    result = map_cdek_order_response(
+        {
+            "entity": {"uuid": "cdek-order-uuid"},
+            "related_entities": [
+                {
+                    "type": "waybill",
+                    "uuid": "waybill-uuid-legacy",
+                },
+            ],
+        }
+    )
+
+    assert result == CDEKOrderRegistrationResult(
+        order_uuid="cdek-order-uuid",
+        waybill_uuid="waybill-uuid-legacy",
+        waybill_url=None,
+    )
+
+
 def test_map_cdek_existing_order_response_returns_waybill_for_duplicate() -> None:
     result = map_cdek_existing_order_response(
         {
@@ -224,10 +246,10 @@ def test_map_cdek_order_info_response_picks_latest_status_by_date_time() -> None
                     {"code": "ACCEPTED", "date_time": "2026-05-24T10:00:00+0000"},
                     {"code": "INVALID", "date_time": "2026-05-24T10:00:05+0000"},
                 ],
-            },
                 "related_entities": [
                     {"type": "waybill", "uuid": "waybill-uuid-1"}
                 ],
+            },
         }
     )