@@ -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