from unittest.mock import patch from django.contrib.auth import get_user_model from django.test import TestCase from django.urls import resolve from rest_framework.test import APIRequestFactory, force_authenticate from external_api_adapter.adapter import AdapterResponse from farm_hub.models import FarmHub, FarmType from .views import AnalyzeView, RiskSummaryView, RiskView class PestDetectionViewTests(TestCase): def setUp(self): self.factory = APIRequestFactory() self.user = get_user_model().objects.create_user( username="farmer", password="secret123", email="farmer@example.com", phone_number="09120000000", ) self.other_user = get_user_model().objects.create_user( username="other-farmer", password="secret123", email="other@example.com", phone_number="09120000001", ) self.farm_type = FarmType.objects.create(name="زراعی") self.farm = FarmHub.objects.create(owner=self.user, farm_type=self.farm_type, name="Farm 1") self.other_farm = FarmHub.objects.create(owner=self.other_user, farm_type=self.farm_type, name="Farm 2") @patch("pest_detection.views.external_api_request") def test_analyze_proxies_to_ai_service(self, mock_external_api_request): mock_external_api_request.return_value = AdapterResponse( status_code=200, data={ "data": { "result": { "has_issue": True, "category": "disease", "confidence": 0.93, "severity": "medium", "summary": "Leaf spot symptoms detected.", "detected_signs": ["Brown leaf spots"], "possible_causes": ["Fungal pressure"], "immediate_actions": ["Isolate affected plants"], "reasoning": ["Pattern matched common fungal lesions"], } } }, ) request = self.factory.post( "/api/pest-detection/analyze/", {"farm_uuid": str(self.farm.farm_uuid), "image_urls": ["https://example.com/leaf.jpg"]}, format="json", ) force_authenticate(request, user=self.user) response = AnalyzeView.as_view()(request) self.assertEqual(response.status_code, 200) self.assertEqual(response.data["code"], 200) self.assertEqual(response.data["data"]["category"], "disease") mock_external_api_request.assert_called_once_with( "ai", "/api/pest-disease/detect/", method="POST", payload={ "farm_uuid": str(self.farm.farm_uuid), "plant_name": "", "query": "", "image_urls": ["https://example.com/leaf.jpg"], }, ) def test_analyze_requires_at_least_one_image(self): request = self.factory.post( "/api/pest-detection/analyze/", {"farm_uuid": str(self.farm.farm_uuid)}, format="json", ) force_authenticate(request, user=self.user) response = AnalyzeView.as_view()(request) self.assertEqual(response.status_code, 400) self.assertEqual(response.data["code"], 400) self.assertIn("images", response.data["data"]) @patch("pest_detection.views.external_api_request") def test_risk_proxies_to_ai_service(self, mock_external_api_request): mock_external_api_request.return_value = AdapterResponse( status_code=200, data={ "data": { "result": { "summary": "Warm humidity raises fungal pressure.", "forecast_window": "72h", "overall_risk": "medium", "disease_risk": {"score": 0.7, "level": "medium", "likely_conditions": [], "reasoning": []}, "pest_risk": {"score": 0.4, "level": "low", "likely_conditions": [], "reasoning": []}, "key_drivers": ["High humidity"], "recommended_actions": ["Scout vulnerable rows"], } } }, ) request = self.factory.post( "/api/pest-detection/risk/", {"farm_uuid": str(self.farm.farm_uuid), "plant_name": "wheat"}, format="json", ) force_authenticate(request, user=self.user) response = RiskView.as_view()(request) self.assertEqual(response.status_code, 200) self.assertEqual(response.data["data"]["overall_risk"], "medium") mock_external_api_request.assert_called_once_with( "ai", "/api/pest-disease/risk/", method="POST", payload={ "farm_uuid": str(self.farm.farm_uuid), "plant_name": "wheat", "growth_stage": "", "query": "", }, ) @patch("pest_detection.views.external_api_request") def test_risk_summary_maps_response_shape(self, mock_external_api_request): mock_external_api_request.return_value = AdapterResponse( status_code=200, data={ "data": { "result": { "disease_risk": {"title": "Disease"}, "pest_risk": {"title": "Pest"}, "drivers": {"humidity": "high"}, } } }, ) request = self.factory.post( "/api/pest-disease/risk-summary/", {"farm_uuid": str(self.farm.farm_uuid)}, format="json", ) force_authenticate(request, user=self.user) response = RiskSummaryView.as_view()(request) self.assertEqual(response.status_code, 200) self.assertEqual(response.data["code"], 200) self.assertEqual(response.data["data"]["farm_uuid"], str(self.farm.farm_uuid)) self.assertEqual(response.data["data"]["diseaseRisk"]["title"], "Disease") self.assertEqual(response.data["data"]["pestRisk"]["title"], "Pest") self.assertEqual(response.data["data"]["drivers"], {"humidity": "high"}) mock_external_api_request.assert_called_once_with( "ai", "/api/pest-disease/risk-summary/", method="POST", payload={"farm_uuid": str(self.farm.farm_uuid)}, ) @patch("pest_detection.views.external_api_request") def test_risk_summary_post_uses_pest_disease_route(self, mock_external_api_request): mock_external_api_request.return_value = AdapterResponse( status_code=200, data={ "data": { "result": { "disease_risk": {"title": "Disease"}, "pest_risk": {"title": "Pest"}, "drivers": {"humidity": "high"}, } } }, ) request = self.factory.post( "/api/pest-disease/risk-summary/", {"farm_uuid": str(self.farm.farm_uuid)}, format="json", ) force_authenticate(request, user=self.user) response = RiskSummaryView.as_view()(request) self.assertEqual(response.status_code, 200) self.assertEqual(response.data["data"]["farm_uuid"], str(self.farm.farm_uuid)) mock_external_api_request.assert_called_once_with( "ai", "/api/pest-disease/risk-summary/", method="POST", payload={"farm_uuid": str(self.farm.farm_uuid)}, ) def test_risk_summary_rejects_foreign_farm_uuid(self): request = self.factory.post( "/api/pest-disease/risk-summary/", {"farm_uuid": str(self.other_farm.farm_uuid)}, format="json", ) force_authenticate(request, user=self.user) response = RiskSummaryView.as_view()(request) self.assertEqual(response.status_code, 404) self.assertEqual(response.data["code"], 404) self.assertEqual(response.data["data"]["farm_uuid"][0], "Farm not found.") def test_risk_summary_get_is_not_allowed(self): request = self.factory.get(f"/api/pest-disease/risk-summary/?farm_uuid={self.farm.farm_uuid}") force_authenticate(request, user=self.user) response = RiskSummaryView.as_view()(request) self.assertEqual(response.status_code, 405) def test_pest_disease_alias_routes_exist(self): self.assertIs(resolve("/api/pest-disease/detect/").func.view_class, AnalyzeView) self.assertIs(resolve("/api/pest-disease/risk/").func.view_class, RiskView) self.assertIs(resolve("/api/pest-disease/risk-summary/").func.view_class, RiskSummaryView)