chore(ci): pass --no-sandbox option only to chrome browser

zama-ai · Oct 14, 2024 · 41b3edf · 41b3edf
1 parent 9a06f62
commit 41b3edf
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/ci/webdriver.py b/ci/webdriver.py
@@ -135,17 +135,17 @@ def __init__(self, browser_path, driver_path, browser_kind, threaded_logs=False)
         match self.browser_kind:
             case BrowserKind.chrome:
                 self.options = ChromeOptions()
+                if os.getuid() == 0:
+                    # If user ID is root then driver needs to run in no-sandbox mode.
+                    print(
+                        "Script is running as root, running browser with --no-sandbox for compatibility"
+                    )
+                self.options.add_argument("--no-sandbox")
             case BrowserKind.firefox:
                 self.options = FirefoxOptions()
 
         self.options.binary_location = self.browser_path
         self.options.add_argument("--headless")
-        if os.getuid() == 0:
-            # If user ID is root then driver needs to run in no-sandbox mode.
-            print(
-                "Script is running as root, running browser with --no-sandbox for compatibility"
-            )
-            self.options.add_argument("--no-sandbox")
 
         self._driver = None