/** * Unwraps the proxy */ @Override public WebDriver deenhance(WebDriver enhancedDriver, Class<? extends Annotation> qualifier) { if (enhancedDriver instanceof GrapheneProxyInstance) { WebDriver driver = ((GrapheneProxyInstance) enhancedDriver).unwrap(); GrapheneContext.removeContextFor(qualifier); return driver; } return enhancedDriver; }
/** * Unwraps the proxy */ @Override public WebDriver deenhance(WebDriver enhancedDriver, Class<? extends Annotation> qualifier) { if (enhancedDriver instanceof GrapheneProxyInstance) { WebDriver driver = ((GrapheneProxyInstance) enhancedDriver).unwrap(); GrapheneContext.removeContextFor(qualifier); return driver; } return enhancedDriver; }
@After public void tearDown() { GrapheneRuntime.popInstance(); GrapheneContext.removeContextFor(Default.class); } }
@After public void tearDown() { GrapheneRuntime.popInstance(); GrapheneContext.removeContextFor(Default.class); }
@After public void tearDown() { GrapheneRuntime.popInstance(); GrapheneContext.removeContextFor(Default.class); }
@Test public void generated_webelement_implements_WrapsElement_interface() { TestPage page = new TestPage(); getGrapheneEnricher().enrich(page); assertTrue(page.element instanceof WrapsElement); GrapheneContext.setContextFor(new GrapheneConfiguration(), driver, Default.class); when(driver.findElement(Mockito.any(By.class))).thenReturn(element); WebElement wrappedElement = ((WrapsElement) page.element).getWrappedElement(); GrapheneContext.removeContextFor(Default.class); assertThat(wrappedElement, is(instanceOf(GrapheneElement.class))); assertEquals(element, ((GrapheneElement)wrappedElement).getWrappedElement()); }