sajad-dev/Backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

UPDATE

Browse Source
This commit is contained in:
Mohammad Sajad Pourajam
2026-04-27 00:40:59 +03:30
parent 2cd96ceec6
commit 64e67c282c
56 changed files with 3912 additions and 745 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pest_detection/tests.py
+229
View File
@@ -0,0 +1,229 @@
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)