setup_folders.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536
  1. #!/usr/bin/env python3
  2. """Open Folder 1 from dashboards/3-folders.adoc."""
  3. import time
  4. from selenium.webdriver.common.by import By
  5. from selenium.webdriver.support.ui import WebDriverWait
  6. def _wait_for_folder_dashboard(driver, timeout=30):
  7. WebDriverWait(driver, timeout).until(
  8. lambda d: d.find_element(By.TAG_NAME, "body").get_attribute("loaded-dashboard")
  9. == "Folder 1"
  10. )
  11. def ready(d):
  12. try:
  13. action1 = d.find_element(By.CSS_SELECTOR, '[title="Action 1"]')
  14. action2 = d.find_element(By.CSS_SELECTOR, '[title="Action 2"]')
  15. subfolder = d.find_element(
  16. By.XPATH,
  17. '//button[contains(@class, "directory-button")]//span[contains(@class, "title") and text()="Subfolder 2"]',
  18. )
  19. except Exception:
  20. return False
  21. return all(
  22. element.is_displayed()
  23. for element in (action1, action2, subfolder)
  24. )
  25. WebDriverWait(driver, timeout).until(ready)
  26. def run(driver):
  27. _wait_for_folder_dashboard(driver)
  28. time.sleep(0.2)