test_ui: Use a timeout instead of sleeping for 10 seconds

india-ultimate · Aug 10, 2023 · a9fd677 · a9fd677
1 parent 238b50c
commit a9fd677
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/server/tests/test_ui.py b/server/tests/test_ui.py
@@ -1,6 +1,5 @@
 import datetime
 import random
-import time
 
 import pytest
 from django.utils.timezone import now
@@ -81,9 +80,8 @@ def test_new_user_login(self) -> None:
             self.type("input#vpa-upi", "punchagan@upi")
             self.click("button#redesign-v15-cta")
             self.switch_to_default_content()
-            time.sleep(10)
 
-            self.click('a[href="#/vaccination/1"]')
+            self.click('a[href="#/vaccination/1"]', timeout=30)
             self.select_option_by_text("select#name", "Covishield")
             self.choose_file("input#certificate", "frontend/assets/favico.png")
             self.click('button:contains("Submit")')