Skip to content

Commit 4dfe49d

Browse files
committed
Fix mocking of methods
1 parent fcd922d commit 4dfe49d

File tree

1 file changed

+8
-22
lines changed

1 file changed

+8
-22
lines changed

graphdatascience/tests/unit/test_gds_sessions.py

Lines changed: 8 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -131,10 +131,8 @@ def test_create_session(mocker: MockerFixture, aura_api: AuraApi) -> None:
131131
def assert_db_credentials(*args: List[Any], **kwargs: Dict[str, Any]) -> None:
132132
assert kwargs == {"gds_url": "fake-url", "gds_user": "neo4j", "initial_pw": "fake-pw", "new_pw": "db_pw"}
133133

134-
mocker.patch(
135-
"graphdatascience.gds_session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs
136-
)
137-
mocker.patch("graphdatascience.gds_session.gds_sessions.GdsSessions._change_initial_pw", assert_db_credentials)
134+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs)
135+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._change_initial_pw", assert_db_credentials)
138136

139137
gds_credentials = sessions.get_or_create(
140138
"my-session",
@@ -163,12 +161,8 @@ def test_create_default_session(mocker: MockerFixture, aura_api: AuraApi) -> Non
163161
sessions = GdsSessions(AuraAPICredentials("", "", "placeholder"))
164162
sessions._aura_api = aura_api
165163

166-
mocker.patch(
167-
"graphdatascience.gds_session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs
168-
)
169-
mocker.patch(
170-
"graphdatascience.gds_session.gds_sessions.GdsSessions._change_initial_pw", lambda *args, **kwargs: kwargs
171-
)
164+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs)
165+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._change_initial_pw", lambda *args, **kwargs: kwargs)
172166

173167
sessions.get_or_create(
174168
"my-session",
@@ -187,12 +181,8 @@ def test_create_session_override_region(mocker: MockerFixture, aura_api: AuraApi
187181
sessions = GdsSessions(AuraAPICredentials("", "", "placeholder"))
188182
sessions._aura_api = aura_api
189183

190-
mocker.patch(
191-
"graphdatascience.gds_session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs
192-
)
193-
mocker.patch(
194-
"graphdatascience.gds_session.gds_sessions.GdsSessions._change_initial_pw", lambda *args, **kwargs: kwargs
195-
)
184+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs)
185+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._change_initial_pw", lambda *args, **kwargs: kwargs)
196186

197187
sessions.get_or_create(
198188
"my-session",
@@ -211,12 +201,8 @@ def test_get_or_create(mocker: MockerFixture, aura_api: AuraApi) -> None:
211201
sessions = GdsSessions(AuraAPICredentials("", "", "placeholder"))
212202
sessions._aura_api = aura_api
213203

214-
mocker.patch(
215-
"graphdatascience.gds_session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs
216-
)
217-
mocker.patch(
218-
"graphdatascience.gds_session.gds_sessions.GdsSessions._change_initial_pw", lambda *args, **kwargs: None
219-
)
204+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._construct_client", lambda *args, **kwargs: kwargs)
205+
mocker.patch("graphdatascience.session.gds_sessions.GdsSessions._change_initial_pw", lambda *args, **kwargs: None)
220206

221207
gds_args1 = sessions.get_or_create(
222208
"my-session",

0 commit comments

Comments
 (0)