cz.etnetera.seb.SebAlert Maven / Gradle / Ivy

Go to download
/* Copyright 2016 Etnetera a.s.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package cz.etnetera.seb;

import java.util.function.Consumer;

import org.openqa.selenium.JavascriptExecutor;

/**
 * Alert and confirm dialog set of methods which
 * support both browser with native dialogs and headless
 * browsers without this capability.
 */
public class SebAlert {

	protected SebContext context;
	
	public SebAlert(SebContext context) {
		this.context = context;
	}

	public void acceptAlertDialog(Consumer showDialog) {
		boolean supportsAlerts = isAlertSupported();
		if (!supportsAlerts) {
			((JavascriptExecutor) context.getDriver()).executeScript("window.alert = function(msg){};");
		}
		showDialog.accept(context);
		if (supportsAlerts) {
			context.getDriver().switchTo().alert().accept();
		}
	}
	
	public void acceptConfirmDialog(Consumer showDialog) {
		boolean supportsAlerts = isAlertSupported();
		if (!supportsAlerts) {
			((JavascriptExecutor) context.getDriver()).executeScript("window.confirm = function(msg){return true;};");
		}
		showDialog.accept(context);
		if (supportsAlerts) {
			context.getDriver().switchTo().alert().accept();
		}
	}

	public void dismissConfirmDialog(Consumer showDialog) {
		boolean supportsAlerts = isAlertSupported();
		if (!supportsAlerts) {
			((JavascriptExecutor) context.getDriver()).executeScript("window.confirm = function(msg){return false;};");
		}
		showDialog.accept(context);
		if (supportsAlerts) {
			context.getDriver().switchTo().alert().dismiss();
		}
	}
	
	public boolean isAlertSupported() {
		return context.getSeb().isAlertSupported();
	}

}