Examples test Chromium run on Ubuntu runners

Signed-off-by: Viet Nguyen Duc <[email protected]>

Author: VietND96

examples/java/src/test/java/dev/selenium/browsers/ChromeTest.java

@@ -26,6 +26,19 @@
 
 
 public class ChromeTest extends BaseTest {
+  private static Path getUserDataDir() {
+    String homeDir = System.getProperty("user.home");
+    Path seleniumDir = Paths.get(homeDir, ".chromium");
+    if (!Files.exists(seleniumDir)) {
+        try {
+            Files.createDirectories(seleniumDir);
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+    }
+    return seleniumDir;
+  }
+
   @AfterEach
   public void clearProperties() {
     System.clearProperty(ChromeDriverService.CHROME_DRIVER_LOG_PROPERTY);
@@ -35,13 +48,14 @@ public void clearProperties() {
   @Test
   public void basicOptions() {
     ChromeOptions options = new ChromeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     driver = new ChromeDriver(options);
   }
 
   @Test
   public void arguments() {
     ChromeOptions options = new ChromeOptions();
-
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.addArguments("--start-maximized");
 
     driver = new ChromeDriver(options);
@@ -50,7 +64,7 @@ public void arguments() {
   @Test
   public void setBrowserLocation() {
     ChromeOptions options = new ChromeOptions();
-
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.setBinary(getChromeLocation());
 
     driver = new ChromeDriver(options);
@@ -59,6 +73,7 @@ public void setBrowserLocation() {
   @Test
   public void extensionOptions() {
     ChromeOptions options = new ChromeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     Path path = Paths.get("src/test/resources/extensions/webextensions-selenium-example.crx");
     File extensionFilePath = new File(path.toUri());
 
@@ -74,7 +89,7 @@ public void extensionOptions() {
   @Test
   public void excludeSwitches() {
     ChromeOptions options = new ChromeOptions();
-
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.setExperimentalOption("excludeSwitches", List.of("disable-popup-blocking"));
 
     driver = new ChromeDriver(options);
@@ -83,6 +98,7 @@ public void excludeSwitches() {
   @Test
   public void loggingPreferences() {
     ChromeOptions options = new ChromeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     LoggingPreferences logPrefs = new LoggingPreferences();
     logPrefs.enable(LogType.PERFORMANCE, Level.ALL);
     options.setCapability(ChromeOptions.LOGGING_PREFS, logPrefs);
@@ -176,6 +192,7 @@ public void disableBuildChecks() throws IOException {
 
   private File getChromeLocation() {
     ChromeOptions options = new ChromeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.setBrowserVersion("stable");
     DriverFinder finder = new DriverFinder(ChromeDriverService.createDefaultService(), options);
     return new File(finder.getBrowserPath());

examples/java/src/test/java/dev/selenium/browsers/EdgeTest.java

@@ -27,6 +27,19 @@
 
 
 public class EdgeTest extends BaseTest {
+  private static Path getUserDataDir() {
+    String homeDir = System.getProperty("user.home");
+    Path seleniumDir = Paths.get(homeDir, ".chromium");
+    if (!Files.exists(seleniumDir)) {
+      try {
+        Files.createDirectories(seleniumDir);
+      } catch (IOException e) {
+        throw new RuntimeException(e);
+      }
+    }
+    return seleniumDir;
+  }
+
   @AfterEach
   public void clearProperties() {
     System.clearProperty(EdgeDriverService.EDGE_DRIVER_LOG_PROPERTY);
@@ -36,13 +49,14 @@ public void clearProperties() {
   @Test
   public void basicOptions() {
     EdgeOptions options = new EdgeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     driver = new EdgeDriver(options);
   }
 
   @Test
   public void arguments() {
     EdgeOptions options = new EdgeOptions();
-
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.addArguments("--start-maximized");
 
     driver = new EdgeDriver(options);
@@ -51,7 +65,7 @@ public void arguments() {
   @Test
   public void setBrowserLocation() {
     EdgeOptions options = new EdgeOptions();
-
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.setBinary(getEdgeLocation());
 
     driver = new EdgeDriver(options);
@@ -60,6 +74,7 @@ public void setBrowserLocation() {
   @Test
   public void extensionOptions() {
     EdgeOptions options = new EdgeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     Path path = Paths.get("src/test/resources/extensions/webextensions-selenium-example.crx");
     File extensionFilePath = new File(path.toUri());
 
@@ -75,7 +90,7 @@ public void extensionOptions() {
   @Test
   public void excludeSwitches() {
     EdgeOptions options = new EdgeOptions();
-
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.setExperimentalOption("excludeSwitches", List.of("disable-popup-blocking"));
 
     driver = new EdgeDriver(options);
@@ -84,6 +99,7 @@ public void excludeSwitches() {
   @Test
   public void loggingPreferences() {
     EdgeOptions options = new EdgeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     LoggingPreferences logPrefs = new LoggingPreferences();
     logPrefs.enable(LogType.PERFORMANCE, Level.ALL);
     options.setCapability(EdgeOptions.LOGGING_PREFS, logPrefs);
@@ -171,6 +187,7 @@ public void disableBuildChecks() throws IOException {
 
   private File getEdgeLocation() {
     EdgeOptions options = new EdgeOptions();
+    options.addArguments(String.format("--user-data-dir=%s", getUserDataDir()));
     options.setBrowserVersion("stable");
     DriverFinder finder = new DriverFinder(EdgeDriverService.createDefaultService(), options);
     return new File(finder.getBrowserPath());

examples/python/tests/browsers/test_chrome.py

@@ -5,16 +5,24 @@
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 
+def get_user_data_dir():
+    home_dir = os.path.expanduser("~")
+    selenium_dir = os.path.join(home_dir, ".chromium")
+    if not os.path.exists(selenium_dir):
+        os.makedirs(selenium_dir)
+    return selenium_dir
+
 def test_basic_options():
     options = webdriver.ChromeOptions()
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     driver = webdriver.Chrome(options=options)
 
     driver.quit()
 
 
 def test_args():
     options = webdriver.ChromeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.add_argument("--start-maximized")
 
     driver = webdriver.Chrome(options=options)
@@ -25,7 +33,7 @@ def test_args():
 
 def test_set_browser_location(chrome_bin):
     options = webdriver.ChromeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.binary_location = chrome_bin
 
     driver = webdriver.Chrome(options=options)
@@ -35,6 +43,7 @@ def test_set_browser_location(chrome_bin):
 
 def test_add_extension():
     options = webdriver.ChromeOptions()
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     extension_file_path = os.path.abspath("tests/extensions/webextensions-selenium-example.crx")
 
     options.add_extension(extension_file_path)
@@ -47,7 +56,7 @@ def test_add_extension():
 
 def test_keep_browser_open():
     options = webdriver.ChromeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.add_experimental_option("detach", True)
 
     driver = webdriver.Chrome(options=options)
@@ -58,7 +67,7 @@ def test_keep_browser_open():
 
 def test_exclude_switches():
     options = webdriver.ChromeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.add_experimental_option('excludeSwitches', ['disable-popup-blocking'])
 
     driver = webdriver.Chrome(options=options)

examples/python/tests/browsers/test_edge.py

@@ -5,16 +5,24 @@
 from selenium import webdriver
 from selenium.webdriver.common.by import By
 
+def get_user_data_dir():
+    home_dir = os.path.expanduser("~")
+    selenium_dir = os.path.join(home_dir, ".chromium")
+    if not os.path.exists(selenium_dir):
+        os.makedirs(selenium_dir)
+    return selenium_dir
+
 def test_basic_options():
     options = webdriver.EdgeOptions()
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     driver = webdriver.Edge(options=options)
 
     driver.quit()
 
 
 def test_args():
     options = webdriver.EdgeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.add_argument("--start-maximized")
 
     driver = webdriver.Edge(options=options)
@@ -25,7 +33,7 @@ def test_args():
 
 def test_set_browser_location(edge_bin):
     options = webdriver.EdgeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.binary_location = edge_bin
 
     driver = webdriver.Edge(options=options)
@@ -35,6 +43,7 @@ def test_set_browser_location(edge_bin):
 
 def test_add_extension():
     options = webdriver.EdgeOptions()
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     extension_file_path = os.path.abspath("tests/extensions/webextensions-selenium-example.crx")
 
     options.add_extension(extension_file_path)
@@ -47,7 +56,7 @@ def test_add_extension():
 
 def test_keep_browser_open():
     options = webdriver.EdgeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.add_experimental_option("detach", True)
 
     driver = webdriver.Edge(options=options)
@@ -58,7 +67,7 @@ def test_keep_browser_open():
 
 def test_exclude_switches():
     options = webdriver.EdgeOptions()
-
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.add_experimental_option('excludeSwitches', ['disable-popup-blocking'])
 
     driver = webdriver.Edge(options=options)