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-23 15:53:59 +03:30
parent 36d6b05a7f
commit 72588fe12c
4 changed files with 892 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files
farm_ai_assistant/tests.py
+171 -25
View File
@@ -5,11 +5,18 @@ from rest_framework.test import APIRequestFactory, force_authenticate
from farm_hub.models import FarmHub, FarmType
from .models import Conversation, Message
from .views import ChatTaskStatusView
from .views import (
ChatDetailView,
ChatListCreateView,
ChatMessagesView,
ChatTaskCreateView,
ChatTaskStatusView,
ContextView,
)
@override_settings(USE_EXTERNAL_API_MOCK=True)
class ChatTaskStatusViewTests(TestCase):
class FarmAiAssistantOptionalFarmUuidTests(TestCase):
def setUp(self):
self.factory = APIRequestFactory()
self.user = get_user_model().objects.create_user(
@@ -24,14 +31,94 @@ class ChatTaskStatusViewTests(TestCase):
farm_type=self.farm_type,
name="Farm 1",
)
self.conversation = Conversation.objects.create(
def test_context_allows_missing_farm_uuid(self):
request = self.factory.get("/api/farm-ai-assistant/context/")
force_authenticate(request, user=self.user)
response = ContextView.as_view()(request)
self.assertEqual(response.status_code, 200)
self.assertEqual(response.data["status"], "success")
self.assertIsNone(response.data["data"]["farm_uuid"])
def test_chat_task_create_allows_missing_farm_uuid_for_landing_chat(self):
request = self.factory.post(
"/api/farm-ai-assistant/chat/task/",
{"content": "Give me a landing page recommendation"},
format="json",
)
force_authenticate(request, user=self.user)
response = ChatTaskCreateView.as_view()(request)
self.assertEqual(response.status_code, 202)
self.assertEqual(response.data["status"], "success")
self.assertEqual(response.data["data"]["task_id"], "farm-ai-chat-task-123")
self.assertIsNone(response.data["data"]["farm_uuid"])
conversation = Conversation.objects.get(uuid=response.data["data"]["conversation_id"])
self.assertIsNone(conversation.farm)
self.assertEqual(conversation.owner_id, self.user.id)
user_message = conversation.messages.get(role=Message.ROLE_USER)
self.assertIsNone(user_message.farm)
self.assertIsNone(user_message.raw_response["farm_uuid"])
def test_status_success_without_farm_uuid_persists_assistant_message(self):
conversation = Conversation.objects.create(
owner=self.user,
farm=self.farm,
title="Irrigation chat",
farm=None,
title="Landing chat",
farm_context={},
)
self.user_message = Message.objects.create(
conversation=self.conversation,
Message.objects.create(
conversation=conversation,
farm=None,
role=Message.ROLE_USER,
content="What should I plant?",
raw_response={
"task_id": "farm-ai-chat-task-123",
"status": "PENDING",
"status_url": "/api/tasks/farm-ai-chat-task-123/status/",
"farm_uuid": None,
},
)
request = self.factory.get("/api/farm-ai-assistant/chat/task/farm-ai-chat-task-123/status/")
force_authenticate(request, user=self.user)
response = ChatTaskStatusView.as_view()(request, task_id="farm-ai-chat-task-123")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.data["status"], "success")
self.assertEqual(response.data["data"]["task_id"], "farm-ai-chat-task-123")
self.assertEqual(response.data["data"]["status"], "SUCCESS")
self.assertEqual(response.data["data"]["conversation_id"], str(conversation.uuid))
self.assertIsNone(response.data["data"]["farm_uuid"])
self.assertEqual(response.data["data"]["result"]["content"], "Here is the recommended plan.")
self.assertEqual(response.data["data"]["result"]["task_id"], "farm-ai-chat-task-123")
assistant_message = (
conversation.messages.filter(role=Message.ROLE_ASSISTANT)
.order_by("-created_at")
.first()
)
self.assertIsNotNone(assistant_message)
self.assertIsNone(assistant_message.farm)
self.assertEqual(assistant_message.content, "Here is the recommended plan.")
self.assertIsNone(assistant_message.raw_response["farm_uuid"])
self.assertEqual(assistant_message.raw_response["task_id"], "farm-ai-chat-task-123")
def test_status_success_with_farm_uuid_still_works_for_farm_chat(self):
conversation = Conversation.objects.create(
owner=self.user,
farm=self.farm,
title="Farm chat",
farm_context={},
)
Message.objects.create(
conversation=conversation,
farm=self.farm,
role=Message.ROLE_USER,
content="What is the best irrigation plan?",
@@ -43,7 +130,6 @@ class ChatTaskStatusViewTests(TestCase):
},
)
def test_status_success_uses_chat_mock_result_and_persists_assistant_message(self):
request = self.factory.get(
f"/api/farm-ai-assistant/chat/task/farm-ai-chat-task-123/status/?farm_uuid={self.farm.farm_uuid}"
)
@@ -52,23 +138,83 @@ class ChatTaskStatusViewTests(TestCase):
response = ChatTaskStatusView.as_view()(request, task_id="farm-ai-chat-task-123")
self.assertEqual(response.status_code, 200)
self.assertEqual(response.data["status"], "success")
self.assertEqual(response.data["data"]["task_id"], "farm-ai-chat-task-123")
self.assertEqual(response.data["data"]["status"], "SUCCESS")
self.assertEqual(response.data["data"]["conversation_id"], str(self.conversation.uuid))
self.assertEqual(response.data["data"]["conversation_id"], str(conversation.uuid))
self.assertEqual(response.data["data"]["farm_uuid"], str(self.farm.farm_uuid))
self.assertEqual(response.data["data"]["result"]["content"], "Here is the recommended plan.")
self.assertEqual(len(response.data["data"]["result"]["sections"]), 3)
self.assertEqual(response.data["data"]["result"]["task_id"], "farm-ai-chat-task-123")
assistant_message = (
self.conversation.messages.filter(role=Message.ROLE_ASSISTANT)
.order_by("-created_at")
.first()
def test_chat_list_create_messages_and_delete_work_without_farm_uuid(self):
landing_conversation = Conversation.objects.create(
owner=self.user,
farm=None,
title="Landing chat",
farm_context={"source": "landing"},
)
self.assertIsNotNone(assistant_message)
self.assertEqual(assistant_message.farm_id, self.farm.id)
self.assertEqual(assistant_message.content, "Here is the recommended plan.")
self.assertEqual(assistant_message.raw_response["task_id"], "farm-ai-chat-task-123")
self.assertEqual(assistant_message.raw_response["farm_uuid"], str(self.farm.farm_uuid))
self.assertEqual(len(assistant_message.raw_response["sections"]), 3)
Message.objects.create(
conversation=landing_conversation,
farm=None,
role=Message.ROLE_USER,
content="Hello from landing",
raw_response={"farm_uuid": None},
)
farm_conversation = Conversation.objects.create(
owner=self.user,
farm=self.farm,
title="Farm chat",
farm_context={},
)
Message.objects.create(
conversation=farm_conversation,
farm=self.farm,
role=Message.ROLE_USER,
content="Hello from farm",
raw_response={"farm_uuid": str(self.farm.farm_uuid)},
)
list_request = self.factory.get("/api/farm-ai-assistant/chats/")
force_authenticate(list_request, user=self.user)
list_response = ChatListCreateView.as_view()(list_request)
self.assertEqual(list_response.status_code, 200)
self.assertEqual(len(list_response.data["data"]), 1)
self.assertEqual(list_response.data["data"][0]["id"], str(landing_conversation.uuid))
self.assertIsNone(list_response.data["data"][0]["farm_uuid"])
create_request = self.factory.post(
"/api/farm-ai-assistant/chats/",
{"title": "New landing conversation"},
format="json",
)
force_authenticate(create_request, user=self.user)
create_response = ChatListCreateView.as_view()(create_request)
self.assertEqual(create_response.status_code, 201)
self.assertIsNone(create_response.data["data"]["farm_uuid"])
created_conversation = Conversation.objects.get(uuid=create_response.data["data"]["id"])
self.assertIsNone(created_conversation.farm)
messages_request = self.factory.get(
f"/api/farm-ai-assistant/chats/{landing_conversation.uuid}/messages/"
)
force_authenticate(messages_request, user=self.user)
messages_response = ChatMessagesView.as_view()(
messages_request,
conversation_id=landing_conversation.uuid,
)
self.assertEqual(messages_response.status_code, 200)
self.assertEqual(messages_response.data["data"]["conversation_id"], str(landing_conversation.uuid))
self.assertIsNone(messages_response.data["data"]["farm_uuid"])
self.assertEqual(len(messages_response.data["data"]["messages"]), 1)
self.assertIsNone(messages_response.data["data"]["messages"][0]["farm_uuid"])
delete_request = self.factory.delete(f"/api/farm-ai-assistant/chats/{landing_conversation.uuid}/")
force_authenticate(delete_request, user=self.user)
delete_response = ChatDetailView.as_view()(
delete_request,
conversation_id=landing_conversation.uuid,
)
self.assertEqual(delete_response.status_code, 200)
self.assertEqual(delete_response.data["data"]["conversation_id"], str(landing_conversation.uuid))
self.assertIsNone(delete_response.data["data"]["farm_uuid"])
self.assertFalse(Conversation.objects.filter(uuid=landing_conversation.uuid).exists())