Method augment

java/src/org/openqa/selenium/remote/Augmenter.java:144–175

Enhance the interfaces implemented by this instance of WebDriver if that instance is a org.openqa.selenium.remote.RemoteWebDriver. The WebDriver that is returned may well be a dynamic proxy. You cannot rely on the concrete implementing class to remain constant. @param driver The driver to e

(WebDriver driver)

   * @return A class implementing the described interfaces.
   */
  public WebDriver augment(WebDriver driver) {
    Require.nonNull("WebDriver", driver);
    Require.precondition(
        driver instanceof HasCapabilities, "Driver must have capabilities", driver);

    if (driver instanceof Decorated<?>) {
      LOG.warning(
          "Warning: In future versions, passing a decorated driver will no longer be allowed.\n"
              + " Instead, augment the driver first and then use it to created a decorated"
              + " driver.\n"
              + " Explanation: Decorated drivers are not aware of the augmentations applied to"
              + " them. It can lead to expected behavior.\n"
              + " For example, augmenting HasDevTools interface to a decorated driver. \n"
              + " The decorated driver is not aware that after augmentation it is an instance of"
              + " HasDevTools. So it does not invoke the close() method of the underlying"
              + " websocket, potentially causing a memory leak. ");
    }

    Capabilities caps = ImmutableCapabilities.copyOf(((HasCapabilities) driver).getCapabilities());

    // Collect the interfaces to apply
    List<Augmentation<?>> matchingAugmenters =
        augmentations.stream()
            // Only consider the augmenters that match interfaces we don't already implement
            .filter(
                augmentation -> !augmentation.interfaceClass.isAssignableFrom(driver.getClass()))
            // And which match an augmentation we have
            .filter(augmentation -> augmentation.whenMatches.test(caps))
            .collect(Collectors.toList());

    return this.augment(driver, matchingAugmenters);
  }

  private WebDriver augment(WebDriver driver, List<Augmentation<?>> matchingAugmenters) {
    Capabilities caps = ImmutableCapabilities.copyOf(((HasCapabilities) driver).getCapabilities());