holmos.webtest.WebDriverBrowserWindow Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of holmos-webtest Show documentation
There is a newer version: 1.0.2u10
package holmos.webtest;

import holmos.webtest.asynchronous.AsynchronousOpen;
import holmos.webtest.basetools.HolmosBaseTools;
import holmos.webtest.constvalue.ConfigConstValue;

import java.awt.Toolkit;
import java.io.File;
import java.io.IOException;
import java.util.Set;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.Alert;
import org.openqa.selenium.Cookie;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.Point;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
/**
 * @author 吴银龙([email protected])
 * */
public class WebDriverBrowserWindow implements BrowserWindow{
	static {
		HolmosBaseTools.configLogProperties();
	}
	/**当前窗口的句柄，此句柄由底层webdriver引擎获取*/
	public WebDriverBrowserWindow(SeleniumDriver driver){
		this.driver= driver;
		this.enginetype=driver.getType();
	}
	private String windowHandle;
	private EngineType enginetype;
	/**获得webdriver页面的句柄*/
	public String getHandle() {
		return windowHandle;
	}

	public void setWidowHandle(String widowHandle) {
		this.windowHandle = widowHandle;
	}

	public void setDriver(SeleniumDriver driver) {
		this.driver = driver;
	}

	/**底层驱动引擎，一定是webdriver类型*/
	private SeleniumDriver driver;
	
	public SeleniumDriver getDriver() {
		return this.driver;
	}

	public String getUrl() {
		focus();
		return ((WebDriver)driver.getEngine()).getCurrentUrl();
	}

	public void refresh() {
		focus();
		((WebDriver)driver.getEngine()).navigate().refresh();
	}

	public void close() {
		focus();
		((WebDriver)driver.getEngine()).close();
		Allocator.getInstance().getWindows().remove(this);
		HolmosBaseTools.killProcess("chromedriver.exe");
		HolmosBaseTools.killProcess("IEDriverServer.exe");
	}
	
	public void goForward() {
		focus();
		((WebDriver)driver.getEngine()).navigate().forward();
	}

	public void goBack() {
		focus();
		((WebDriver)driver.getEngine()).navigate().back();
	}

	public EngineType getEngineType() {
		return this.enginetype;
	}

	public void open(String url) {
		focus();
		AsynchronousOpen getThread=new AsynchronousOpen(this, url);
		getThread.start();
		int count=0;
		while(count++ConfigConstValue.defaultWaitCount){
			System.out.println("页面加载超时了!但是case继续执行了，不影响~");
		}
	}

	public void focus() {
		if(Allocator.getInstance().currentWindow.equals(this))
			return;
		((WebDriver)driver.getEngine()).switchTo().window(getHandle()); 
		Allocator.getInstance().currentWindow=this;
	}

	public void maxSizeWindow() {
		int width=Toolkit.getDefaultToolkit().getScreenSize().width;
		int height=Toolkit.getDefaultToolkit().getScreenSize().height;
		focus();
		moveWindowTo(0, 0);
		resizeTo(width, height);
	}

	public void moveWindowTo(int xLocation, int yLocation) {
		focus();
		StringBuilder message=new StringBuilder("移动窗口位置");
		if(xLocation>=0&&yLocation>=0){
			message.append("x:"+xLocation+"  y:"+yLocation);
			((WebDriver)driver.getEngine()).manage().window().setPosition(new Point(xLocation, yLocation));
			logger.info(message);
		}else{
			message.append("窗口位置设置错误！");
			logger.error(message);
		}
	}

	public void resizeTo(int horizonSize, int verticalSize) {
		focus();
		StringBuilder message=new StringBuilder("窗口重新设置大小");
		if(horizonSize>=0&&verticalSize>=0){
			message.append("宽度:"+horizonSize+"  高度:"+verticalSize);
			((WebDriver)driver.getEngine()).manage().window().setSize(new Dimension(horizonSize, verticalSize));
			logger.info(message);
		}else{
			message.append("窗口大小设置错误！");
			logger.error(message);
		}
	}

	public void start() {
//		((WebDriver)getDriver().getEngine()).get(getUrl())
	}
	
	public String dealAlert() {
		String alertMessage=null;
		focus();
		try{
			Alert alert=((WebDriver)getDriver().getEngine()).switchTo().alert();
			alertMessage= alert.getText();
			alert.accept();
			return alertMessage;
		}catch (Exception e) {
			logger.error("没有找到Alert窗口!");
		}
		return null;
	}

	public String dealPrompt(String input,boolean isYes) {
		String promptMessage=null;
		focus();
		try{
			Alert alert=((WebDriver)getDriver().getEngine()).switchTo().alert();
			alert.sendKeys(input);
			promptMessage= alert.getText();
			if(isYes)
				alert.accept();
			else 
				alert.dismiss();
			return promptMessage;
		}catch (Exception e) {
			logger.error("没有找到Prompt窗口!");
		}return null;
	}

	public String dealConfirm(boolean isYes) {
		String confirmMessage=null;
		focus();
		try{
			Alert alert=((WebDriver)getDriver().getEngine()).switchTo().alert();
			confirmMessage=alert.getText();
			if(isYes)alert.accept();
			else alert.dismiss();
			return confirmMessage;
		}catch (Exception e) {
			logger.error("没有找到Prompt窗口!");
		}return null;
	}

	@Override
	public void openNewWindow(String url) {
		open(url);
		this.windowHandle=((WebDriver)driver.getEngine()).getWindowHandle();
	}
	
	@Override
	public void quit() {
		((WebDriver)driver.getEngine()).quit();
	}
	/**
	 * 截图位置放到了
	 * */
	public void TakeScreenshot(String fileName){
		TakesScreenshot TCS=(TakesScreenshot) ((WebDriver)driver.getEngine());
		File screenShot=TCS.getScreenshotAs(OutputType.FILE);
		try {
			FileUtils.copyFile(screenShot, new File(Allocator.getInstance().getScreenShotDir()+fileName));
			logger.info("截图成功!");
		} catch (IOException e) {
			logger.error("截图失败!");
			e.printStackTrace();
		}
	}

	@Override
	public void addCookie(Cookie cookie) {
		focus();
		((WebDriver)driver.getEngine()).manage().addCookie(cookie);
	}

	@Override
	public Set getAllCookies() {
		return ((WebDriver)driver.getEngine()).manage().getCookies();
	}

	@Override
	public Cookie getCookieByName(String name) {
		return ((WebDriver)driver.getEngine()).manage().getCookieNamed(name);
	}

	@Override
	public void removeCookieByName(String name) {
		((WebDriver)driver.getEngine()).manage().deleteCookieNamed(name);
	}

	@Override
	public void removeAllCookies() {
		((WebDriver)driver.getEngine()).manage().deleteAllCookies();
	}
	public void removeCookie(Cookie cookie){
		((WebDriver)driver.getEngine()).manage().deleteCookie(cookie);
	}

	@Override
	public String getTitle() {
		focus();
		return ((WebDriver)driver.getEngine()).getTitle();
	}

	@Override
	public void setHandle(String handle) {
		this.windowHandle=handle;
	}
}