Skip to content

Commit

Permalink
Add selenium test
Browse files Browse the repository at this point in the history
  • Loading branch information
edan-bainglass committed Dec 4, 2024
1 parent cd0554b commit 19a5cca
Showing 1 changed file with 33 additions and 0 deletions.
33 changes: 33 additions & 0 deletions tests_notebooks/test_notification.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
import time
from pathlib import Path

from selenium.webdriver.common.by import By
from selenium.webdriver.remote.webdriver import WebDriver


def test_home_notification(selenium_driver, final_screenshot):
selenium: WebDriver = selenium_driver("start.ipynb")
selenium.set_window_size(1000, 941)

config_dir = Path.home() / ".aiidalab"
warning_file = config_dir / "home_app_warning.md"

# If the file exists, we are testing locally
# We should make sure not to overwrite or remove the file
testing_locally = warning_file.exists()

if not testing_locally:
config_dir.mkdir(exist_ok=True)
warning_file.write_text("This is a test warning message.")
selenium.refresh()
time.sleep(2)

notifications = selenium.find_elements(By.CLASS_NAME, "home-notification")
assert len(notifications) == 1

home_warning = notifications[0]
content = warning_file.read_text()
assert content in home_warning.text

if not testing_locally:
warning_file.unlink()

0 comments on commit 19a5cca

Please sign in to comment.