bunkerweb/tests/ui/main.py

from contextlib import suppress
from datetime import datetime, timedelta
from functools import partial
from os import getenv, listdir, sep
from os.path import join
from pathlib import Path
from random import randint
from time import sleep
from traceback import format_exc
from typing import List, Union
from pyotp import TOTP
from requests import get
from requests.exceptions import RequestException
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.firefox.options import Options
from selenium.webdriver.firefox.service import Service
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.remote.webelement import WebElement
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import ElementClickInterceptedException, TimeoutException, WebDriverException

default_server = "127.0.0.1"

os_release_path = Path(sep, "etc", "os-release")
if os_release_path.is_file() and "Alpine" in os_release_path.read_text(encoding="utf-8"):
    default_server = "192.168.0.2"

ready = False
retries = 0
while not ready:
    with suppress(RequestException):
        status_code = get(f"http://{default_server}/setup").status_code

        if status_code > 500 and status_code != 502:
            print("An error occurred with the server, exiting ...", flush=True)
            exit(1)

        ready = status_code < 400

    if retries > 20:
        print("UI took too long to be ready, exiting ...", flush=True)
        exit(1)
    elif not ready:
        retries += 1
        print("Waiting for UI to be ready, retrying in 5s ...", flush=True)
        sleep(5)

print("UI is ready, starting tests ...", flush=True)

TEST_TYPE = getenv("TEST_TYPE", "docker")

firefox_options = Options()
if "geckodriver" not in listdir(Path.cwd()):
    firefox_options.add_argument("--headless")
firefox_options.log.level = "trace"

print("Starting Firefox ...", flush=True)


def safe_get_element(driver, by: By, _id: str, *, multiple: bool = False, error: bool = False) -> Union[WebElement, List[WebElement]]:
    try:
        return WebDriverWait(driver, 4).until(EC.presence_of_element_located((by, _id)) if not multiple else EC.presence_of_all_elements_located((by, _id)))
    except TimeoutException as e:
        if error:
            raise e

        print(f'Element searched by {by}: "{_id}" not found, exiting ...', flush=True)
        exit(1)


def assert_button_click(driver, button: Union[str, WebElement]):
    clicked = False
    while not clicked:
        with suppress(ElementClickInterceptedException):
            if isinstance(button, str):
                button = safe_get_element(driver, By.XPATH, button)

            sleep(0.5)

            button.click()
            clicked = True
    return clicked


def assert_alert_message(driver, message: str):
    safe_get_element(driver, By.XPATH, "//button[@data-flash-sidebar-open='']")

    sleep(0.3)

    assert_button_click(driver, "//button[@data-flash-sidebar-open='']")

    error = False
    while True:
        try:
            alerts = safe_get_element(
                driver,
                By.XPATH,
                "//aside[@data-flash-sidebar='']/div[2]/div",
                multiple=True,
                error=True,
            )
            break
        except TimeoutException:
            if error:
                print("Messages list not found, exiting ...", flush=True)
                exit(1)
            error = True
            driver.refresh()

    is_in = False
    for alert in alerts:
        if message in alert.text:
            is_in = True
            break

    if not is_in:
        print(
            f'Message "{message}" not found in one of the messages in the list, exiting ...',
            flush=True,
        )
        exit(1)

    print(f'Message "{message}" found in one of the messages in the list', flush=True)

    assert_button_click(driver, "//button[@data-flash-sidebar-close='']/*[local-name() = 'svg']")


def access_page(
    driver,
    driver_wait: WebDriverWait,
    button: Union[str, WebElement],
    name: str,
    message: bool = True,
    *,
    retries: int = 0,
    clicked: bool = False,
):
    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.title():
            print(f"Didn't get redirected to {name} page, exiting ...", flush=True)
            exit(1)
    except TimeoutException:
        if "/loading" in driver.current_url:
            sleep(2)
            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)
    except WebDriverException as we:
        if "connectionFailure" in str(we):
            print("Connection failure, retrying in 5s ...", flush=True)
            driver.refresh()
            sleep(5)
            return access_page(driver, driver_wait, button, name, message, retries=retries + 1, clicked=clicked)
        raise we

    if message:
        print(
            f"{name.title()} page loaded successfully",
            flush=True,
        )


driver_func = partial(webdriver.Firefox, service=Service(log_output="./geckodriver.log"), options=firefox_options)
if TEST_TYPE == "dev":
    driver_func = partial(
        webdriver.Firefox,
        service=Service(executable_path="./geckodriver" if "geckodriver" in listdir(Path.cwd()) else "/usr/local/bin/geckodriver", log_output="./geckodriver.log"),
        options=firefox_options,
    )

with driver_func() as driver:
    try:
        driver.delete_all_cookies()
        driver.maximize_window()
        driver_wait = WebDriverWait(driver, 60)

        print(f"Navigating to http://{default_server}/setup ...", flush=True)

        driver.get(f"http://{default_server}/setup")

        ### WIZARD PAGE

        try:
            title = driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/main/div/div/h1")))

            if title.text != "Setup Wizard":
                print("Didn't get redirected to setup page, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("Didn't get redirected to setup page, exiting ...", flush=True)
            exit(1)

        print("Setup page loaded successfully, filling the form ...", flush=True)

        admin_username_input = safe_get_element(driver, By.ID, "admin_username")
        password_input = safe_get_element(driver, By.ID, "admin_password")
        password_check_input = safe_get_element(driver, By.ID, "admin_password_check")
        ui_url = safe_get_element(driver, By.ID, "ui_url").get_attribute("value")

        admin_username_input.send_keys("admin")
        password_input.send_keys("S$cr3tP@ssw0rd")
        password_check_input.send_keys("S$cr3tP@ssw0rd")

        assert_button_click(driver, "//button[@id='setup-button']")

        print("Submitted the form, waiting for the wizard to finish ...", flush=True)

        current_time = datetime.now()

        while current_time + timedelta(minutes=5) > datetime.now() and not driver.current_url.endswith("/login"):
            sleep(1)

        ### LOGIN PAGE

        if not driver.current_url.endswith("/login"):
            print("Didn't get redirected to login page, exiting ...", flush=True)
            exit(1)

        print("Redirected to login page, waiting for login form ...", flush=True)

        safe_get_element(driver, By.TAG_NAME, "form")

        print(
            "Form found, trying to access another page without being logged in ...",
            flush=True,
        )

        driver.get(f"http://www.example.com{ui_url}/home")

        print("Waiting for toast ...", flush=True)

        toast = safe_get_element(driver, By.XPATH, "//div[@data-flash-message='']")

        print("Toast found", flush=True)

        if "Please log in to access this page." not in toast.text:
            print("Toast doesn't contain the expected message, exiting ...", flush=True)
            exit(1)

        print(
            "Toast contains the expected message, filling login form with wrong credentials ...",
            flush=True,
        )

        sleep(1)

        safe_get_element(driver, By.TAG_NAME, "form")

        username_input = safe_get_element(driver, By.ID, "username")
        password_input = safe_get_element(driver, By.ID, "password")
        username_input.send_keys("hackerman")
        password_input.send_keys("password")
        password_input.send_keys(Keys.RETURN)

        sleep(0.3)

        try:
            title = driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/main/div[1]/div/h1")))

            if title.text != "Log in":
                print("Didn't get redirected to login page, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("Login page didn't load in time, exiting ...", flush=True)
            exit(1)

        print(
            "Got redirected to login page successfully, filling login form with good credentials ...",
            flush=True,
        )

        username_input = safe_get_element(driver, By.ID, "username")
        password_input = safe_get_element(driver, By.ID, "password")
        username_input.send_keys("admin")
        password_input.send_keys("S$cr3tP@ssw0rd")

        access_page(
            driver,
            driver_wait,
            "//button[@value='login']",
            "home",
        )

        ### HOME PAGE

        print("Trying instances page ...", flush=True)

        access_page(
            driver,
            driver_wait,
            "/html/body/aside[1]/div[1]/div[3]/ul/li[2]/a",
            "instances",
        )

        ### INSTANCES PAGE

        no_errors = True
        retries = 0
        action = "reload" if TEST_TYPE == "docker" else "restart"
        while no_errors:
            print(f"Trying to {action} BunkerWeb instance ...", flush=True)

            try:
                form = WebDriverWait(driver, 2).until(EC.presence_of_element_located((By.XPATH, "//form[starts-with(@id, 'form-instance-')]")))
            except TimeoutException:
                print("No instance form found, exiting ...", flush=True)
                exit(1)

            try:
                access_page(
                    driver,
                    driver_wait,
                    f"//form[starts-with(@id, 'form-instance-')]//button[@value='{action}']",
                    "instances",
                    False,
                )

                print(
                    f"Instance was {action}ed successfully, checking the message ...",
                    flush=True,
                )

                assert_alert_message(driver, f"has been {action}ed")

                no_errors = False
            except:
                if retries >= 3:
                    exit(1)
                retries += 1

                print("WARNING: message list doesn't contain the expected message or is empty, retrying...")

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        print("Trying global config page ...")

        access_page(
            driver,
            driver_wait,
            "/html/body/aside[1]/div[1]/div[3]/ul/li[3]/a",
            "global config",
        )

        ### GLOBAL CONFIG PAGE

        no_errors = True
        retries = 0
        while no_errors:
            try:
                print(
                    "Trying to save the global config without changing anything ...",
                    flush=True,
                )

                safe_get_element(driver, By.ID, "form-edit-global-configs")

                access_page(
                    driver,
                    driver_wait,
                    "//form[@id='form-edit-global-configs']//button[@type='submit']",
                    "global config",
                    False,
                )

                print(
                    "The page reloaded successfully, checking the message ...",
                    flush=True,
                )

                assert_alert_message(
                    driver,
                    "The global configuration was not edited because no values were changed.",
                )

                no_errors = False
            except:
                if retries >= 3:
                    exit(1)
                retries += 1

                print("WARNING: message list doesn't contain the expected message or is empty, retrying...")

        print(
            'Checking if the "DATASTORE_MEMORY_SIZE" input have the overridden value ...',
            flush=True,
        )

        input_datastore = safe_get_element(driver, By.ID, "DATASTORE_MEMORY_SIZE")

        if not input_datastore.get_attribute("disabled"):
            print(
                'The input "DATASTORE_MEMORY_SIZE" is not disabled, even though it should be, exiting ...',
                flush=True,
            )
            exit(1)
        elif input_datastore.get_attribute("value") != "384m":
            print("The value is not the expected one, exiting ...", flush=True)
            exit(1)

        print(
            "The value is the expected one and the input is disabled, trying to edit the global config with wrong values ...",
            flush=True,
        )

        input_worker = safe_get_element(driver, By.ID, "WORKER_RLIMIT_NOFILE")

        input_worker.clear()
        input_worker.send_keys("ZZZ")

        assert_button_click(driver, "//form[@id='form-edit-global-configs']//button[@type='submit']")

        assert_alert_message(
            driver,
            "The global configuration was not edited because no values were changed.",
        )

        print(
            "The form was not submitted, trying to edit the global config with good values ...",
            flush=True,
        )

        input_worker.clear()
        input_worker.send_keys("4096")

        access_page(
            driver,
            driver_wait,
            "//form[@id='form-edit-global-configs']//button[@type='submit']",
            "global config",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        input_worker = safe_get_element(driver, By.ID, "WORKER_RLIMIT_NOFILE")

        if input_worker.get_attribute("value") != "4096":
            print("The value was not updated, exiting ...", flush=True)
            exit(1)

        print(
            "The value was updated successfully, trying to navigate through the global config tabs ...",
            flush=True,
        )

        buttons = safe_get_element(
            driver,
            By.XPATH,
            "//div[@data-global-config-tabs-desktop='']/button",
            multiple=True,
        )
        buttons.reverse()
        for button in buttons:
            assert_button_click(driver, button)

        print("Trying to filter the global config ...", flush=True)

        safe_get_element(driver, By.ID, "settings-filter").send_keys("Datastore")

        if (
            len(
                safe_get_element(
                    driver,
                    By.XPATH,
                    "//form[@id='form-edit-global-configs']//div[@data-setting-container='' and not(contains(@class, 'hidden'))]",
                    multiple=True,
                )
            )
            != 1
        ):
            print("The filter didn't work, exiting ...", flush=True)
            exit(1)

        print("Trying services page ...")

        access_page(
            driver,
            driver_wait,
            "/html/body/aside[1]/div[1]/div[3]/ul/li[4]/a",
            "services",
        )

        ## SERVICES PAGE

        print("Checking the services page ...", flush=True)

        if safe_get_element(driver, By.XPATH, "//div[@data-services-service='www.example.com']//h5").text.strip() != "www.example.com":
            print("The service is not present, exiting ...", flush=True)
            exit(1)

        if safe_get_element(driver, By.XPATH, "//div[@data-services-service='www.example.com']//h6").text.strip() != "ui":
            print(
                "The service should have been created by the ui, exiting ...",
                flush=True,
            )
            exit(1)

        print("Service www.example.com is present, trying to edit it ...", flush=True)

        assert_button_click(
            driver,
            "//div[@data-services-service='www.example.com']//button[@data-services-action='edit']",
        )

        try:
            modal = safe_get_element(driver, By.XPATH, "//div[@data-services-modal='']", error=True)
        except TimeoutException:
            print("Modal not found, exiting ...", flush=True)
            exit(1)

        if "hidden" in modal.get_attribute("class"):
            print("Modal is hidden even though it shouldn't be, exiting ...", flush=True)
            exit(1)

        input_server_name = safe_get_element(driver, By.ID, "SERVER_NAME")

        if input_server_name.get_attribute("value") != "www.example.com":
            print("The value is not the expected one, exiting ...", flush=True)
            exit(1)

        print(
            'The value for the "SERVER_NAME" input is the expected one, trying to edit the config ...',
            flush=True,
        )

        assert_button_click(
            driver,
            safe_get_element(driver, By.XPATH, "//button[@data-tab-handler='gzip']"),
        )

        gzip_select = safe_get_element(driver, By.XPATH, "//button[@data-setting-select='gzip-comp-level']")

        assert_button_click(driver, gzip_select)

        assert_button_click(
            driver,
            safe_get_element(
                driver,
                By.XPATH,
                "//button[@data-setting-select-dropdown-btn='gzip-comp-level' and @value='6']",
            ),
        )

        access_page(
            driver,
            driver_wait,
            "//button[@data-services-modal-submit='']",
            "services",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        print(
            "The page reloaded successfully, checking if the setting has been updated ...",
            flush=True,
        )

        assert_button_click(
            driver,
            "//div[@data-services-service='www.example.com']//button[@data-services-action='edit']",
        )

        modal = safe_get_element(driver, By.XPATH, "//div[@data-services-modal='']")

        if "hidden" in modal.get_attribute("class"):
            print("Modal is hidden even though it shouldn't be, exiting ...", flush=True)
            exit(1)

        assert_button_click(
            driver,
            driver.find_element(By.XPATH, "//button[@data-tab-handler='gzip']"),
        )

        gzip_true_select = safe_get_element(driver, By.ID, "GZIP_COMP_LEVEL")

        if safe_get_element(driver, By.XPATH, "//select[@id='GZIP_COMP_LEVEL']/option[@selected='']").get_attribute("value") != "6":
            print("The value is not the expected one, exiting ...", flush=True)
            exit(1)

        assert_button_click(driver, "//button[@data-services-modal-close='']/*[local-name() = 'svg']")

        print("Creating a new service ...", flush=True)

        assert_button_click(driver, "//button[@data-services-action='new']")

        server_name_input: WebElement = safe_get_element(driver, By.ID, "SERVER_NAME")  # type: ignore
        server_name_input.clear()
        server_name_input.send_keys("app1.example.com")

        if TEST_TYPE == "docker":
            assert_button_click(driver, "//button[@data-tab-handler='reverseproxy']")

            assert_button_click(driver, safe_get_element(driver, By.ID, "USE_REVERSE_PROXY"))

            assert_button_click(driver, "//button[@data-services-multiple-add='reverse-proxy']")

            safe_get_element(driver, By.ID, "REVERSE_PROXY_HOST").send_keys("http://app1:8080")
            safe_get_element(driver, By.ID, "REVERSE_PROXY_URL").send_keys("/")

        access_page(
            driver,
            driver_wait,
            "//button[@data-services-modal-submit='']",
            "services",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://app1.example.com/ready", headers={"Host": "app1.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        try:
            services = safe_get_element(driver, By.XPATH, "//div[@data-services-service]", multiple=True, error=True)
        except TimeoutException:
            print("Services not found, exiting ...", flush=True)
            exit(1)

        if len(services) < 2:
            print("The service hasn't been created, exiting ...", flush=True)
            exit(1)

        if safe_get_element(driver, By.XPATH, "//div[@data-services-service='app1.example.com']//h5").text.strip() != "app1.example.com":
            print('The service "app1.example.com" is not present, exiting ...', flush=True)
            exit(1)

        if safe_get_element(driver, By.XPATH, "//div[@data-services-service='app1.example.com']//h6").text.strip() != "ui":
            print(
                "The service should have been created by the ui, exiting ...",
                flush=True,
            )
            exit(1)

        print("Service app1.example.com is present, trying it ...", flush=True)

        try:
            safe_get_element(
                driver,
                By.XPATH,
                "//button[@data-services-action='edit' and @data-services-name='app1.example.com']//ancestor::div//a",
                error=True,
            )
        except TimeoutException:
            print(
                "Delete button hasn't been found, even though it should be, exiting ...",
                flush=True,
            )
            exit(1)

        ready = False
        retries = 0
        while not ready:
            with suppress(RequestException):
                status_code = get("http://app1.example.com/").status_code

                if status_code > 500:
                    print("The service is not working, exiting ...", flush=True)
                    exit(1)

                ready = status_code < 400

            if retries > 20:
                print("The service took too long to be ready, exiting ...", flush=True)
                exit(1)
            elif not ready:
                retries += 1
                print(
                    "Waiting for the service to be ready, retrying in 5s ...",
                    flush=True,
                )
                sleep(5)

        print("The service is working, trying to clone it ...", flush=True)

        try:
            clone_button = safe_get_element(
                driver,
                By.XPATH,
                "//button[@data-services-action='clone' and @data-services-name='app1.example.com']",
                error=True,
            )
        except TimeoutException:
            print(
                "Clone button hasn't been found, even though it should be, exiting ...",
                flush=True,
            )
            exit(1)

        assert_button_click(driver, clone_button)

        server_name_input: WebElement = safe_get_element(driver, By.ID, "SERVER_NAME")  # type: ignore

        if server_name_input.get_attribute("value"):
            print("The cloned service input is not empty, exiting ...", flush=True)
            exit(1)

        server_name_input.clear()
        server_name_input.send_keys("app2.example.com")

        access_page(
            driver,
            driver_wait,
            "//button[@data-services-modal-submit='']",
            "services",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://app2.example.com/ready", headers={"Host": "app2.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        try:
            services = safe_get_element(driver, By.XPATH, "//div[@data-services-service]", multiple=True, error=True)
        except TimeoutException:
            print("Services not found, exiting ...", flush=True)
            exit(1)

        if len(services) < 3:
            print("The service hasn't been created, exiting ...", flush=True)
            exit(1)

        if safe_get_element(driver, By.XPATH, "//div[@data-services-service='app2.example.com']//h5").text.strip() != "app2.example.com":
            print('The service "app2.example.com" is not present, exiting ...', flush=True)
            exit(1)

        if safe_get_element(driver, By.XPATH, "//div[@data-services-service='app2.example.com']//h6").text.strip() != "ui":
            print(
                "The service should have been created by the ui, exiting ...",
                flush=True,
            )
            exit(1)

        print("Service app2.example.com is present, trying it ...", flush=True)

        try:
            safe_get_element(
                driver,
                By.XPATH,
                "//button[@data-services-action='edit' and @data-services-name='app2.example.com']//ancestor::div//a",
                error=True,
            )
        except TimeoutException:
            print(
                "Delete button hasn't been found, even though it should be, exiting ...",
                flush=True,
            )
            exit(1)

        ready = False
        retries = 0
        while not ready:
            with suppress(RequestException):
                status_code = get("http://app2.example.com/").status_code

                if status_code > 500:
                    print("The service is not working, exiting ...", flush=True)
                    exit(1)

                ready = status_code < 400

            if retries > 20:
                print("The service took too long to be ready, exiting ...", flush=True)
                exit(1)
            elif not ready:
                retries += 1
                print(
                    "Waiting for the service to be ready, retrying in 5s ...",
                    flush=True,
                )
                sleep(5)

        print("The service is working, trying to delete it ...", flush=True)

        try:
            delete_button = safe_get_element(
                driver,
                By.XPATH,
                "//button[@data-services-action='delete' and @data-services-name='app2.example.com']",
                error=True,
            )
        except TimeoutException:
            print(
                "Delete button hasn't been found, even though it should be, exiting ...",
                flush=True,
            )
            exit(1)

        print(
            "Delete button is present, as expected, deleting the service ...",
            flush=True,
        )

        assert_button_click(driver, delete_button)

        access_page(
            driver,
            driver_wait,
            "//form[@data-services-modal-form-delete='']//button[@type='submit']",
            "services",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        assert_alert_message(driver, "has been deleted.")

        print(
            "Service app2.example.com has been deleted, checking if it's still present ...",
            flush=True,
        )

        try:
            services = safe_get_element(driver, By.XPATH, "//div[@data-services-service='']", multiple=True, error=True)
        except TimeoutException:
            print("Services not found, exiting ...", flush=True)
            exit(1)

        if len(services) > 2:
            print("The service hasn't been deleted, exiting ...", flush=True)
            exit(1)

        print(
            "The service has been deleted, successfully, trying configs page ...",
            flush=True,
        )

        access_page(
            driver,
            driver_wait,
            "/html/body/aside[1]/div[1]/div[3]/ul/li[5]/a",
            "configs",
        )

        ### CONFIGS PAGE

        print("Trying to create a new config ...", flush=True)

        assert_button_click(
            driver,
            "//div[@data-configs-element='server-http' and @data-_type='folder']",
        )
        assert_button_click(driver, "//li[@data-configs-add-file='']/button")

        safe_get_element(driver, By.XPATH, "//div[@data-configs-modal-path='']/input").send_keys("hello")
        safe_get_element(driver, By.XPATH, "//div[@data-configs-modal-editor='']/textarea").send_keys(
            """
location /hello {
    default_type 'text/plain';
    content_by_lua_block {
        ngx.say('hello app1')
    }
}
            """
        )

        access_page(
            driver,
            driver_wait,
            "//button[@data-configs-modal-submit='']",
            "configs",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        assert_alert_message(driver, "was successfully created")

        sleep(30)

        driver.execute_script("window.open('http://www.example.com/hello','_blank');")
        driver.switch_to.window(driver.window_handles[1])
        driver.switch_to.default_content()

        try:
            if safe_get_element(driver, By.XPATH, "//pre", error=True).text.strip() != "hello app1":
                print("The config hasn't been created correctly, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("The config hasn't been created, exiting ...", flush=True)
            exit(1)

        print(
            "The config has been created and is working, trying to delete it ...",
            flush=True,
        )

        driver.close()
        driver.switch_to.window(driver.window_handles[0])

        assert_button_click(
            driver,
            "//div[@data-configs-element='server-http' and @data-_type='folder']",
        )
        assert_button_click(driver, "//div[@data-configs-action-button='hello.conf']")
        assert_button_click(
            driver,
            "//div[@data-configs-action-dropdown='hello.conf']/button[@value='delete' and @data-configs-action-dropdown-btn='hello.conf']",
        )

        access_page(
            driver,
            driver_wait,
            "//button[@data-configs-modal-submit='']",
            "configs",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        assert_alert_message(driver, "was successfully deleted")

        print("The config has been deleted, trying plugins page ...", flush=True)

        access_page(
            driver,
            driver_wait,
            "/html/body/aside[1]/div[1]/div[3]/ul/li[6]/a",
            "plugins",
        )

        ### PLUGINS PAGE

        print("Trying to reload the plugins without adding any ...", flush=True)

        reload_button = safe_get_element(driver, By.XPATH, "//div[@data-plugins-upload='']//button[@type='submit']")

        if reload_button.get_attribute("disabled") is None:
            print("The reload button is not disabled, exiting ...", flush=True)
            exit(1)

        print("Trying to filter the plugins ...", flush=True)

        safe_get_element(driver, By.XPATH, "//input[@placeholder='key words']").send_keys("Anti")

        plugins = safe_get_element(driver, By.XPATH, "//div[@data-plugins-list='']", multiple=True)

        if len(plugins) != 1:
            print("The filter is not working, exiting ...", flush=True)
            exit(1)

        print("The filter is working, trying to add a bad plugin ...", flush=True)

        safe_get_element(driver, By.XPATH, "//input[@type='file' and @name='file']").send_keys(join(Path.cwd(), "test.zip"))

        sleep(2)

        access_page(
            driver,
            driver_wait,
            "//div[@data-plugins-upload='']//button[@type='submit']",
            "plugins",
            False,
        )

        print(
            "The bad plugin has been rejected, trying to add a good plugin ...",
            flush=True,
        )

        safe_get_element(driver, By.XPATH, "//input[@type='file' and @name='file']").send_keys(join(Path.cwd(), "discord.zip"))

        sleep(2)

        access_page(
            driver,
            driver_wait,
            "//div[@data-plugins-upload='']//button[@type='submit']",
            "plugins",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        external_plugins = safe_get_element(
            driver,
            By.XPATH,
            "//div[@data-plugins-external=' external ']",
            multiple=True,
        )

        if len(external_plugins) != 1:
            print("The plugin hasn't been added, exiting ...", flush=True)
            exit(1)

        print("The plugin has been added, trying delete it ...", flush=True)

        assert_button_click(
            driver,
            "//button[@data-plugins-action='delete' and @name='discord']",
        )

        access_page(
            driver,
            driver_wait,
            "//form[@data-plugins-modal-form-delete='']//button[@type='submit']",
            "plugins",
            False,
        )

        if TEST_TYPE == "linux":
            ready = False
            retries = 0
            while not ready:
                with suppress(RequestException):
                    resp = get("http://www.example.com/ready", headers={"Host": "www.example.com"}, verify=False)
                    status_code = resp.status_code
                    text = resp.text

                    if resp.status_code >= 500:
                        print("❌ An error occurred with the server, exiting ...", flush=True)
                        exit(1)

                    ready = status_code < 400 and text == "ready"

                if retries > 10:
                    print("❌ BunkerWeb took too long to be ready, exiting ...", flush=True)
                    exit(1)
                elif not ready:
                    retries += 1
                    print("⚠️ Waiting for BunkerWeb to be ready, retrying in 5s ...", flush=True)
                    sleep(5)

        with suppress(TimeoutException):
            title = WebDriverWait(driver, 2).until(
                EC.presence_of_element_located(
                    (
                        By.XPATH,
                        "//button[@data-plugins-action='delete' and @name='discord']",
                    )
                )
            )

            if title:
                print("The plugin hasn't been deleted, exiting ...", flush=True)
                exit(1)

        print("The plugin has been deleted, trying cache page ...", flush=True)

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[3]/ul/li[7]/a", "cache")

        ### CACHE PAGE

        print("Trying to open a cache file ...", flush=True)

        assert_button_click(driver, "//div[@data-cache-element='mmdb-asn/asn.mmdb']")

        if (
            safe_get_element(
                driver,
                By.XPATH,
                "//div[@data-cache-modal-editor='']/div[@class='ace_scroller']//div[@class='ace_line']",
            ).text.strip()
            != "Download file to view content"
        ):
            print("The cache file content is not correct, exiting ...", flush=True)
            exit(1)

        assert_button_click(driver, "//button[@data-cache-modal-submit='']")

        sleep(3)

        print("The cache file content is correct, trying reporting page ...", flush=True)

        get(f"http://www.example.com{ui_url}/home?id=/etc/passwd")

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[3]/ul/li[8]/a", "reports")

        ### REPORTS PAGE

        print("Trying to filter the reports ...", flush=True)

        reports_list = safe_get_element(driver, By.XPATH, "//ul[@data-reports-list='']/li", multiple=True)

        if not reports_list:
            print("No reports found, exiting ...", flush=True)
            exit(1)

        filter_input = safe_get_element(driver, By.ID, "keyword")
        filter_input.send_keys("abcde")

        with suppress(TimeoutException):
            WebDriverWait(driver, 2).until(
                EC.presence_of_element_located(
                    (
                        By.XPATH,
                        "//ul[@data-reports-list='']/li[not(contains(@class, 'hidden'))]",
                    )
                )
            )
            print("The keyword filter is not working, exiting ...", flush=True)
            exit(1)

        print("The reports have been filtered, trying bans page ...", flush=True)

        ### BANS PAGE

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[3]/ul/li[9]/a", "bans")

        try:
            safe_get_element(driver, By.XPATH, "/html/body/main/div/div[2]/div/h5", error=True)
        except TimeoutException:
            print("Bans present even though they shouldn't be, exiting ...", flush=True)
            exit(1)

        print("No bans found, as expected, trying to add a ban ...", flush=True)

        assert_button_click(driver, "//button[@data-add-ban='']")

        try:
            WebDriverWait(driver, 2).until(
                EC.presence_of_element_located(
                    (
                        By.XPATH,
                        "//ul[@data-bans-add-ban-list='']/li",
                    )
                )
            )
        except TimeoutException:
            print("No bans found, exiting ...", flush=True)
            exit(1)

        assert_button_click(driver, "//button[@data-add-ban-delete-all-item='']")

        with suppress(TimeoutException):
            WebDriverWait(driver, 2).until(
                EC.presence_of_element_located(
                    (
                        By.XPATH,
                        "//ul[@data-bans-add-ban-list='']/li",
                    )
                )
            )
            print("Bans present even though they shouldn't be, exiting ...", flush=True)
            exit(1)

        print("No bans found, as expected, trying to add multiple bans ...", flush=True)

        add_entry_button = safe_get_element(driver, By.XPATH, "//button[@data-ban-add-new='']")

        assert_button_click(driver, add_entry_button)

        ip_input = safe_get_element(driver, By.ID, "ip-1")
        ip_input.send_keys(f"127.0.0.{randint(10, 122)}")

        assert_button_click(driver, add_entry_button)

        ip_input = safe_get_element(driver, By.ID, "ip-2")
        ip_input.send_keys(f"127.0.0.{randint(123, 255)}")

        access_page(driver, driver_wait, "//button[@data-bans-modal-submit='']", "bans", False)

        try:
            entries = safe_get_element(driver, By.XPATH, "//ul[@data-bans-list='']/li", multiple=True, error=True)
        except TimeoutException:
            print("No ban found, exiting ...", flush=True)
            exit(1)

        if len(entries) != 2:
            print("The bans are present but there should be 2, exiting ...", flush=True)
            exit(1)

        print("Bans found, trying to delete them ...", flush=True)

        assert_button_click(driver, "//input[@id='ban-item-2']")

        access_page(driver, driver_wait, "//button[@data-unban-btn='']", "bans", False)

        try:
            entries = safe_get_element(driver, By.XPATH, "//ul[@data-bans-list='']/li", multiple=True, error=True)
        except TimeoutException:
            print("No bans found, exiting ...", flush=True)
            exit(1)

        if len(entries) != 1:
            print("The bans are present but there should be 1, exiting ...", flush=True)
            exit(1)

        print("Ban deleted successfully, trying jobs page ...", flush=True)

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[3]/ul/li[10]/a", "jobs")

        ### JOBS PAGE

        print("Trying to filter jobs ...", flush=True)

        jobs_list = safe_get_element(driver, By.XPATH, "//ul[@data-jobs-list='']/li", multiple=True)

        if len(jobs_list) == 0:
            print("No jobs found, exiting ...", flush=True)
            exit(1)

        filter_input = safe_get_element(driver, By.ID, "keyword")

        filter_input.send_keys("abcde")

        with suppress(TimeoutException):
            if len(jobs_list) == len(
                safe_get_element(
                    driver,
                    By.XPATH,
                    "//ul[@data-jobs-list='']/li[not(contains(@class, 'hidden'))]",
                    multiple=True,
                    error=True,
                )
            ):
                print("The keyword filter is not working, exiting ...", flush=True)
                exit(1)

        filter_input.clear()

        print(
            "Keyword filter is working, trying to filter by success state ...",
            flush=True,
        )

        sleep(0.3)

        assert_button_click(driver, "//button[@data-jobs-setting-select='success']")

        assert_button_click(
            driver,
            "//div[@data-jobs-setting-select-dropdown='success']/button[@value='false']",
        )

        with suppress(TimeoutException):
            if (
                len(jobs_list)
                == len(
                    safe_get_element(
                        driver,
                        By.XPATH,
                        "//ul[@data-jobs-list='']/li[not(contains(@class, 'hidden'))]",
                        multiple=True,
                        error=True,
                    )
                )
            ) and len(jobs_list) != len(
                safe_get_element(
                    driver,
                    By.XPATH,
                    "//ul[@data-jobs-list='']//p[@data-jobs-success='false']",
                    multiple=True,
                    error=True,
                )
            ):
                print("The success filter is not working, exiting ...", flush=True)
                exit(1)

        assert_button_click(driver, "//button[@data-jobs-setting-select='success']")

        assert_button_click(
            driver,
            "//div[@data-jobs-setting-select-dropdown='success']/button[@value='all']",
        )

        print(
            "Success filter is working, trying to filter by reload state ...",
            flush=True,
        )

        sleep(0.3)

        assert_button_click(driver, "//button[@data-jobs-setting-select='reload']")

        assert_button_click(
            driver,
            "//div[@data-jobs-setting-select-dropdown='reload']/button[@value='true']",
        )

        with suppress(TimeoutException):
            if (
                len(jobs_list)
                == len(
                    safe_get_element(
                        driver,
                        By.XPATH,
                        "//ul[@data-jobs-list='']/li[not(contains(@class, 'hidden'))]",
                        multiple=True,
                        error=True,
                    )
                )
            ) and len(jobs_list) != len(
                safe_get_element(
                    driver,
                    By.XPATH,
                    "//ul[@data-jobs-list='']//p[@data-jobs-reload='true']",
                    multiple=True,
                    error=True,
                )
            ):
                print("The reload filter is not working, exiting ...", flush=True)
                exit(1)

        assert_button_click(driver, "//button[@data-jobs-setting-select='reload']")

        assert_button_click(
            driver,
            "//div[@data-jobs-setting-select-dropdown='reload']/button[@value='all']",
        )

        print("Reload filter is working, trying jobs cache ...", flush=True)

        sleep(0.3)

        resp = get(f"http://www.example.com{ui_url}/jobs/download?job_name=mmdb-country&file_name=country.mmdb")

        if resp.status_code != 200:
            print("The cache download is not working, exiting ...", flush=True)
            exit(1)

        print("Jobs cache download is working, trying logs page ...", flush=True)

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[3]/ul/li[11]/a", "logs")

        ### LOGS PAGE

        print("Selecting correct instance ...", flush=True)

        assert_button_click(driver, "//button[@data-logs-setting-select='instances']")

        instances = safe_get_element(
            driver,
            By.XPATH,
            "//div[@data-logs-setting-select-dropdown='instances']/button",
            multiple=True,
        )

        first_instance = instances[0].text

        if len(instances) == 0:
            print("No instances found, exiting ...", flush=True)
            exit(1)

        assert_button_click(driver, instances[0])
        assert_button_click(driver, safe_get_element(driver, By.ID, "submit-data"))

        sleep(3)

        logs_list = safe_get_element(driver, By.XPATH, "//ul[@data-logs-list='']/li", multiple=True)

        if len(logs_list) == 0:
            print("No logs found, exiting ...", flush=True)
            exit(1)

        print("Logs found, trying auto refresh ...", flush=True)

        assert_button_click(driver, safe_get_element(driver, By.ID, "live-update"))
        assert_button_click(driver, safe_get_element(driver, By.ID, "submit-live"))

        sleep(3)

        if len(logs_list) == len(
            safe_get_element(
                driver,
                By.XPATH,
                "//ul[@data-logs-list='']/li[not(contains(@class, 'hidden'))]",
                multiple=True,
            )
        ):
            print("Auto refresh is not working, exiting ...", flush=True)
            exit(1)

        print("Auto refresh is working, deactivating it ...", flush=True)

        assert_button_click(driver, safe_get_element(driver, By.ID, "live-update"))
        assert_button_click(driver, safe_get_element(driver, By.ID, "submit-data"))

        sleep(3)

        logs_list = safe_get_element(driver, By.XPATH, "//ul[@data-logs-list='']/li", multiple=True)

        print("Trying filters ...", flush=True)

        filter_input = safe_get_element(driver, By.ID, "keyword")

        filter_input.send_keys("gen")

        sleep(3)

        if len(logs_list) == len(
            safe_get_element(
                driver,
                By.XPATH,
                "//ul[@data-logs-list='']/li[not(contains(@class, 'hidden'))]",
                multiple=True,
            )
        ):
            print("The keyword filter is not working, exiting ...", flush=True)
            exit(1)

        filter_input.clear()

        print("Keyword filter is working, trying type filter ...", flush=True)

        assert_button_click(driver, "//button[@data-logs-setting-select='types']")

        assert_button_click(
            driver,
            "//div[@data-logs-setting-select-dropdown='types']/button[@value='warn']",
        )

        if len(logs_list) == len(
            safe_get_element(
                driver,
                By.XPATH,
                "//ul[@data-logs-list='']/li[not(contains(@class, 'hidden'))]",
                multiple=True,
            )
        ):
            print("The keyword filter is not working, exiting ...", flush=True)
            exit(1)

        assert_button_click(driver, "//button[@data-logs-setting-select='types']")

        assert_button_click(
            driver,
            "//div[@data-logs-setting-select-dropdown='types']/button[@value='all']",
        )

        print("Type filter is working, trying to filter by date ...", flush=True)

        current_date = datetime.now()
        resp = get(
            f"http://www.example.com{ui_url}/logs/{first_instance}?from_date={int((current_date - timedelta(weeks=1)).timestamp())}&to_date={int((current_date - timedelta(days=1)).timestamp())}",
            headers={"Host": "www.example.com", "User-Agent": driver.execute_script("return navigator.userAgent;")},
            cookies={"session": driver.get_cookies()[0]["value"]},
        )

        if len(resp.json()["logs"]) != 0:
            print("The date filter is not working, exiting ...", flush=True)
            exit(1)

        print("Date filter is working, trying jobs page ...", flush=True)

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[2]/a", "account")

        ### ACCOUNT PAGE

        username_input = safe_get_element(driver, By.ID, "admin_username")

        if username_input.get_attribute("value") != "admin":
            print("The username is not correct, exiting ...", flush=True)
            exit(1)

        username_input.clear()
        username_input.send_keys("admin2")

        password_input = safe_get_element(driver, By.ID, "curr_password")

        if password_input.get_attribute("value") != "":
            print("The current password is not empty, exiting ...", flush=True)
            exit(1)

        password_input.send_keys("S$cr3tP@ssw0rd")

        assert_button_click(driver, "//button[@id='username-button' and @class='edit-btn']")

        try:
            title = driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/main/div[1]/div/h1")))

            if title.text != "Log in":
                print("Didn't get redirected to login page, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("Login page didn't load in time, exiting ...", flush=True)
            exit(1)

        print("Successfully changed username, trying to log in with new username ...", flush=True)

        username_input = safe_get_element(driver, By.ID, "username")
        password_input = safe_get_element(driver, By.ID, "password")
        username_input.send_keys("admin2")
        password_input.send_keys("S$cr3tP@ssw0rd")

        access_page(driver, driver_wait, "//button[@value='login']", "home")

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[2]/a", "account")

        username_input = safe_get_element(driver, By.ID, "admin_username")

        if username_input.get_attribute("value") != "admin2":
            print("The username is not correct, exiting ...", flush=True)
            exit(1)

        print("Successfully logged in with new username, trying to change password ...", flush=True)

        assert_button_click(driver, "//button[@data-tab-handler='password']")

        password_input = safe_get_element(driver, By.XPATH, "//form[@data-plugin-item='password']//input[@id='curr_password']")

        if password_input.get_attribute("value") != "":
            print("The current password is not empty, exiting ...", flush=True)
            exit(1)

        password_input.send_keys("S$cr3tP@ssw0rd")

        new_password_input = safe_get_element(driver, By.ID, "admin_password")

        if new_password_input.get_attribute("value") != "":
            print("The new password is not empty, exiting ...", flush=True)
            exit(1)

        new_password_input.send_keys("P@ssw0rd")

        new_password_check_input = safe_get_element(driver, By.ID, "admin_password_check")

        if new_password_check_input.get_attribute("value") != "":
            print("The new password check is not empty, exiting ...", flush=True)
            exit(1)

        new_password_check_input.send_keys("P@ssw0rd")

        assert_button_click(driver, "//button[@id='pw-button' and @class='edit-btn']")

        try:
            title = driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/main/div[1]/div/h1")))

            if title.text != "Log in":
                print("Didn't get redirected to login page, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("Login page didn't load in time, exiting ...", flush=True)
            exit(1)

        print("Successfully changed username, trying to log in with new password ...", flush=True)

        username_input = safe_get_element(driver, By.ID, "username")
        password_input = safe_get_element(driver, By.ID, "password")
        username_input.send_keys("admin2")
        password_input.send_keys("P@ssw0rd")

        access_page(driver, driver_wait, "//button[@value='login']", "home")

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[2]/a", "account")

        print("Successfully logged in with new password, trying 2FA ...", flush=True)

        assert_button_click(driver, "//button[@data-tab-handler='totp']")

        secret_token_input = safe_get_element(driver, By.ID, "secret_token")
        secret_token = secret_token_input.get_attribute("value")

        driver.refresh()

        driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/div/header/div/nav/h6")))

        assert_button_click(driver, "//button[@data-tab-handler='totp']")

        secret_token_input = safe_get_element(driver, By.ID, "secret_token")
        new_secret_token = secret_token_input.get_attribute("value")

        if new_secret_token == secret_token:
            print("The secret token hasn't been changed, exiting ...", flush=True)
            exit(1)

        print("The secret token has been changed, trying to activate 2FA ...", flush=True)

        totp = TOTP(new_secret_token)
        totp_input = safe_get_element(driver, By.ID, "totp_token")
        totp_input.send_keys(totp.now())

        password_input = safe_get_element(driver, By.XPATH, "//form[@data-plugin-item='totp']//input[@id='curr_password']")

        if password_input.get_attribute("value") != "":
            print("The new password check is not empty, exiting ...", flush=True)
            exit(1)

        password_input.send_keys("P@ssw0rd")

        access_page(driver, driver_wait, "//button[@id='totp-button' and @class='valid-btn']", "account")

        assert_button_click(driver, "//button[@data-tab-handler='totp']")

        try:
            totp_state = safe_get_element(driver, By.XPATH, "/html/body/main/div/div/form[2]/h5")

            if totp_state.text != "TOTP IS CURRENTLY ON":
                print("TOTP is not activated, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("TOTP has not been activated, exiting ...", flush=True)
            exit(1)

        print("2FA has been activated, trying to log out ...", flush=True)

        assert_button_click(driver, "//a[@href='logout']")

        try:
            title = driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/main/div[1]/div/h1")))

            if title.text != "Log in":
                print("Didn't get redirected to login page, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("Login page didn't load in time, exiting ...", flush=True)
            exit(1)

        print("Successfully logged out, trying to log in with 2FA ...", flush=True)

        username_input = safe_get_element(driver, By.ID, "username")
        password_input = safe_get_element(driver, By.ID, "password")
        username_input.send_keys("admin2")
        password_input.send_keys("P@ssw0rd")

        assert_button_click(driver, "//button[@value='login']")

        try:
            totp_input = safe_get_element(driver, By.ID, "totp_token")
        except TimeoutException:
            print("Didn't get redirected to 2FA page, exiting ...", flush=True)
            exit(1)

        totp_input.send_keys("0000000")
        assert_button_click(driver, "//button[@value='login']")

        sleep(5)

        if not driver.current_url.endswith("/totp"):
            print("Didn't get redirected back to 2FA page, exiting ...", flush=True)
            exit(1)

        totp_input = safe_get_element(driver, By.ID, "totp_token")
        totp_input.send_keys(totp.now())

        access_page(
            driver,
            driver_wait,
            "//button[@value='login']",
            "home",
        )

        print("Successfully logged in with 2FA, trying to deactivate 2FA ...", flush=True)

        access_page(driver, driver_wait, "/html/body/aside[1]/div[1]/div[2]/a", "account")

        assert_button_click(driver, "//button[@data-tab-handler='totp']")

        totp_input = safe_get_element(driver, By.ID, "totp_token")
        totp_input.send_keys(totp.now())

        password_input = safe_get_element(driver, By.XPATH, "//form[@data-plugin-item='totp']//input[@id='curr_password']")
        password_input.send_keys("P@ssw0rd")

        access_page(
            driver,
            driver_wait,
            "//button[@id='totp-button' and @class='delete-btn']",
            "account",
        )

        assert_button_click(driver, "//button[@data-tab-handler='totp']")

        try:
            totp_state = safe_get_element(driver, By.XPATH, "/html/body/main/div/div/form[2]/h5")

            if totp_state.text != "TOTP IS CURRENTLY OFF":
                print("TOTP is not deactivated, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("TOTP has not been deactivated, exiting ...", flush=True)
            exit(1)

        print("2FA has been deactivated, trying to log out ...", flush=True)

        assert_button_click(driver, "//a[@href='logout']")

        try:
            title = driver_wait.until(EC.presence_of_element_located((By.XPATH, "/html/body/main/div[1]/div/h1")))

            if title.text != "Log in":
                print("Didn't get redirected to login page, exiting ...", flush=True)
                exit(1)
        except TimeoutException:
            print("Login page didn't load in time, exiting ...", flush=True)
            exit(1)

        print("Successfully logged out, trying to log in without 2FA ...", flush=True)

        username_input = safe_get_element(driver, By.ID, "username")
        password_input = safe_get_element(driver, By.ID, "password")
        username_input.send_keys("admin2")
        password_input.send_keys("P@ssw0rd")

        access_page(
            driver,
            driver_wait,
            "//button[@value='login']",
            "home",
        )

        print("Successfully logged in without 2FA, tests are done, exiting ...", flush=True)
    except KeyboardInterrupt:
        pass
    except SystemExit:
        exit(1)
    except:
        print(f"Something went wrong, exiting ...\n{format_exc()}", flush=True)
        driver.save_screenshot("error.png")
        exit(1)

exit(0)