UPDATE

2026-04-27 00:40:59 +03:30
parent 2cd96ceec6
commit 64e67c282c
56 changed files with 3912 additions and 745 deletions
@@ -0,0 +1,110 @@
+from django.contrib.auth import get_user_model
+from django.test import TestCase
+from django.urls import Resolver404, resolve
+from rest_framework.test import APIRequestFactory, force_authenticate
+
+from external_api_adapter.adapter import AdapterResponse
+from farm_hub.models import FarmHub, FarmType
+from unittest.mock import patch
+
+from .views import CropHealthSummaryView, NdviHealthView
+
+
+class NdviHealthViewTests(TestCase):
+    def setUp(self):
+        self.factory = APIRequestFactory()
+        self.user = get_user_model().objects.create_user(
+            username="ndvi-user",
+            password="secret123",
+            email="ndvi@example.com",
+            phone_number="09120000020",
+        )
+        self.other_user = get_user_model().objects.create_user(
+            username="ndvi-other-user",
+            password="secret123",
+            email="ndvi-other@example.com",
+            phone_number="09120000021",
+        )
+        self.farm_type = FarmType.objects.create(name="NDVI Farm Type")
+        self.farm = FarmHub.objects.create(
+            owner=self.user,
+            farm_type=self.farm_type,
+            name="NDVI Farm",
+        )
+
+    @patch("crop_health.views.external_api_request")
+    def test_post_ndvi_health_returns_expected_payload(self, mock_external_api_request):
+        mock_external_api_request.return_value = AdapterResponse(
+            status_code=200,
+            data={"data": {"result": {"ndviIndex": 0.78, "mean_ndvi": 0.78, "vegetation_health_class": "Healthy", "satellite_source": "sentinel-2"}}},
+        )
+
+        request = self.factory.post(
+            "/api/crop-health/ndvi-health/",
+            {"farm_uuid": str(self.farm.farm_uuid)},
+            format="json",
+        )
+        force_authenticate(request, user=self.user)
+
+        response = NdviHealthView.as_view()(request)
+
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(response.data["code"], 200)
+        self.assertEqual(response.data["msg"], "success")
+        self.assertEqual(response.data["data"]["ndviIndex"], 0.78)
+        self.assertEqual(response.data["data"]["mean_ndvi"], 0.78)
+        self.assertEqual(response.data["data"]["vegetation_health_class"], "Healthy")
+        self.assertEqual(response.data["data"]["satellite_source"], "sentinel-2")
+        mock_external_api_request.assert_called_once_with(
+            "ai",
+            "/api/soil-data/ndvi-health/",
+            method="POST",
+            payload={"farm_uuid": str(self.farm.farm_uuid)},
+        )
+
+    def test_post_ndvi_health_requires_farm_uuid(self):
+        request = self.factory.post("/api/crop-health/ndvi-health/", {}, format="json")
+        force_authenticate(request, user=self.user)
+
+        response = NdviHealthView.as_view()(request)
+
+        self.assertEqual(response.status_code, 400)
+        self.assertIn("farm_uuid", response.data)
+
+    def test_post_ndvi_health_returns_404_for_missing_farm(self):
+        request = self.factory.post(
+            "/api/crop-health/ndvi-health/",
+            {"farm_uuid": "11111111-1111-1111-1111-111111111111"},
+            format="json",
+        )
+        force_authenticate(request, user=self.user)
+
+        response = NdviHealthView.as_view()(request)
+
+        self.assertEqual(response.status_code, 404)
+        self.assertEqual(response.data["msg"], "Farm not found.")
+
+    def test_post_ndvi_health_does_not_expose_other_users_farm(self):
+        request = self.factory.post(
+            "/api/crop-health/ndvi-health/",
+            {"farm_uuid": str(self.farm.farm_uuid)},
+            format="json",
+        )
+        force_authenticate(request, user=self.other_user)
+
+        response = NdviHealthView.as_view()(request)
+
+        self.assertEqual(response.status_code, 404)
+        self.assertEqual(response.data["msg"], "Farm not found.")
+
+    def test_crop_health_routes_exist(self):
+        self.assertIs(resolve("/api/crop-health/ndvi-health/").func.view_class, NdviHealthView)
+        self.assertIs(resolve("/api/crop-health/summary/").func.view_class, CropHealthSummaryView)
+
+    def test_removed_soil_health_alias_routes_no_longer_resolve(self):
+        with self.assertRaises(Resolver404):
+            resolve("/api/soil/health/ndvi-health/")
+        with self.assertRaises(Resolver404):
+            resolve("/api/soil/health/summary/")
+        with self.assertRaises(Resolver404):
+            resolve("/api/soil-data/ndvi-health/")