fix country code

tests/controllers/v1/test_delivery.py

@@ -177,6 +177,7 @@ def test_post_delivery_price_returns_prices_and_delegates_to_service() -> None:
         entity=DeliveryEntity.INDIVIDUAL,
         from_city="Moscow",
         to_city="Kazan",
+        country_code=None,
         weight_kg=2.5,
         length_cm=30.0,
         width_cm=20.0,
@@ -184,6 +185,28 @@ def test_post_delivery_price_returns_prices_and_delegates_to_service() -> None:
     )
 
 
+def test_post_delivery_price_accepts_optional_country_code() -> None:
+    service = StubAggregatorService(response=[])
+    app = create_app()
+    _install_service_override(app, service)
+    payload = _valid_payload()
+    payload["country_code"] = "kz"
+
+    async def run_request() -> httpx.Response:
+        transport = httpx.ASGITransport(app=app)
+        async with httpx.AsyncClient(
+            transport=transport,
+            base_url="http://testserver",
+        ) as client:
+            return await client.post("/api/v1/delivery/price", json=payload)
+
+    response = asyncio.run(run_request())
+
+    assert response.status_code == 200
+    assert len(service.calls) == 1
+    assert service.calls[0].country_code == "kz"
+
+
 def test_post_delivery_price_rejects_invalid_payload() -> None:
     service = StubAggregatorService(response=[])
     app = create_app()