Unit tests for LiteralDataLayer.

backend/tests/conftest.py

@@ -1,3 +1,4 @@
+import datetime
 from contextlib import asynccontextmanager
 from unittest.mock import AsyncMock, Mock
 
@@ -10,10 +11,12 @@
 
 
 @pytest.fixture
-def mock_persisted_user():
- mock = Mock(spec=PersistedUser)
- mock.id = "test_user_id"
- return mock
+def persisted_test_user():
+ return PersistedUser(
+ id="test_user_id",
+ createdAt=datetime.datetime.now().isoformat(),
+ identifier="test_user_identifier",
+ )
 
 
 @pytest.fixture
@@ -44,8 +47,8 @@ async def create_chainlit_context(mock_session):
 
 
 @pytest_asyncio.fixture
-async def mock_chainlit_context(mock_persisted_user, mock_session):
- mock_session.user = mock_persisted_user
+async def mock_chainlit_context(persisted_test_user, mock_session):
+ mock_session.user = persisted_test_user
  return create_chainlit_context(mock_session)
 
 

backend/tests/data/conftest.py

@@ -3,6 +3,7 @@
 from unittest.mock import AsyncMock
 
 from chainlit.data.base import BaseStorageClient
+from chainlit.user import User
 
 
 @pytest.fixture
@@ -13,3 +14,8 @@ def mock_storage_client():
  "object_key": "test_user/test_element/test.txt",
  }
  return mock_client
+
+
+@pytest.fixture
+def test_user() -> User:
+ return User(identifier="test_user_identifier", metadata={})