Add selenium test

Author: edan-bainglass

tests_notebooks/test_notification.py

@@ -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()