librewolf/bunkerized-nginx
4
0
Fork 1
mirror of https://github.com/bunkerity/bunkerized-nginx synced 2023-12-13 21:30:18 +01:00
Code Issues Projects Releases Wiki Activity

tests - retry UI access in case of network exception

Browse source
This commit is contained in:
florian 2023-10-29 13:05:43 +01:00
parent 86875f4863
commit 0e3d8e59cc
No known key found for this signature in database
GPG key ID: 93EE47CC3D061500
1 changed files with 11 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/ui/main.py
View file

@ -73,7 +73,7 @@ def assert_button_click(driver, button: Union[str, WebElement]):
button.click()
clicked = True
return clicked
def assert_alert_message(driver, message: str):
safe_get_element(driver, By.XPATH, "//button[@data-flash-sidebar-open='']")
@ -126,20 +126,25 @@ def access_page(
message: bool = True,
*,
retries: int = 0,
clicked: bool = False,
):
if retries == 0:
assert_button_click(driver, button)
if retries > 5:
print("Too many retries...", flush=True)
exit(1)
try:
if not clicked:
clicked = assert_button_click(driver, button)
title = driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/div/header/div/nav/h6")))
if title.text != name.replace(" ", "_").title():
print(f"Didn't get redirected to {name} page, exiting ...", flush=True)
exit(1)
except TimeoutException:
if retries < 3 and "/loading" in driver.current_url:
if "/loading" in driver.current_url:
sleep(2)
access_page(driver, driver_wait, button, name, message, retries=retries + 1)
return access_page(driver, driver_wait, button, name, message, retries=retries + 1, clicked=clicked)
print(f"{name.title()} page didn't load in time, exiting ...", flush=True)
exit(1)
@ -148,7 +153,7 @@ def access_page(
print("Connection failure, retrying in 5s ...", flush=True)
driver.refresh()
sleep(5)
return access_page(driver, driver_wait, button, name, message, retries=1)
return access_page(driver, driver_wait, button, name, message, retries=retries + 1, clicked=clicked)
raise we
if message: