UPDATE
This commit is contained in:
@@ -0,0 +1,197 @@
|
||||
from datetime import timedelta
|
||||
from unittest.mock import patch
|
||||
|
||||
from django.contrib.auth import get_user_model
|
||||
from django.test import SimpleTestCase, TestCase
|
||||
from django.utils import timezone
|
||||
from rest_framework.test import APIRequestFactory, force_authenticate
|
||||
|
||||
from external_api_adapter.adapter import AdapterResponse
|
||||
from farm_hub.models import FarmHub, FarmType
|
||||
from notifications.models import FarmNotification
|
||||
|
||||
from .models import FarmAlert
|
||||
from .serializers import FarmAlertsTrackerRequestSerializer
|
||||
from .views import AlertTrackerView
|
||||
|
||||
|
||||
class FarmAlertsTrackerRequestSerializerTests(SimpleTestCase):
|
||||
def test_accepts_farm_uuid_and_optional_alerts(self):
|
||||
serializer = FarmAlertsTrackerRequestSerializer(
|
||||
data={
|
||||
"farm_uuid": "11111111-1111-1111-1111-111111111111",
|
||||
"alerts": [
|
||||
{
|
||||
"alert_id": "soil-1",
|
||||
"level": "warning",
|
||||
"title": "Low moisture",
|
||||
}
|
||||
],
|
||||
}
|
||||
)
|
||||
|
||||
self.assertTrue(serializer.is_valid(), serializer.errors)
|
||||
|
||||
def test_rejects_extra_fields(self):
|
||||
serializer = FarmAlertsTrackerRequestSerializer(
|
||||
data={
|
||||
"farm_uuid": "11111111-1111-1111-1111-111111111111",
|
||||
"unexpected": True,
|
||||
}
|
||||
)
|
||||
|
||||
self.assertFalse(serializer.is_valid())
|
||||
self.assertEqual(serializer.errors["unexpected"][0], "This field is not allowed.")
|
||||
|
||||
|
||||
class FarmAlertsTrackerViewTests(TestCase):
|
||||
def setUp(self):
|
||||
self.factory = APIRequestFactory()
|
||||
self.user = get_user_model().objects.create_user(
|
||||
username="farm-alerts-user",
|
||||
password="secret123",
|
||||
email="farm-alerts@example.com",
|
||||
phone_number="09120000999",
|
||||
)
|
||||
self.other_user = get_user_model().objects.create_user(
|
||||
username="farm-alerts-other",
|
||||
password="secret123",
|
||||
email="farm-alerts-other@example.com",
|
||||
phone_number="09120000998",
|
||||
)
|
||||
self.farm_type = FarmType.objects.create(name="مرکبات")
|
||||
self.farm = FarmHub.objects.create(owner=self.user, farm_type=self.farm_type, name="Farm Alerts")
|
||||
|
||||
@patch("farm_alerts.views.external_api_request")
|
||||
def test_tracker_persists_incoming_alerts_and_sends_recent_notifications_to_ai(self, mock_external_api_request):
|
||||
recent_notification = FarmNotification.objects.create(
|
||||
farm=self.farm,
|
||||
endpoint="tracker",
|
||||
title="Recent alert",
|
||||
message="Recent notification",
|
||||
level="warning",
|
||||
)
|
||||
old_notification = FarmNotification.objects.create(
|
||||
farm=self.farm,
|
||||
endpoint="tracker",
|
||||
title="Old alert",
|
||||
message="Old notification",
|
||||
level="info",
|
||||
)
|
||||
FarmNotification.objects.filter(id=old_notification.id).update(created_at=timezone.now() - timedelta(days=4))
|
||||
old_notification.refresh_from_db()
|
||||
|
||||
mock_external_api_request.return_value = AdapterResponse(
|
||||
status_code=200,
|
||||
data={
|
||||
"data": {
|
||||
"headline": "وضعیت هشدارها",
|
||||
"overview": "دو مورد نیاز به پیگیری دارد.",
|
||||
"status_level": "warning",
|
||||
"tracker": {"active": 2},
|
||||
"notifications": [
|
||||
{
|
||||
"title": "افت رطوبت خاک",
|
||||
"message": "تنش رطوبتی ادامه دارد.",
|
||||
"level": "warning",
|
||||
"suggested_action": "آبیاری جبرانی انجام شود.",
|
||||
"source_alert_id": "soil-1",
|
||||
"source_metric_type": "moisture",
|
||||
"payload": {"current_value": 38.5},
|
||||
}
|
||||
],
|
||||
"structured_context": {"source": "ai"},
|
||||
}
|
||||
},
|
||||
)
|
||||
|
||||
request = self.factory.post(
|
||||
"/api/farm-alerts/tracker/",
|
||||
{
|
||||
"farm_uuid": str(self.farm.farm_uuid),
|
||||
"alerts": [
|
||||
{
|
||||
"alert_id": "soil-1",
|
||||
"level": "danger",
|
||||
"title": "افت رطوبت خاک",
|
||||
"message": "رطوبت خاک کمتر از حد مطلوب است.",
|
||||
"suggested_action": "آبیاری اصلاحی بررسی شود.",
|
||||
"source_metric_type": "moisture",
|
||||
"payload": {"current_value": 38.5},
|
||||
}
|
||||
],
|
||||
},
|
||||
format="json",
|
||||
)
|
||||
force_authenticate(request, user=self.user)
|
||||
|
||||
response = AlertTrackerView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertEqual(FarmAlert.objects.filter(farm=self.farm).count(), 1)
|
||||
|
||||
saved_alert = FarmAlert.objects.get(farm=self.farm)
|
||||
self.assertEqual(saved_alert.external_alert_id, "soil-1")
|
||||
self.assertEqual(saved_alert.color, "error")
|
||||
self.assertEqual(saved_alert.source_metric_type, "moisture")
|
||||
|
||||
mock_external_api_request.assert_called_once()
|
||||
outbound_payload = mock_external_api_request.call_args.kwargs["payload"]
|
||||
self.assertEqual(outbound_payload["farm_uuid"], str(self.farm.farm_uuid))
|
||||
self.assertEqual(len(outbound_payload["alerts"]), 1)
|
||||
self.assertEqual(len(outbound_payload["recent_notifications"]), 1)
|
||||
self.assertEqual(outbound_payload["recent_notifications"][0]["id"], recent_notification.id)
|
||||
|
||||
self.assertEqual(response.data["data"]["headline"], "وضعیت هشدارها")
|
||||
self.assertEqual(response.data["data"]["status_level"], "warning")
|
||||
self.assertEqual(len(response.data["data"]["notifications"]), 1)
|
||||
self.assertEqual(response.data["data"]["notifications"][0]["endpoint"], "tracker")
|
||||
|
||||
persisted_notification = FarmNotification.objects.filter(
|
||||
farm=self.farm,
|
||||
title="افت رطوبت خاک",
|
||||
endpoint="tracker",
|
||||
).latest("id")
|
||||
self.assertEqual(persisted_notification.source_alert_id, "soil-1")
|
||||
self.assertEqual(persisted_notification.suggested_action, "آبیاری جبرانی انجام شود.")
|
||||
|
||||
@patch("farm_alerts.views.external_api_request")
|
||||
def test_tracker_limits_recent_notifications_to_ten(self, mock_external_api_request):
|
||||
for index in range(12):
|
||||
FarmNotification.objects.create(
|
||||
farm=self.farm,
|
||||
endpoint="tracker",
|
||||
title=f"Notification {index}",
|
||||
message="msg",
|
||||
)
|
||||
|
||||
mock_external_api_request.return_value = AdapterResponse(
|
||||
status_code=200,
|
||||
data={"data": {"headline": "", "overview": "", "status_level": "info", "notifications": []}},
|
||||
)
|
||||
|
||||
request = self.factory.post(
|
||||
"/api/farm-alerts/tracker/",
|
||||
{"farm_uuid": str(self.farm.farm_uuid)},
|
||||
format="json",
|
||||
)
|
||||
force_authenticate(request, user=self.user)
|
||||
|
||||
response = AlertTrackerView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
outbound_payload = mock_external_api_request.call_args.kwargs["payload"]
|
||||
self.assertEqual(len(outbound_payload["recent_notifications"]), 10)
|
||||
|
||||
def test_tracker_rejects_unowned_farm(self):
|
||||
request = self.factory.post(
|
||||
"/api/farm-alerts/tracker/",
|
||||
{"farm_uuid": str(self.farm.farm_uuid)},
|
||||
format="json",
|
||||
)
|
||||
force_authenticate(request, user=self.other_user)
|
||||
|
||||
response = AlertTrackerView.as_view()(request)
|
||||
|
||||
self.assertEqual(response.status_code, 400)
|
||||
self.assertEqual(response.data["farm_uuid"][0], "Farm not found.")
|
||||
Reference in New Issue
Block a user