org.openqa.selenium.Alert Java Examples

/**
 * Accepts or declines prompt with sending message
 *
 * @param alertAction accept or decline
 * @param text        message to send
 */
public void handlePromptAlert(AlertActions alertAction, String text) {
    try {
        Alert alert = getDriver().switchTo().alert();
        if (text != null && !text.isEmpty()) {
            getDriver().switchTo().alert().sendKeys(text);
        }
        if (alertAction.equals(AlertActions.ACCEPT)) {
            alert.accept();
        } else {
            alert.dismiss();
        }
    } catch (NoAlertPresentException exception) {
        localizedLogger.fatal("loc.browser.alert.fail", exception);
        throw exception;
    }
}
 

/**
 * 切换到alter，并点击确认框
 */
public void clickAlert(String msg) {
	WebDriverWait webDriverWait = new WebDriverWait(driver, 10);
	Alert alert;
	baseOpt.wait(20);
	// Alert alert = driver.switchTo().alert();

	try { // 不稳定，原因待查。还是用上面的吧
	// wait and switchTo, Otherwise, throws a TimeoutException
		alert = webDriverWait.until(ExpectedConditions.alertIsPresent());
	} catch (Exception e) {
		this.screenShot();
		LogUtil.info(driver.manage().logs() + "==>alert等待超时!");
		alert = driver.switchTo().alert(); // 与waituntil功能重复，但until经常失败，为了增强健壮性才如此写
	}

	if (msg != null) {
		AssertUtil.assertEquals(alert.getText(), msg, "提示语错误");
	}

	alert.accept();
	baseOpt.wait(30);
}
 

public String takeScreenshot(final String errorMessage, final String testName) {
    try {
        BufferedImage image;
        if (Configuration.browser.equals("chrome")) {
            image = ImageIO.read(new ChromeScreenshoter().getFullScreenshotAs(OutputType.FILE));
        } else {
            image = ImageIO.read(((TakesScreenshot)DriverHolder.getDriver()).getScreenshotAs(OutputType.FILE));
        }

        printStrings(image, removeNL(testName, errorMessage));

        final String pathName = getFilenameFor(testName);
        final File screenShotWithProjectPath = new File(pathName);
        ImageIO.write(image, "png", screenShotWithProjectPath);
        attachScreenShotToAllure(errorMessage, testName, screenShotWithProjectPath);
        return screenShotWithProjectPath.getAbsolutePath();
    } catch (IOException e) {
        e.printStackTrace();
        return "";
    } catch (UnhandledAlertException alertException) {
        Alert alert = DriverHolder.getDriver().switchTo().alert();
        alert.dismiss();
        return takeScreenshot(errorMessage, testName);
    }
}
 

/**
 * 验证alert对话框提示信息是否与预期值一致
 * @param expectAlertText alert 提示框预期信息
 * @param Message 验证中文描述
 * @author Administrator 郑树恒
 */
public static void VerityAlertText(String expectAlertText,String Message)
{
	Alert alert=driver.switchTo().alert();
	String  alertText=alert.getText();
	String verityStr="【Assert验证】:弹出的对话框的文本内容是否一致{"+"实际值："+alertText+","+"预期值："+expectAlertText+"}";
	log.info(Message+":"+verityStr);
	try {
		Assert.assertEquals(alertText, expectAlertText);
		AssertPassLog();

		assertInfolList.add(Message+verityStr+":pass");
		messageList.add(Message+":pass");
	} catch (Error e) {
		// TODO: handle exception
		AssertFailedLog();
		errors.add(e);
		errorIndex++;

		assertInfolList.add(Message+verityStr+":failed");
		messageList.add(Message+":failed");
		Assertion.snapshotInfo();
		//throw e;
	}

}
 

public static ExpectedCondition<Alert> alertIsPresent() {
  return new ExpectedCondition<Alert>() {
    @Override
    public Alert apply(WebDriver driver) {
      try {
        return driver.switchTo().alert();
      } catch (NoAlertPresentException e) {
        return null;
      }
    }

    @Override
    public String toString() {
      return "alert to be present";
    }
  };
}
 

/**
 * 验证alert对话框提示信息是否与预期值一致
 * @param expectAlertText alert 提示框预期信息
 * @author Administrator 郑树恒
 */
public static void VerityAlertText(String expectAlertText)
{
	Alert alert=driver.switchTo().alert();
	String  alertText=alert.getText();
	String verityStr="【Assert验证】:弹出的对话框的文本内容是否一致{"+alertText+","+expectAlertText+"}";
	log.info("【Assert验证】:弹出的对话框的文本内容是否一致{"+"实际值："+alertText+","+"预期值"+expectAlertText+"}");
	try {
		Assert.assertEquals(alertText, expectAlertText);
		AssertPassLog();
		assertInfolList.add(verityStr+":pass");
	} catch (Error e) {
		// TODO: handle exception
		AssertFailedLog();
		errors.add(e);
		errorIndex++;
		assertInfolList.add(verityStr+":failed");
		Assertion.snapshotInfo();
		//throw e;
	}

}
 

public static String alertOperation(IOSDriver<IOSElement> appium, String operation) {
	String result = "";
	Alert alert = appium.switchTo().alert();
	switch (operation) {
	case "alertaccept":
		alert.accept();
		LogUtil.APP.info("�����������ͬ��...");
		break;
	case "alertdismiss":
		alert.dismiss();
		LogUtil.APP.info("�����������ȡ��...");
		break;
	case "alertgettext":
		result = "��ȡ����ֵ�ǡ�" + alert.getText() + "��";
		LogUtil.APP.info("���������ͨ��getText��ȡ����text����...��Text����ֵ:{}��",alert.getText());
		break;
	default:
		break;
	}
	return result;
}
 

public static String alertOperation(WebDriver wd, String operation) {
    String result = "";
    Alert alert = wd.switchTo().alert();
    switch (operation) {
        case "alertaccept":
            alert.accept();
            result = "�����������ͬ��...";
            LogUtil.APP.info(result);
            break;
        case "alertdismiss":
            alert.dismiss();
            result = "�����������ȡ��...";
            LogUtil.APP.info(result);
            break;
        case "alertgettext":
            result = "��ȡ����ֵ�ǡ�" + alert.getText() + "��";
            LogUtil.APP.info("���������ͨ��getText��ȡ����text����...��Text����ֵ:{}��",alert.getText());
            break;
        default:
            break;
    }
    return result;
}
 

@Test public void listenableObjectSample() {
    try {
        ContextAware listenableContextAware =
                getEventFiringObject(contextAware, emptyWebDriver, contextListener, alertListener);
        WebDriver webDriver = listenableContextAware.context("NATIVE_APP");
        assertTrue(contextAwarePredicate.test(listenableContextAware));

        Alert alert = webDriver.switchTo().alert();
        assertTrue(alertPredicate.test(alert));

        assertTrue(webDriverPredicate.test(getEventFiringWebDriver(webDriver, searchingListener)));
    } finally {
        listeners.get(ContextListener.class).messages.clear();
        listeners.get(AlertListener.class).messages.clear();
        listeners.get(SearchingListener.class).messages.clear();
    }
}
 

@PublicAtsApi
public boolean isElementPresent() {

    // with the current Selenium implementation we do not know whether the opened modal dialog
    // is alert, prompt or confirmation
    if (element instanceof UiAlert) {

        return getAlert() != null;
    } else if (element instanceof UiPrompt) {

        Alert prompt = getAlert();
        return prompt != null && prompt.getText() != null;
    } else if (element instanceof UiConfirm) {

        Alert confirm = getAlert();
        return confirm != null && confirm.getText() != null;
    }

    HtmlNavigator.getInstance().navigateToFrame(webDriver, element);

    return RealHtmlElementLocator.findElements(element).size() > 0;
}
 

@Test
public void shouldKeepTrackOfPreviousWebDriverTargetIfAlertIsCurrentTarget() {
    WebDriver mockDriver = mock(WebDriver.class);
    TargetLocator mockLocator = mock(TargetLocator.class);

    when(mockDriver.switchTo()).thenReturn(mockLocator);
    when(mockLocator.alert()).thenReturn(mock(Alert.class));

    CachingTargetLocator targetLocator = new CachingTargetLocator(
            WebDriverTargets.window("test"), mockDriver);
    targetLocator.alert();
    targetLocator.frame("frame");

    assertEquals(WebDriverTargets.frame(WebDriverTargets.window("test"), "frame"),
            targetLocator.getCurrentTarget());
}
 

@Test
public void alertEventTest() {
    try {
        Alert alert = driver.switchTo().alert();
        alert.accept();
        alert.dismiss();
        alert.sendKeys("Keys");

        assertThat(listener.messages,
                hasItems(WEBDRIVER_EVENT_LISTENER + "Attempt to accept alert",
                        WEBDRIVER_EVENT_LISTENER + "The alert was accepted",
                        WEBDRIVER_EVENT_LISTENER + "Attempt to dismiss alert",
                        WEBDRIVER_EVENT_LISTENER + "The alert was dismissed"));
    } finally {
        listener.messages.clear();
    }
}
 

/**
 * 获取对话框文本
 * @return 返回String
 */
public String getAlertText()
{
	Alert alert=driver.switchTo().alert();
	try {
		String text=alert.getText().toString();
		log.info("获取对话框文本："+text);
		return text;
	} catch (NoAlertPresentException notFindAlert) {
		// TODO: handle exception
		log.error("找不到对话框");
		//return "找不到对话框";
		throw notFindAlert;

	}
}
 

@Override
public Alert alert() {
    if (alert == null) {
        // Cache the alert to avoid repeat switches.
        // If alert is switched _from_, it must be nulled out.
        alert = driver.switchTo().alert();
    }

    return alert;
}
 

/**
 * Accept the javascript alert (clicking OK)
 *
*/
protected void acceptAlert()
{
    Alert alert = driver.switchTo().alert();
    //update is executed
    alert.accept();
}
 

@Override
public void promptInputPressOK(String inputMessage) {
	Alert alert = waitForAlert();
	alert.sendKeys(inputMessage);
	alert.accept();

}
 

public void AcceptPrompt(String text) {
	
	if (!isAlertPresent())
		return;
	
	Alert alert = getAlert();
	alert.sendKeys(text);
	alert.accept();
	oLog.info(text);
}
 

/**
 * Dismiss the javascript alert (clicking Cancel)
 *
*/
protected void dismissAlert()
{
    Alert alert = driver.switchTo().alert();
    //update is executed
    alert.dismiss();
}
 

@Test(groups = "wso2.cep", description = "Verifying XSS Vulnerability in event data sources - description field")
public void testXSSVenerabilityDescriptionField() throws Exception {
    boolean isVulnerable = false;

    // Login
    driver.get(getLoginURL());
    driver.findElement(By.id("txtUserName")).clear();
    driver.findElement(By.id("txtUserName")).sendKeys(cepServer.getContextTenant().getContextUser().getUserName());
    driver.findElement(By.id("txtPassword")).clear();
    driver.findElement(By.id("txtPassword")).sendKeys(cepServer.getContextTenant().getContextUser().getPassword());
    driver.findElement(By.cssSelector("input.button")).click();

    // Sending request to even-tracer admin service
    String url = backendURL.substring(0, 22) + "/carbon/ndatasource/newdatasource.jsp?";
    List<NameValuePair> params = new ArrayList<>();
    params.add(new BasicNameValuePair("description", "RiskScoringDB\"><script>alert(1)</script><example attr=\""));
    params.add(new BasicNameValuePair("edit", "true"));
    url += URLEncodedUtils.format(params, "UTF-8");

    driver.get(url);
    try {
        // Alert appears if vulnerable to XSS attack.
        Alert alert = driver.switchTo().alert();
        alert.accept();
        isVulnerable = true;
    } catch (NoAlertPresentException e) {
        // XSS vulnerability is not there
    }
    Assert.assertFalse(isVulnerable);
    driver.close();
}
 

/**
 * check state pop up browser window
 *
 * @return
 */
public boolean isAlertPresent() {
  try {
    Alert alert = seleniumWebDriver.switchTo().alert();
    return !alert.getText().isEmpty();
  } catch (Exception e) {
    return false;
  }
}
 

@Override
public void processAlert(Action action)
{
    Alert alert = switchToAlert(webDriverProvider.get());
    if (alert != null)
    {
        action.process(alert, webDriverManager);
    }
}
 

private String closeAlertAndGetItsText() {
    try {
        Alert alert = driver.switchTo().alert();
        String alertText = alert.getText();
        if (acceptNextAlert) {
            alert.accept();
        } else {
            alert.dismiss();
        }
        return alertText;
    } finally {
        acceptNextAlert = true;
    }
}
 

@Test
public void shouldNotSwitchDriverToAlertIfCurrentTargetIsAlert() {
    WebDriver mockDriver = mock(WebDriver.class);
    TargetLocator mockLocator = mock(TargetLocator.class);

    when(mockDriver.switchTo()).thenReturn(mockLocator);
    when(mockLocator.alert()).thenReturn(mock(Alert.class));

    CachingTargetLocator targetLocator = new CachingTargetLocator(
            WebDriverTargets.window("test"), mockDriver);
    targetLocator.alert();
    targetLocator.alert();

    verify(mockLocator, times(1)).alert();
}
 

@WaitUntil
public boolean dismissAlert() {
    Alert alert = getAlert();
    boolean result = false;
    if (alert != null) {
        alert.dismiss();
        onAlertHandled(false);
        result = true;
    }
    return result;
}
 

@Test
public void alertEvents() {
  final WebDriver mockedDriver = mock(WebDriver.class);
  final Alert mockedAlert = mock(Alert.class);
  final WebDriver.TargetLocator mockedTargetLocator = mock(WebDriver.TargetLocator.class);

  when(mockedDriver.switchTo()).thenReturn(mockedTargetLocator);
  when(mockedTargetLocator.alert()).thenReturn(mockedAlert);

  WebDriverEventListener listener = mock(WebDriverEventListener.class);

  EventFiringWebDriver testedDriver = new EventFiringWebDriver(mockedDriver).register(listener);

  testedDriver.switchTo().alert().accept();
  testedDriver.switchTo().alert().dismiss();

  InOrder order = Mockito.inOrder(mockedDriver, mockedAlert, listener);
  order.verify(mockedDriver).switchTo();
  order.verify(listener).beforeAlertAccept(any(WebDriver.class));
  order.verify(mockedAlert).accept();
  order.verify(listener).afterAlertAccept(any(WebDriver.class));
  order.verify(mockedDriver).switchTo();
  order.verify(listener).beforeAlertDismiss(any(WebDriver.class));
  order.verify(mockedAlert).dismiss();
  order.verify(listener).afterAlertDismiss(any(WebDriver.class));
  verifyNoMoreInteractions(mockedDriver, mockedAlert, listener);
}
 

private String closeAlertAndGetItsText() {
    try {
        Alert alert = driver.switchTo().alert();
        String alertText = alert.getText();
        if (acceptNextAlert) {
            alert.accept();
        } else {
            alert.dismiss();
        }
        return alertText;
    } finally {
        acceptNextAlert = true;
    }
}
 

/**
 * Accept any displayed alert message. If no alert is displayed, an exception will be thrown.
 * <p>
 * Fires {@link AcceptedAlertEvent} in case a alert was successfully accepted.
 *
 * @throws NoAlertPresentException in case no alert is present
 * @see Alert#accept()
 * @since 2.0
 */
public void accept() throws NoAlertPresentException {
    StringBuilder builder = new StringBuilder();
    ActionTemplate.browser(browser()).execute(browser -> {
        Alert alert = webDriver().switchTo().alert();
        builder.append(alert.getText());
        alert.accept();
    }).fireEvent(browser -> new AcceptedAlertEvent(builder.toString()));
    log.debug("alert was accepted");
}
 

@Override
public void promptInputPressCancel(String inputMessage) {
	Alert alert = waitForAlert();
	alert.sendKeys(inputMessage);
	alert.dismiss();

}
 

private String closeAlertAndGetItsText() {
    try {
        Alert alert = driver.switchTo().alert();
        String alertText = alert.getText();
        if (acceptNextAlert) {
            alert.accept();
        } else {
            alert.dismiss();
        }
        return alertText;
    } finally {
        acceptNextAlert = true;
    }
}
 

private String closeAlertAndGetItsText() {
    try {
        Alert alert = driver.switchTo().alert();
        String alertText = alert.getText();
        if (acceptNextAlert) {
            alert.accept();
        } else {
            alert.dismiss();
        }
        return alertText;
    } finally {
        acceptNextAlert = true;
    }
}