Setup infra / Implement basic frontend test

Author: Kirandawadi

compose/local/django/Dockerfile

@@ -52,6 +52,7 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
   && wget --quiet -O - https://www.postgresql.org/media/keys/ACCC4CF8.asc | apt-key add - \
   && apt-get update \
   && apt-get install -y postgresql-15 postgresql-client-15 \
+  && apt-get install -y chromium chromium-driver \
   # cleaning up unused files
   && apt-get purge -y --auto-remove -o APT::AutoRemove::RecommendsImportant=false \
   && rm -rf /var/lib/apt/lists/*

requirements/local.txt

@@ -13,6 +13,9 @@ pytest==8.0.0  # https://github.com/pytest-dev/pytest
 pytest-sugar==1.0.0  # https://github.com/Frozenball/pytest-sugar
 types-requests # maybe instead, we should add `mypy --install-types` to the dockerfile?
 types-xmltodict
+pytest-xdist>=3.3.1
+pytest-cov>=4.1.0
+selenium>=4.15.2    # Selenium (Frontend Testing)
 
 # Documentation
 # ------------------------------------------------------------------------------

sde_collections/tests/frontend/__init__.py

@@ -0,0 +1,59 @@
+import shutil
+import subprocess
+
+import pytest
+from django.contrib.staticfiles.testing import StaticLiveServerTestCase
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.support.ui import WebDriverWait
+
+
+class BaseTestCase(StaticLiveServerTestCase):
+    """Base class for all frontend tests using Selenium."""
+
+    @classmethod
+    def setUpClass(cls):
+        super().setUpClass()
+
+        # Verify ChromeDriver is available
+        chromedriver_path = shutil.which("chromedriver")
+        if not chromedriver_path:
+            pytest.fail("ChromeDriver not found. Please ensure chromium-driver is installed.")
+
+        # # Set up Chrome options
+        chrome_options = Options()
+        chrome_options.add_argument("--headless")
+        chrome_options.add_argument("--no-sandbox")
+        chrome_options.add_argument("--disable-dev-shm-usage")
+        chrome_options.binary_location = "/usr/bin/chromium"
+
+        try:
+            # Create service with explicit path
+            service = Service(executable_path=chromedriver_path, log_path="/tmp/chromedriver.log")
+
+            # Initialize WebDriver with service and options
+            cls.driver = webdriver.Chrome(service=service, options=chrome_options)
+
+            cls.driver.set_window_size(1920, 1080)
+            cls.driver.implicitly_wait(10)
+            cls.wait = WebDriverWait(cls.driver, 10)
+
+        except Exception as e:
+            # Print debugging information
+            subprocess.run(["which", "chromium"])
+            subprocess.run(["which", "chromedriver"])
+            subprocess.run(["chromium", "--version"])
+            subprocess.run(["chromedriver", "--version"])
+            pytest.fail(f"Failed to initialize ChromeDriver: {str(e)}")
+
+    @classmethod
+    def tearDownClass(cls):
+        if hasattr(cls, "driver"):
+            cls.driver.quit()
+        super().tearDownClass()
+
+    def setUp(self):
+        """Set up test case."""
+        super().setUp()
+        # Add any additional setup here

sde_collections/tests/frontend/base.py

@@ -0,0 +1,16 @@
+from .base import BaseTestCase
+
+
+class TestSetup(BaseTestCase):
+    """Verify Selenium setup is working correctly."""
+
+    def test_basic_page_load(self):
+        """Test that we can load a page."""
+        # Print the live server URL
+        print(f"\nTest server running at: {self.live_server_url}")
+
+        self.driver.get(self.live_server_url)
+        print(f"Current URL: {self.driver.current_url}")
+        print(f"Page Title: {self.driver.title}")
+
+        assert self.driver.title == "Sign In | COSMOS"