UPDATE

2026-04-28 19:00:38 +03:30
parent 8471d648a3
commit cb60254c81
8 changed files with 971 additions and 86 deletions
@@ -0,0 +1,64 @@
+from __future__ import annotations
+
+from unittest.mock import patch
+
+from django.test import TestCase, override_settings
+from rest_framework.test import APIClient
+
+
+@override_settings(ROOT_URLCONF="irrigation.urls")
+class IrrigationRecommendApiTests(TestCase):
+    def setUp(self):
+        self.client = APIClient()
+
+    @patch("rag.services.irrigation.get_irrigation_recommendation")
+    def test_recommend_api_returns_water_balance(self, mock_get_irrigation_recommendation):
+        mock_get_irrigation_recommendation.return_value = {
+            "plan": {
+                "frequencyPerWeek": 4,
+                "durationMinutes": 38,
+                "bestTimeOfDay": "05:30 تا 08:00 صبح",
+                "moistureLevel": 72,
+                "warning": "در ساعات گرم روز آبیاری انجام نشود",
+            },
+            "water_balance": {
+                "active_kc": 0.93,
+                "crop_profile": {
+                    "kc_initial": 0.55,
+                    "kc_mid": 1.05,
+                    "kc_end": 0.78,
+                },
+                "daily": [
+                    {
+                        "forecast_date": "2025-02-12",
+                        "et0_mm": 5.4,
+                        "etc_mm": 4.9,
+                        "effective_rainfall_mm": 0,
+                        "gross_irrigation_mm": 17,
+                        "irrigation_timing": "05:30 - 07:00",
+                    }
+                ],
+            },
+            "timeline": [],
+            "sections": [],
+            "simulation_optimizer": {"engine": "crop_simulation_heuristic"},
+            "selected_irrigation_method": {"name": "آبیاری قطره‌ای"},
+        }
+
+        response = self.client.post(
+            "/recommend/",
+            data={
+                "farm_uuid": "11111111-1111-1111-1111-111111111111",
+                "plant_name": "گوجه‌فرنگی",
+                "growth_stage": "گلدهی",
+                "irrigation_method_name": "آبیاری قطره‌ای",
+            },
+            format="json",
+        )
+
+        self.assertEqual(response.status_code, 200)
+        data = response.json()["data"]
+        self.assertIn("water_balance", data)
+        self.assertEqual(data["water_balance"]["active_kc"], 0.93)
+        self.assertNotIn("simulation_optimizer", data)
+        self.assertNotIn("selected_irrigation_method", data)