[java] Add remotedriverbuilder tests

pujagani · web-flow · commit dded957a46bc · 2022-01-11T10:30:46.000+05:30
diff --git a/java/test/org/openqa/selenium/remote/FakeWebDriverInfo.java b/java/test/org/openqa/selenium/remote/FakeWebDriverInfo.java
@@ -9,6 +9,7 @@
 import org.openqa.selenium.SessionNotCreatedException;
 import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.WebDriverInfo;
+import org.openqa.selenium.devtools.HasDevTools;
 
 import java.util.Optional;
 
@@ -32,7 +33,7 @@ public boolean isSupporting(Capabilities capabilities) {
 
   @Override
   public boolean isSupportingCdp() {
-    return false;
+    return true;
   }
 
   @Override
@@ -52,7 +53,7 @@ public Optional<WebDriver> createDriver(Capabilities capabilities)
     return Optional.of(new FakeWebDriver());
   }
 
-  private static class FakeWebDriver extends RemoteWebDriver {
+  public static class FakeWebDriver extends RemoteWebDriver {
 
     @Override
     protected void startSession(Capabilities capabilities) {
diff --git a/java/test/org/openqa/selenium/remote/RemoteWebDriverBuilderTest.java b/java/test/org/openqa/selenium/remote/RemoteWebDriverBuilderTest.java
@@ -23,6 +23,7 @@
 import org.openqa.selenium.ImmutableCapabilities;
 import org.openqa.selenium.Platform;
 import org.openqa.selenium.SessionNotCreatedException;
+import org.openqa.selenium.WebDriver;
 import org.openqa.selenium.chrome.ChromeOptions;
 import org.openqa.selenium.firefox.FirefoxOptions;
 import org.openqa.selenium.ie.InternetExplorerOptions;
@@ -351,10 +352,36 @@ public void commandsShouldBeSentWithW3CHeaders() {
 
   @Test
   public void shouldUseWebDriverInfoToFindAMatchingDriverImplementationForRequestedCapabilitiesIfRemoteUrlNotSet() {
+    WebDriver driver = RemoteWebDriver.builder()
+      .oneOf(new ImmutableCapabilities("browser", "selenium-test"))
+      .connectingWith(config -> req -> CANNED_SESSION_RESPONSE)
+      .build();
+
+    assertThat(driver).isInstanceOf(FakeWebDriverInfo.FakeWebDriver.class);
   }
 
   @Test
   public void shouldAugmentDriverIfPossible() {
+    HttpResponse response = new HttpResponse()
+      .setContent(Contents.asJson(ImmutableMap.of(
+        "value", ImmutableMap.of(
+          "sessionId", SESSION_ID,
+          "capabilities", new ImmutableCapabilities("firefox", "caps")))));
+
+    Augmenter augmenter = new Augmenter().addDriverAugmentation("firefox",
+                                                                AugmenterTest.HasMagicNumbers.class,
+                                                                (c, exe) -> () -> 1);
+    WebDriver driver = RemoteWebDriver.builder()
+      .oneOf(new FirefoxOptions())
+      .augmentUsing(augmenter)
+      .address("http://localhost:34576")
+      .connectingWith(config -> req -> response)
+      .build();
+
+    int number = ((AugmenterTest.HasMagicNumbers)driver).getMagicNumber();
+
+    assertThat(driver).isInstanceOf(AugmenterTest.HasMagicNumbers.class);
+    assertThat(number).isEqualTo(1);
   }
 
   @SuppressWarnings("unchecked")
