Fix shenanigans in web UI automatic tests

bunkerity · TheophileDiot · Jun 14, 2024 · Jun 7, 2024 · Jun 8, 2024 · Jun 8, 2024
commit 0aabffb6209b645dd985fa4c77b5242457ca14cb
diff --git a/tests/ui/configs_page.py b/tests/ui/configs_page.py
@@ -66,7 +66,7 @@
 
     assert_alert_message(DRIVER, "Created")
 
-    sleep(30)
+    sleep(5)
 
     DRIVER.execute_script("window.open('http://www.example.com/hello','_blank');")
     DRIVER.switch_to.window(DRIVER.window_handles[1])
@@ -165,7 +165,7 @@
 
     assert_alert_message(DRIVER, "Deleted")
 
-    sleep(30)
+    sleep(5)
 
     resp = get("http://www.example.com/hello", verify=False)
 
@@ -202,7 +202,7 @@
 
     assert_alert_message(DRIVER, "Created")
 
-    sleep(30)
+    sleep(5)
 
     DRIVER.execute_script("window.open('http://app1.example.com/hello','_blank');")
     DRIVER.switch_to.window(DRIVER.window_handles[1])

diff --git a/tests/ui/jobs_page.py b/tests/ui/jobs_page.py
@@ -86,7 +86,13 @@
 
     log_info("Reload filter is working, trying jobs cache ...")
 
-    resp = get(f"http://www.example.com{UI_URL}/jobs/download?job_name=mmdb-country&file_name=country.mmdb", verify=False)
+    cookie = DRIVER.get_cookies()[0]
+    resp = get(
+        f"https://www.example.com{UI_URL}/jobs/download?job_name=mmdb-country&file_name=country.mmdb",
+        headers={"Host": "www.example.com", "User-Agent": DRIVER.execute_script("return navigator.userAgent;")},
+        cookies={cookie["name"]: cookie["value"]},
+        verify=False,
+    )
 
     if resp.status_code != 200:
         log_error("The cache download is not working, exiting ...")

diff --git a/tests/ui/plugins_page.py b/tests/ui/plugins_page.py
@@ -110,7 +110,7 @@
     DEACTIVATED_PLUGINS = ("antibot", "blacklist", "bunkernet", "cors", "country", "greylist", "redis", "reversescan")
     for plugin in DEACTIVATED_PLUGINS:
         log_info(f"Trying {plugin} plugin page ...")
-        DRIVER.get(f"http://www.example.com{UI_URL}/plugins/{plugin}")
+        DRIVER.get(f"https://www.example.com{UI_URL}/plugins/{plugin}")
         first_card = safe_get_element(DRIVER, By.XPATH, "/html/body/main/div/div/div/div[1]/h5")
         assert isinstance(first_card, WebElement), "First card is not a WebElement"
 
@@ -122,7 +122,7 @@
         DRIVER.back()
 
     log_info("Trying bad behavior plugin page ...")
-    DRIVER.get(f"http://www.example.com{UI_URL}/plugins/badbehavior")
+    DRIVER.get(f"https://www.example.com{UI_URL}/plugins/badbehavior")
 
     sleep(5)
 
@@ -143,7 +143,7 @@
     DRIVER.back()
 
     log_info("Trying dnsbl plugin page ...")
-    DRIVER.get(f"http://www.example.com{UI_URL}/plugins/dnsbl")
+    DRIVER.get(f"https://www.example.com{UI_URL}/plugins/dnsbl")
 
     sleep(5)
 
@@ -158,7 +158,7 @@
     DRIVER.back()
 
     log_info("Trying errors plugin page ...")
-    DRIVER.get(f"http://www.example.com{UI_URL}/plugins/errors")
+    DRIVER.get(f"https://www.example.com{UI_URL}/plugins/errors")
 
     sleep(5)
 
@@ -179,7 +179,7 @@
     DRIVER.back()
 
     log_info("Trying limit plugin page ...")
-    DRIVER.get(f"http://www.example.com{UI_URL}/plugins/limit")
+    DRIVER.get(f"https://www.example.com{UI_URL}/plugins/limit")
 
     limit_info_elem = safe_get_element(DRIVER, By.XPATH, "/html/body/main/div/div/div[1]/h5")
     assert isinstance(limit_info_elem, WebElement), "Limit info element is not a WebElement"
@@ -192,7 +192,7 @@
     DRIVER.back()
 
     log_info("Trying miscellaneous plugin page ...")
-    DRIVER.get(f"http://www.example.com{UI_URL}/plugins/misc")
+    DRIVER.get(f"https://www.example.com{UI_URL}/plugins/misc")
 
     sleep(5)
 
@@ -209,7 +209,7 @@
     DRIVER.back()
 
     log_info("Trying whitelist plugin page ...")
-    DRIVER.get(f"http://www.example.com{UI_URL}/plugins/whitelist")
+    DRIVER.get(f"https://www.example.com{UI_URL}/plugins/whitelist")
 
     sleep(5)