Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
shangyian committed Feb 4, 2025
1 parent 5d97115 commit 618e841
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions datajunction-clients/python/tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -274,6 +274,7 @@ def module__server( # pylint: disable=too-many-statements
module__session: AsyncSession,
module__settings: Settings,
module__query_service_client: QueryServiceClient,
module_mocker,
) -> Iterator[TestClient]:
"""
Create a mock server for testing APIs that contains a mock query service.
Expand All @@ -290,6 +291,11 @@ async def get_session_override() -> AsyncSession:
def get_settings_override() -> Settings:
return module__settings

module_mocker.patch(
"datajunction_server.api.materializations.get_query_service_client",
get_query_service_client_override,
)

app.dependency_overrides[get_session] = get_session_override
app.dependency_overrides[get_settings] = get_settings_override
app.dependency_overrides[
Expand Down

0 comments on commit 618e841

Please sign in to comment.