Skip to content

Commit 415c7b2

Browse files
Mats-SXDarthMax
andcommitted
Fix monkey patching for unit tests
Co-authored-by: Max Kießling <max.kiessling@neotechnology.com>
1 parent cdb4637 commit 415c7b2

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

graphdatascience/tests/unit/conftest.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,8 +115,9 @@ def gds(runner: CollectingQueryRunner) -> Generator[GraphDataScience, None, None
115115
@pytest.fixture
116116
def aura_gds(runner: CollectingQueryRunner, mocker: MockerFixture) -> Generator[AuraGraphDataScience, None, None]:
117117
mocker.patch("graphdatascience.query_runner.neo4j_query_runner.Neo4jQueryRunner.create", return_value=runner)
118-
mocker.patch("graphdatascience.query_runner.aura_db_query_runner.AuraArrowQueryRunner.__new__", return_value=runner)
118+
mocker.patch("graphdatascience.query_runner.aura_db_query_runner.AuraDbQueryRunner.__new__", return_value=runner)
119119
mocker.patch("graphdatascience.query_runner.arrow_query_runner.ArrowQueryRunner.create", return_value=runner)
120+
mocker.patch("graphdatascience.query_runner.gds_arrow_client.GdsArrowClient.create", return_value=None)
120121
aura_gds = AuraGraphDataScience(
121122
gds_session_connection_info=DbmsConnectionInfo("address", "some", "auth"),
122123
aura_db_connection_info=DbmsConnectionInfo("address", "some", "auth"),

0 commit comments

Comments
 (0)