Best Selenium code snippet using org.openqa.selenium.support.decorators.DefaultDecorated.onError
Source:WebDriverDecorator.java  
...78 *   WebDriverDecorator and override some of the following methods:79 *   {@link #beforeCall(Decorated, Method, Object[])},80 *   {@link #afterCall(Decorated, Method, Object[], Object)},81 *   {@link #call(Decorated, Method, Object[])} and82 *   {@link #onError(Decorated, Method, Object[], InvocationTargetException)}</li>83 *   <li>if you want to modify behavior of a specific class instances only84 *   (e.g. behaviour of WebElement instances) you can override one of the85 *   overloaded <code>createDecorated</code> methods to create a non-trivial86 *   decorator for the specific class only.</li>87 * </ul>88 * Let's consider both approaches by examples.89 * <p>90 * One of the most widely used decorator examples is a logging decorator.91 * In this case we want to add the same piece of logging code before and after92 * each invoked method:93 * <code>94 *   public class LoggingDecorator extends WebDriverDecorator {95 *     final Logger logger = LoggerFactory.getLogger(Thread.currentThread().getName());96 *97 *     @Override98 *     public void beforeCall(Decorated<?> target, Method method, Object[] args) {99 *       logger.debug("before {}.{}({})", target, method, args);100 *     }101 *     @Override102 *     public void afterCall(Decorated<?> target, Method method, Object[] args, Object res) {103 *       logger.debug("after {}.{}({}) => {}", target, method, args, res);104 *     }105 *   }106 * </code>107 * For the second example let's implement a decorator that implicitly waits108 * for an element to be visible before any click or sendKeys method call.109 * <code>110 *   public class ImplicitlyWaitingDecorator extends WebDriverDecorator {111 *     private WebDriverWait wait;112 *113 *     @Override114 *     public Decorated<WebDriver> createDecorated(WebDriver driver) {115 *       wait = new WebDriverWait(driver, Duration.ofSeconds(10));116 *       return super.createDecorated(driver);117 *     }118 *     @Override119 *     public Decorated<WebElement> createDecorated(WebElement original) {120 *       return new DefaultDecorated<>(original, this) {121 *         @Override122 *         public void beforeCall(Method method, Object[] args) {123 *           String methodName = method.getName();124 *           if ("click".equals(methodName) || "sendKeys".equals(methodName)) {125 *             wait.until(d -> getOriginal().isDisplayed());126 *           }127 *         }128 *       };129 *     }130 *   }131 * </code>132 * This class is not a pure decorator, it allows to not only add new behavior133 * but also replace "normal" behavior of a WebDriver or derived objects.134 * <p>135 * Let's suppose you want to use JavaScript-powered clicks instead of normal136 * ones (yes, this allows to interact with invisible elements, it's a bad137 * practice in general but sometimes it's inevitable). This behavior change138 * can be achieved with the following "decorator":139 * <code>140 *   public class JavaScriptPoweredDecorator extends WebDriverDecorator {141 *     @Override142 *     public Decorated<WebElement> createDecorated(WebElement original) {143 *       return new DefaultDecorated<>(original, this) {144 *         @Override145 *         public Object call(Method method, Object[] args) throws Throwable {146 *           String methodName = method.getName();147 *           if ("click".equals(methodName)) {148 *             JavascriptExecutor executor = (JavascriptExecutor) getDecoratedDriver().getOriginal();149 *             executor.executeScript("arguments[0].click()", getOriginal());150 *             return null;151 *           } else {152 *             return super.call(method, args);153 *           }154 *         }155 *       };156 *     }157 *   }158 * </code>159 * It is possible to apply multiple decorators to compose behaviors added160 * by each of them. For example, if you want to log method calls and161 * implicitly wait for elements visibility you can use two decorators:162 * <code>163 *   WebDriver original = new FirefoxDriver();164 *   WebDriver decorated =165 *     new ImplicitlyWaitingDecorator().decorate(166 *       new LoggingDecorator().decorate(original));167 * </code>168 */169@Beta170public class WebDriverDecorator {171  private Decorated<WebDriver> decorated;172  public final WebDriver decorate(WebDriver original) {173    Require.nonNull("WebDriver", original);174    decorated = createDecorated(original);175    return createProxy(decorated);176  }177  public Decorated<WebDriver> getDecoratedDriver() {178    return decorated;179  }180  public Decorated<WebDriver> createDecorated(WebDriver driver) {181    return new DefaultDecorated<>(driver, this);182  }183  public Decorated<WebElement> createDecorated(WebElement original) {184    return new DefaultDecorated<>(original, this);185  }186  public Decorated<WebDriver.TargetLocator> createDecorated(WebDriver.TargetLocator original) {187    return new DefaultDecorated<>(original, this);188  }189  public Decorated<WebDriver.Navigation> createDecorated(WebDriver.Navigation original) {190    return new DefaultDecorated<>(original, this);191  }192  public Decorated<WebDriver.Options> createDecorated(WebDriver.Options original) {193    return new DefaultDecorated<>(original, this);194  }195  public Decorated<WebDriver.Timeouts> createDecorated(WebDriver.Timeouts original) {196    return new DefaultDecorated<>(original, this);197  }198  public Decorated<WebDriver.Window> createDecorated(WebDriver.Window original) {199    return new DefaultDecorated<>(original, this);200  }201  public Decorated<Alert> createDecorated(Alert original) {202    return new DefaultDecorated<>(original, this);203  }204  public Decorated<VirtualAuthenticator> createDecorated(VirtualAuthenticator original) {205    return new DefaultDecorated<>(original, this);206  }207  public void beforeCall(Decorated<?> target, Method method, Object[] args) {}208  public Object call(Decorated<?> target, Method method, Object[] args) throws Throwable {209    return decorateResult(method.invoke(target.getOriginal(), args));210  }211  public void afterCall(Decorated<?> target, Method method, Object[] args, Object res) {}212  public Object onError(Decorated<?> target, Method method, Object[] args,213                        InvocationTargetException e) throws Throwable214  {215    throw e.getTargetException();216  }217  private Object decorateResult(Object toDecorate) {218    if (toDecorate instanceof WebDriver) {219      return createProxy(getDecoratedDriver());220    }221    if (toDecorate instanceof WebElement) {222      return createProxy(createDecorated((WebElement) toDecorate));223    }224    if (toDecorate instanceof Alert) {225      return createProxy(createDecorated((Alert) toDecorate));226    }227    if (toDecorate instanceof VirtualAuthenticator) {228      return createProxy(createDecorated((VirtualAuthenticator) toDecorate));229    }230    if (toDecorate instanceof WebDriver.Navigation) {231      return createProxy(createDecorated((WebDriver.Navigation) toDecorate));232    }233    if (toDecorate instanceof WebDriver.Options) {234      return createProxy(createDecorated((WebDriver.Options) toDecorate));235    }236    if (toDecorate instanceof WebDriver.TargetLocator) {237      return createProxy(createDecorated((WebDriver.TargetLocator) toDecorate));238    }239    if (toDecorate instanceof WebDriver.Timeouts) {240      return createProxy(createDecorated((WebDriver.Timeouts) toDecorate));241    }242    if (toDecorate instanceof WebDriver.Window) {243      return createProxy(createDecorated((WebDriver.Window) toDecorate));244    }245    if (toDecorate instanceof List) {246      return ((List<?>) toDecorate).stream()247        .map(this::decorateResult)248        .collect(Collectors.toList());249    }250    return toDecorate;251  }252  protected final <Z> Z createProxy(final Decorated<Z> decorated) {253    Set<Class<?>> decoratedInterfaces = extractInterfaces(decorated);254    Set<Class<?>> originalInterfaces = extractInterfaces(decorated.getOriginal());255    final InvocationHandler handler = (proxy, method, args) -> {256      try {257        if (method.getDeclaringClass().equals(Object.class)258            || decoratedInterfaces.contains(method.getDeclaringClass())) {259          return method.invoke(decorated, args);260        }261        if (originalInterfaces.contains(method.getDeclaringClass())) {262          decorated.beforeCall(method, args);263          Object result = decorated.call(method, args);264          decorated.afterCall(method, result, args);265          return result;266        }267        return method.invoke(decorated.getOriginal(), args);268      } catch (InvocationTargetException e) {269        return decorated.onError(method, e, args);270      }271    };272    Set<Class<?>> allInterfaces = new HashSet<>();273    allInterfaces.addAll(decoratedInterfaces);274    allInterfaces.addAll(originalInterfaces);275    Class<?>[] allInterfacesArray = allInterfaces.toArray(new Class<?>[0]);276    return (Z) Proxy.newProxyInstance(277      this.getClass().getClassLoader(), allInterfacesArray, handler);278  }279  static Set<Class<?>> extractInterfaces(final Object object) {280    return extractInterfaces(object.getClass());281  }282  private static Set<Class<?>> extractInterfaces(final Class<?> clazz) {283    Set<Class<?>> allInterfaces = new HashSet<>();...Source:DefaultDecorated.java  
...42  public void afterCall(Method method, Object result, Object[] args) {43    getDecorator().afterCall(this, method, args, result);44  }45  @Override46  public Object onError(Method method, InvocationTargetException e, Object[] args) throws Throwable {47    return getDecorator().onError(this, method, args, e);48  }49  @Override50  public String toString() {51    return String.format("Decorated {%s}", original);52  }53  @Override54  public boolean equals(Object o) {55    if (this == o) return true;56    if (o instanceof Decorated) {57      Decorated<?> that = (Decorated<?>) o;58      return original.equals(that.getOriginal());59    } else {60      return this.original.equals(o);61    }...onError
Using AI Code Generation
1public class DecoratorTest {2    public void testDecorator() {3        WebDriver driver = new FirefoxDriver();4        driver = new HtmlUnitDriver();5        driver = new Augmenter().augment(driver);6        driver = new EventFiringWebDriver(driver);7        driver = new EventFiringWebDriver(new Augmenter().augment(new HtmlUnitDriver()));8        driver = new Augmenter().augment(new EventFiringWebDriver(new HtmlUnitDriver()));9        driver = new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new HtmlUnitDriver())));10        driver = new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new HtmlUnitDriver()))));11        driver = new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new HtmlUnitDriver())))));12        driver = new Augmenter().augment(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new HtmlUnitDriver()))))));13        driver = new Augmenter().augment(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new HtmlUnitDriver()))))))));14        driver = new Augmenter().augment(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new HtmlUnitDriver()))))))))));15        driver = new Augmenter().augment(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new EventFiringWebDriver(new Augmenter().augment(new HtmlUnitDriver()))))))))))));onError
Using AI Code Generation
1DefaultDecorated decorated = new DefaultDecorated();2decorated.onError = { Throwable t -> 3    println "Error: " + t.getMessage()4}5WebDriverDefaultHandler handler = new WebDriverDefaultHandler();6handler.onError = { Throwable t -> 7    println "Error: " + t.getMessage()8}9WebDriverDecorator decorator = new WebDriverDecorator();10decorator.onError = { Throwable t -> 11    println "Error: " + t.getMessage()12}13WebDriverDecorator decorator = new WebDriverDecorator();14decorator.onError = { Throwable t -> 15    println "Error: " + t.getMessage()16}17WebDriverDecorator decorator = new WebDriverDecorator();18decorator.onError = { Throwable t -> 19    println "Error: " + t.getMessage()20}21WebDriverDecorator decorator = new WebDriverDecorator();22decorator.onError = { Throwable t -> 23    println "Error: " + t.getMessage()24}25WebDriverDecorator decorator = new WebDriverDecorator();26decorator.onError = { Throwable t -> 27    println "Error: " + t.getMessage()28}29WebDriverDecorator decorator = new WebDriverDecorator();30decorator.onError = { Throwable t -> 31    println "Error: " + t.getMessage()32}33WebDriverDecorator decorator = new WebDriverDecorator();34decorator.onError = { Throwable t -> 35    println "Error: " + t.getMessage()36}37WebDriverDecorator decorator = new WebDriverDecorator();38decorator.onError = { Throwable t -> 39    println "Error: " + t.getMessage()40}41WebDriverDecorator decorator = new WebDriverDecorator();42decorator.onError = { Throwable t -> 43    println "Error: " + t.getMessage()44}45WebDriverDecorator decorator = new WebDriverDecorator();onError
Using AI Code Generation
1import org.openqa.selenium.support.decorators.DefaultDriverDecorator;2import org.openqa.selenium.support.decorators.WebDriverDecorator;3import org.openqa.selenium.support.decorators.WebDriverDecorators;4import org.openqa.selenium.support.decorators.WebDriverHandler;5import org.openqa.selenium.support.decorators.handler.DefaultElementHandler;6import org.openqa.selenium.support.decorators.handler.DefaultErrorHandler;7import org.openqa.selenium.support.decorators.handler.DefaultLocatableHandler;8import org.openqa.selenium.support.decorators.handler.DefaultSearchContextHandler;9import org.openqa.selenium.support.decorators.handler.DefaultWebDriverHandler;10import org.openqa.selenium.support.decorators.handler.DefaultWrapsElementHandler;11import org.openqa.selenium.support.decorators.handler.DefaultWrapsHandler;12import org.openqa.selenium.support.decorators.handler.DefaultWrapsLocatableHandler;13import org.openqa.selenium.support.decorators.handler.DefaultWrapsSearchContextHandler;14import org.openqa.selenium.support.decorators.handler.DefaultWrapsWebDriverHandler;15import java.lang.reflect.Method;16import java.util.ArrayList;17import java.util.Arrays;18import java.util.List;19import static org.junit.Assert.assertEquals;20import static org.junit.Assert.fail;21import static org.mockito.Mockito.mock;22import static org.mockito.Mockito.when;23import org.junit.Test;24import org.openqa.selenium.By;25import org.openqa.selenium.NoSuchElementException;26import org.openqa.selenium.StaleElementReferenceException;27import org.openqa.selenium.WebDriver;28import org.openqa.selenium.WebElement;29import org.openqa.selenium.support.pagefactory.ElementLocator;30import org.openqa.selenium.support.pagefactory.ElementLocatorFactory;31import org.openqa.selenium.support.pagefactory.FieldDecorator;32import org.openqa.selenium.support.pagefactory.internal.LocatingElementHandler;33import org.openqa.selenium.support.ui.ExpectedConditions;34import org.openqa.selenium.support.ui.WebDriverWait;35public class DecoratorTest {36    public void testDecorator() {37        WebDriver driver = mock(WebDriver.class);38        WebElement element = mock(WebElement.class);39        when(driver.findElement(By.id("someId"))).thenReturn(element);40        WebDriverDecorator decorator = new DefaultDriverDecorator();41        WebDriver decoratedDriver = decorator.decorate(driver);42        WebElement decoratedElement = decoratedDriver.findElement(By.id("someId"));43        decoratedElement.click();44    }45    public void testLocatingElementHandler() throws Exception {46        WebDriver driver = mock(WebDriver.class);47        WebElement element = mock(WebElement.class);48        ElementLocator locator = mock(ElementLocator.class);49        when(locator.findElement()).thenReturn(element);50        ElementLocatorFactory factory = mock(ElementLocatorFactory.class);51        when(factory.createLocator(By.id("someId"))).thenReturn(locator);onError
Using AI Code Generation
1import org.openqa.selenium.support.decorators.DefaultDecorated;2import org.openqa.selenium.support.decorators.Decorated;3import org.openqa.selenium.support.decorators.Decorator;4import org.openqa.selenium.support.decorators.WithTimeout;5import org.openqa.selenium.support.decorators.WithTimeoutProcessor;6public class CustomException extends RuntimeException {7    public CustomException(String message, Throwable cause) {8        super(message, cause);9    }10}11public class CustomDecorator implements Decorator {onError
Using AI Code Generation
1import org.openqa.selenium.NoSuchElementException;2import org.openqa.selenium.support.decorators.DefaultElementLocatorFactory;3import org.openqa.selenium.support.decorators.DefaultErrorHandler;4import org.openqa.selenium.support.decorators.DefaultWebDriverDecorator;5import org.openqa.selenium.support.pagefactory.DefaultElementLocator;6import org.openqa.selenium.support.pagefactory.DefaultElementLocatorFactory;7import org.openqa.selenium.support.pagefactory.DefaultFieldDecorator;8import org.openqa.selenium.support.pagefactory.ElementLocator;9import org.openqa.selenium.support.pagefactory.ElementLocatorFactory;10import org.openqa.selenium.support.pagefactory.FieldDecorator;11import java.lang.reflect.Field;12import java.lang.reflect.InvocationHandler;13import java.lang.reflect.Method;14import java.lang.reflect.Proxy;15import java.util.List;16import org.openqa.selenium.By;17import org.openqa.selenium.NoSuchElementException;18import org.openqa.selenium.SearchContext;19import org.openqa.selenium.WebDriver;20import org.openqa.selenium.WebElement;21public class DefaultDecorator extends DefaultWebDriverDecorator {22    public DefaultDecorator(WebDriver driver) {23        super(driver);24    }25    protected ElementLocatorFactory createElementLocatorFactory(SearchContext context) {26        return new DefaultElementLocatorFactory(context) {27            public ElementLocator createLocator(Field field) {28                return new DefaultElementLocator(this, field) {29                    public WebElement findElement() {30                        try {31                            return super.findElement();32                        } catch (NoSuchElementException e) {33                            return null;34                        }35                    }36                    public List<WebElement> findElements() {37                        try {38                            return super.findElements();39                        } catch (NoSuchElementException e) {40                            return null;41                        }42                    }43                };44            }45        };46    }47}48import org.openqa.selenium.NoSuchElementException;49import org.openqa.selenium.support.decorators.DefaultElementLocatorFactory;50import org.openqa.selenium.support.decorators.DefaultErrorHandler;51import orgLambdaTest’s Selenium 4 tutorial is covering every aspects of Selenium 4 testing with examples and best practices. Here you will learn basics, such as how to upgrade from Selenium 3 to Selenium 4, to some advanced concepts, such as Relative locators and Selenium Grid 4 for Distributed testing. Also will learn new features of Selenium 4, such as capturing screenshots of specific elements, opening a new tab or window on the browser, and new protocol adoptions.
Upgrading From Selenium 3 To Selenium 4?: In this chapter, learn in detail how to update Selenium 3 to Selenium 4 for Java binding. Also, learn how to upgrade while using different build tools such as Maven or Gradle and get comprehensive guidance for upgrading Selenium.
What’s New In Selenium 4 & What’s Being Deprecated? : Get all information about new implementations in Selenium 4, such as W3S protocol adaption, Optimized Selenium Grid, and Enhanced Selenium IDE. Also, learn what is deprecated for Selenium 4, such as DesiredCapabilites and FindsBy methods, etc.
Selenium 4 With Python: Selenium supports all major languages, such as Python, C#, Ruby, and JavaScript. In this chapter, learn how to install Selenium 4 for Python and the features of Python in Selenium 4, such as Relative locators, Browser manipulation, and Chrom DevTool protocol.
Selenium 4 Is Now W3C Compliant: JSON Wireframe protocol is retiring from Selenium 4, and they are adopting W3C protocol to learn in detail about the advantages and impact of these changes.
How To Use Selenium 4 Relative Locator? : Selenium 4 came with new features such as Relative Locators that allow constructing locators with reference and easily located constructors nearby. Get to know its different use cases with examples.
Selenium Grid 4 Tutorial For Distributed Testing: Selenium Grid 4 allows you to perform tests over different browsers, OS, and device combinations. It also enables parallel execution browser testing, reads up on various features of Selenium Grid 4 and how to download it, and runs a test on Selenium Grid 4 with best practices.
Selenium Video Tutorials: Binge on video tutorials on Selenium by industry experts to get step-by-step direction from automating basic to complex test scenarios with Selenium.
LambdaTest also provides certification for Selenium testing to accelerate your career in Selenium automation testing.
Get 100 minutes of automation test minutes FREE!!
