test(chat): use transaction case for async stream assertion
This commit is contained in:
@@ -3,7 +3,7 @@ from unittest.mock import patch
|
|||||||
|
|
||||||
from django.contrib.auth import get_user_model
|
from django.contrib.auth import get_user_model
|
||||||
from django.test import TestCase
|
from django.test import TestCase
|
||||||
from rest_framework.test import APITestCase
|
from rest_framework.test import APITransactionTestCase
|
||||||
|
|
||||||
from adventures.models import Collection, CollectionItineraryItem
|
from adventures.models import Collection, CollectionItineraryItem
|
||||||
from chat.agent_tools import add_to_itinerary, get_trip_details
|
from chat.agent_tools import add_to_itinerary, get_trip_details
|
||||||
@@ -165,7 +165,7 @@ class ChatViewSetToolValidationBoundaryTests(TestCase):
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
class ChatViewSetSearchPlacesClarificationTests(APITestCase):
|
class ChatViewSetSearchPlacesClarificationTests(APITransactionTestCase):
|
||||||
@patch("chat.views.execute_tool")
|
@patch("chat.views.execute_tool")
|
||||||
@patch("chat.views.stream_chat_completion")
|
@patch("chat.views.stream_chat_completion")
|
||||||
@patch("integrations.utils.auto_profile.update_auto_preference_profile")
|
@patch("integrations.utils.auto_profile.update_auto_preference_profile")
|
||||||
|
|||||||
Reference in New Issue
Block a user