[test]: fix RBE tests

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

Author: VietND96

java/test/org/openqa/selenium/chrome/ChromeOptionsFunctionalTest.java

@@ -33,6 +33,7 @@
 import org.openqa.selenium.testing.JupiterTestBase;
 import org.openqa.selenium.testing.NoDriverBeforeTest;
 import org.openqa.selenium.testing.drivers.Browser;
+import org.openqa.selenium.testing.drivers.WebDriverBuilder;
 
 class ChromeOptionsFunctionalTest extends JupiterTestBase {
 
@@ -82,7 +83,7 @@ public void canSetAcceptInsecureCerts() {
   public void canAddExtensionFromFile() {
     ChromeOptions options = createChromeOptionsForExtensions();
     options.addExtensions(InProject.locate(EXT_PATH).toFile());
-    localDriver = seleniumExtension.createNewDriver(options);
+    localDriver = new WebDriverBuilder().get(options);
 
     localDriver.get(toLocalUrl(pages.echoPage));
     WebElement footerElement = localDriver.findElement(By.id("webextensions-selenium-example"));
@@ -97,7 +98,7 @@ public void canAddExtensionFromStringEncodedInBase64() throws IOException {
     ChromeOptions options = createChromeOptionsForExtensions();
     options.addEncodedExtensions(
         Base64.getEncoder().encodeToString(Files.readAllBytes(InProject.locate(EXT_PATH))));
-    localDriver = seleniumExtension.createNewDriver(options);
+    localDriver = new WebDriverBuilder().get(options);
 
     localDriver.get(pages.echoPage);
 
@@ -109,7 +110,9 @@ public void canAddExtensionFromStringEncodedInBase64() throws IOException {
 
   private ChromeOptions createChromeOptionsForExtensions() {
     ChromeOptions options = (ChromeOptions) Browser.CHROME.getCapabilities();
+    options.addArguments("--handle_prefs");
     return options.setExperimentalOption(
-        "prefs", ImmutableMap.of("extensions.ui.developer_mode", true));
+        "prefs",
+        ImmutableMap.of("extensions.ui.developer_mode", true, "extensions-on-chrome-urls", true));
   }
 }

java/test/org/openqa/selenium/edge/EdgeOptionsFunctionalTest.java

@@ -20,6 +20,7 @@
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.openqa.selenium.remote.CapabilityType.ACCEPT_INSECURE_CERTS;
 
+import com.google.common.collect.ImmutableMap;
 import java.io.IOException;
 import java.nio.file.Files;
 import java.util.Base64;
@@ -94,6 +95,9 @@ public void canAddExtensionFromFile() {
   @NoDriverBeforeTest
   public void canAddExtensionFromStringEncodedInBase64() throws IOException {
     EdgeOptions options = new EdgeOptions();
+    options.setExperimentalOption(
+        "prefs",
+        ImmutableMap.of("extensions.ui.developer_mode", true, "extensions-on-chrome-urls", true));
     options.addEncodedExtensions(
         Base64.getEncoder().encodeToString(Files.readAllBytes(InProject.locate(EXT_PATH))));
     localDriver = new WebDriverBuilder().get(options);