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");
             }
         });
     }
 }