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

@@ -35,13 +35,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 +51,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 +60,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 +76,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 +85,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 +179,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());
@@ -258,4 +262,12 @@ public void getBrowserLogs() {
     Assertions.assertTrue(logFound, "No matching log message found.");
     driver.quit();
   }
+
+  private static Path getUserDataDir() {
+    try {
+      return Files.createTempDirectory("").toAbsolutePath();
+    } catch (IOException e) {
+      throw new RuntimeException(e);
+    }
+  }
 }

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

@@ -36,13 +36,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 +52,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 +61,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 +77,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 +86,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 +174,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());
@@ -253,4 +257,12 @@ public void getBrowserLogs() {
     Assertions.assertTrue(logFound, "No matching log message found.");
     driver.quit();
   }
+
+  private static Path getUserDataDir() {
+    try {
+      return Files.createTempDirectory("").toAbsolutePath();
+    } catch (IOException e) {
+      throw new RuntimeException(e);
+    }
+  }
 }

examples/python/tests/browsers/test_chrome.py

@@ -7,14 +7,15 @@
 
 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 +26,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 +36,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 +49,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 +60,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)
@@ -188,3 +190,9 @@ def test_get_browser_logs():
     # Assert that at least one log contains the expected message
     assert any("I am console error" in log['message'] for log in logs), "No matching log message found."
     driver.quit()
+
+def get_user_data_dir():
+    import tempfile
+    import os
+    temp_dir = tempfile.mkdtemp(prefix="")
+    return os.path.abspath(temp_dir)

examples/python/tests/browsers/test_edge.py

@@ -7,14 +7,15 @@
 
 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 +26,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 +36,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 +49,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 +60,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)
@@ -188,3 +190,9 @@ def test_get_browser_logs():
     # Assert that at least one log contains the expected message
     assert any("I am console error" in log['message'] for log in logs), "No matching log message found."
     driver.quit()
+
+def get_user_data_dir():
+    import tempfile
+    import os
+    temp_dir = tempfile.mkdtemp(prefix="")
+    return os.path.abspath(temp_dir)

examples/python/tests/drivers/test_service.py

@@ -10,6 +10,7 @@ def test_basic_service():
 
 def test_driver_location(chromedriver_bin, chrome_bin):
     options = webdriver.ChromeOptions()
+    options.add_argument(f"--user-data-dir={get_user_data_dir()}")
     options.binary_location = chrome_bin
 
     service = webdriver.ChromeService(executable_path=chromedriver_bin)
@@ -25,3 +26,9 @@ def test_driver_port():
     driver = webdriver.Chrome(service=service)
 
     driver.quit()
+
+def get_user_data_dir():
+    import tempfile
+    import os
+    temp_dir = tempfile.mkdtemp(prefix="")
+    return os.path.abspath(temp_dir)