Fix gateway integration test

OpenMined · Oct 18, 2023 · bf67a1f · bf67a1f
1 parent 41c1701
commit bf67a1f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/integration/network/gateway_test.py b/tests/integration/network/gateway_test.py
@@ -39,8 +39,12 @@ def test_domain_connect_to_gateway(domain_1_port, gateway_port):
     assert proxy_domain_client.metadata == domain_client.metadata
     assert proxy_domain_client.user_role == ServiceRole.NONE
 
-    domain_client.login(email="[email protected]", password="changethis")
-    proxy_domain_client.login(email="[email protected]", password="changethis")
+    domain_client = domain_client.login(
+        email="[email protected]", password="changethis"
+    )
+    proxy_domain_client = proxy_domain_client.login(
+        email="[email protected]", password="changethis"
+    )
 
     assert proxy_domain_client.logged_in_user == "[email protected]"
     assert proxy_domain_client.user_role == ServiceRole.ADMIN