Setup ChromeDriver CICD

.github/workflows/github-actions.yml

@@ -32,6 +32,16 @@ jobs:
       run: |
         pdm install
 
+    - name: Setup Chrome and ChromeDriver
+      run: |
+        sudo apt-get update
+        sudo apt-get install -y google-chrome-stable
+        CHROME_VERSION=$(google-chrome --version | cut -d ' ' -f 3 | cut -d '.' -f 1)
+        wget https://chromedriver.storage.googleapis.com/$(curl -s https://chromedriver.storage.googleapis.com/LATEST_RELEASE_$CHROME_VERSION)/chromedriver_linux64.zip
+        unzip chromedriver_linux64.zip
+        sudo mv chromedriver /usr/local/bin/chromedriver
+        sudo chmod +x /usr/local/bin/chromedriver
+
     - name: Check Code Formatting with Black
       run: |
         pdm run black --check src/ tests/

tests/test_index.py

@@ -1,4 +1,5 @@
 import pytest
+import os
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 from selenium.webdriver.chrome.service import Service
@@ -8,7 +9,10 @@
 @pytest.fixture(scope="class")
 def browser():
     # Setup Selenium
-    service = Service(executable_path="/opt/homebrew/bin/chromedriver")
+    if os.getenv('CI'):  
+        service = Service()
+    else:
+        service = Service(executable_path="/opt/homebrew/bin/chromedriver")
     options = Options()
     options.add_argument("--headless")
     driver = webdriver.Chrome(service=service, options=options)