fix tests

Author: pcrespov

services/web/server/tests/unit/isolated/test_utils_rate_limiting.py

@@ -81,9 +81,10 @@ async def test_global_rate_limit_route(requests_per_second: float, client: TestC
 
     msg = []
     for i, task in enumerate(tasks):
-        event = asyncio.Event()
-        task.add_done_callback(lambda _: event.set())
-        await event.wait()
+
+        while not task.done():
+            await asyncio.sleep(0.01)
+
         assert not task.cancelled()
         assert not task.exception()
         msg.append(

services/web/server/tests/unit/with_dbs/03/login/test_login_2fa.py

@@ -301,7 +301,7 @@ def _get_confirmation_link_from_email():
         },
     )
     data, _ = await assert_status(response, status.HTTP_200_OK)
-    assert data["message"] == "You are logged in"
+    assert "logged in" in data["message"]
 
 
 async def test_can_register_same_phone_in_different_accounts(
@@ -358,7 +358,7 @@ async def test_can_register_same_phone_in_different_accounts(
         )
         data, error = await assert_status(response, status.HTTP_202_ACCEPTED)
         assert data
-        assert "Code" in data["message"]
+        assert "SMS" in data["message"]
         assert data["name"] == CODE_2FA_SMS_CODE_REQUIRED
         assert not error