UPDATE

2026-04-07 01:08:41 +03:30
parent ff464cb4a5
commit 5acee1fa2c
5 changed files with 198 additions and 34 deletions
@@ -1,4 +1,5 @@
 from datetime import date
+from unittest.mock import patch
 import uuid

 from django.test import TestCase
@@ -193,3 +194,68 @@ class FarmDataUpsertApiTests(TestCase):
        self.assertEqual(str(farm.center_location.latitude), "50.010000")
        self.assertEqual(str(farm.center_location.longitude), "50.010000")
        self.assertIsNone(farm.weather_forecast_id)
+
+    @patch("farm_data.services.update_weather_for_location")
+    @patch("farm_data.services.fetch_soil_data_for_coordinates")
+    def test_post_fetches_missing_location_and_weather_data(
+        self,
+        mock_fetch_soil_data_for_coordinates,
+        mock_update_weather_for_location,
+    ):
+        missing_boundary = square_boundary_for_center(36.0, 52.0)
+        farm_uuid = uuid.uuid4()
+
+        def soil_side_effect(latitude, longitude, task_id="", progress_callback=None):
+            location = SoilLocation.objects.get(
+                latitude="36.000000",
+                longitude="52.000000",
+            )
+            SoilDepthData.objects.update_or_create(
+                soil_location=location,
+                depth_label="0-5cm",
+                defaults={"clay": 20.0},
+            )
+            SoilDepthData.objects.update_or_create(
+                soil_location=location,
+                depth_label="5-15cm",
+                defaults={"clay": 18.0},
+            )
+            SoilDepthData.objects.update_or_create(
+                soil_location=location,
+                depth_label="15-30cm",
+                defaults={"clay": 16.0},
+            )
+            return {"status": "completed", "location_id": location.id, "depths": ["0-5cm", "5-15cm", "15-30cm"]}
+
+        def weather_side_effect(location):
+            WeatherForecast.objects.update_or_create(
+                location=location,
+                forecast_date=date(2026, 4, 12),
+                defaults={
+                    "temperature_min": 10.0,
+                    "temperature_max": 20.0,
+                    "temperature_mean": 15.0,
+                },
+            )
+            return {"status": "success", "location_id": location.id, "days_updated": 1}
+
+        mock_fetch_soil_data_for_coordinates.side_effect = soil_side_effect
+        mock_update_weather_for_location.side_effect = weather_side_effect
+
+        response = self.client.post(
+            "/api/farm-data/",
+            data={
+                "farm_uuid": str(farm_uuid),
+                "farm_boundary": missing_boundary,
+                "sensor_payload": {"sensor-7-1": {"soil_moisture": 44.0}},
+            },
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, 201)
+        mock_fetch_soil_data_for_coordinates.assert_called_once()
+        mock_update_weather_for_location.assert_called_once()
+
+        farm = SensorData.objects.get(farm_uuid=farm_uuid)
+        self.assertEqual(farm.center_location.depths.count(), 3)
+        self.assertIsNotNone(farm.weather_forecast_id)