chore: import upstream snapshot with attribution
CI / Run CI (push) Has been cancelled
CI / check-backend (push) Has been cancelled
CI / check-frontend (push) Has been cancelled
CI / tests (push) Has been cancelled
CI / e2e-tests (push) Has been cancelled
Copilot Setup Steps / copilot-setup-steps (push) Has been cancelled
CI / Run CI (push) Has been cancelled
CI / check-backend (push) Has been cancelled
CI / check-frontend (push) Has been cancelled
CI / tests (push) Has been cancelled
CI / e2e-tests (push) Has been cancelled
Copilot Setup Steps / copilot-setup-steps (push) Has been cancelled
This commit is contained in:
@@ -0,0 +1,18 @@
|
||||
from unittest.mock import AsyncMock, Mock
|
||||
|
||||
from chainlit.data import get_data_layer
|
||||
|
||||
|
||||
async def test_get_data_layer(
|
||||
mock_data_layer: AsyncMock,
|
||||
mock_get_data_layer: Mock,
|
||||
):
|
||||
# Check whether the data layer is properly set
|
||||
assert mock_data_layer == get_data_layer()
|
||||
|
||||
mock_get_data_layer.assert_called_once()
|
||||
|
||||
# Getting the data layer again, should not result in additional call
|
||||
assert mock_data_layer == get_data_layer()
|
||||
|
||||
mock_get_data_layer.assert_called_once()
|
||||
Reference in New Issue
Block a user