Selenium - WebDriver 小结(1)

public class Base {
SimpleDateFormat formatterTime = new SimpleDateFormat("yyyyMMdd_hhmmssa");
    SimpleDateFormat formatterDate = new SimpleDateFormat("yyyyMMdd");

    private Calendar now() {
        Calendar now = Calendar.getInstance();
        return now;
    }

public static Map<String, String> initialProperties(String properityName) {
        String key = "";
        String value = "";
        Map<String, String> map = null;
        ResourceBundle bound = PropertyResourceBundle.getBundle(properityName, Locale.ENGLISH);
        Enumeration<String> enumeration = bound.getKeys();
        if (enumeration != null) {
            map = new HashMap<String, String>();
            while (enumeration.hasMoreElements()) {
                key = (String) enumeration.nextElement();
                value = bound.getString(key);
                map.put(key, value);
            }
        }
        return map;
    }

    public static WebDriver setUp_IEDriver() throws Exception {
        System.setProperty("webdriver.ie.driver", "./IEDriverServer.exe");
        DesiredCapabilities capabilities = DesiredCapabilities.internetExplorer();
        capabilities.setCapability(InternetExplorerDriver.INTRODUCE_FLAKINESS_BY_IGNORING_SECURITY_DOMAINS, true);
        WebDriver driver = new InternetExplorerDriver(capabilities);
        driver.manage().window().maximize();
        return driver;
    }

public WebDriver setUp_ChromeDriver(){
        System.setProperty("webdriver.chrome.driver", "./chromedriver.exe");
        WebDriver driver = new ChromeDriver();
        driver.manage().window().maximize();
        return driver;
    }
    
    public WebDriver setUp_FirefoxDriver(){
        System.setProperty("webdriver.ie.driver", "C:/Program Files/Mozilla Firefox");
        WebDriver driver = new FirefoxDriver();
        return driver;
    }

public static boolean isElementPresent(WebDriver driver, By by) {
        try {
            driver.findElement(by);
            return true;
        } catch (NoSuchElementException e) {
            e.printStackTrace();
            return false;
        }
    }

public void screenshot(WebDriver driver, String path){        
        File folder = new File(path);
        try {
            if(!folder.exists()){
                folder.mkdirs();
            }
            Dimension window_size = driver.manage().window().getSize();            
//            BufferedImage image = new Robot().createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));
            BufferedImage image = new Robot().createScreenCapture(new Rectangle(window_size.getWidth(), window_size.getHeight()));
            String screenshotName = path+formatterTime.format(now().getTime())+".png";
            ImageIO.write(image, "png", new File(screenshotName));
            System.out.println("Screenshot "+screenshotName+" has been saved to " + path);
        } catch (HeadlessException e) {
            e.printStackTrace();
        } catch (AWTException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }        
    }
    
    public void pageshot(WebDriver driver, String path){        
        File folder = new File(path);
        try {
            if(!folder.exists()){
                folder.mkdirs();
            }
            String screenshotName = path+formatterTime.format(now().getTime())+".png";
            File screenshot = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
            FileUtils.copyFile(screenshot, new File(screenshotName));            
            System.out.println("Screenshot "+screenshotName+" has been saved to " + path);
        } catch (HeadlessException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } 
    }

public boolean windowStatus(WebDriver driver,  final String window_status){
        WebDriverWait wait = new WebDriverWait(driver, 120);
        boolean windowStatus = wait.until(new ExpectedCondition<Boolean>(){
            public Boolean apply(WebDriver driver){
            JavascriptExecutor js = (JavascriptExecutor) driver;
            return js.executeScript("return window.status").toString().trim().equals(window_status);
        }});
        return windowStatus;
    }
    
    public void waitForElementLoadByXpath(WebDriver driver, final String xpath){
        WebDriverWait wait = new WebDriverWait(driver, 300);
        wait.until(new ExpectedCondition<WebElement>(){
            public WebElement apply(WebDriver driver) { 
                WebElement element = driver.findElement(By.xpath(xpath));
                return element;
            }
        });    
    }    
    
    public void waitForPageLoad(WebDriver driver){
        WebDriverWait wait = new WebDriverWait(driver, 300);
        wait.until(new ExpectedCondition<Boolean>(){
            public Boolean apply(WebDriver driver) {
                  return ((JavascriptExecutor)driver).executeScript("return document.readyState").toString().equalsIgnoreCase("complete");
            }
        });    
    }
    
    public static void waitForDivLoad(WebDriver driver) {
        WebDriverWait wait = new WebDriverWait(driver, 30);
        wait.until(new ExpectedCondition<Boolean>() {
            public Boolean apply(WebDriver driver) {
                return ((JavascriptExecutor) driver).executeScript("return document.getElementById('ctl00_ContentPlaceHolder1_ClientProgress').style.display").toString().equalsIgnoreCase("block");
            }
        });
    }
}