Fix: incorrect attribute in convo info service (#11503)

This commit is contained in:
Rohit Malhotra
2025-10-24 18:33:36 -04:00
committed by GitHub
parent 47776ae2ad
commit 2631294e79
2 changed files with 22 additions and 6 deletions

View File

@@ -73,7 +73,8 @@ def service(async_session) -> SQLAppConversationInfoService:
def service_with_user(async_session) -> SQLAppConversationInfoService:
"""Create a SQLAppConversationInfoService instance with a user_id for testing."""
return SQLAppConversationInfoService(
db_session=async_session, user_id='test_user_123'
db_session=async_session,
user_context=SpecifyUserContext(user_id='test_user_123'),
)
@@ -444,6 +445,23 @@ class TestSQLAppConversationInfoService:
count = await service.count_app_conversation_info()
assert count == len(multiple_conversation_infos)
@pytest.mark.asyncio
async def test_count_conversation_info_with_user_id(
self,
service_with_user: SQLAppConversationInfoService,
multiple_conversation_infos: list[AppConversationInfo],
):
"""Test count without any filters."""
# Save all conversation infos
for info in multiple_conversation_infos:
await service_with_user.save_app_conversation_info(info)
# Count without filters
count = await service_with_user.count_app_conversation_info(
updated_at__gte=datetime(1900, 1, 1, tzinfo=timezone.utc)
)
assert count == len(multiple_conversation_infos)
@pytest.mark.asyncio
async def test_count_conversation_info_with_filters(
self,