WebDriverAbstractBaseScreen.java

package com.photon.phresco.Screens;

import java.awt.AWTException;
import java.io.File;
import java.io.IOException;

import org.apache.commons.io.FileUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.support.ui.WebDriverWait;

import com.google.common.base.Function;
import com.photon.phresco.selenium.util.GetCurrentDir;
import com.photon.phresco.selenium.util.ScreenActionFailedException;
import com.photon.phresco.selenium.util.ScreenException;

public class WebDriverAbstractBaseScreen extends BaseScreen {
	
	public static WebElement element;	
	private Log log = LogFactory.getLog(getClass());
	private static final long TIMEOUT=200;
    
	
	protected WebDriverAbstractBaseScreen(){
		
	}
	protected WebDriverAbstractBaseScreen(String host,int port,String browser,String url,String speed,String context) throws AWTException, IOException, ScreenActionFailedException{
		super(host,port,browser,url,speed,context);
		log.info("Entering:********WebDriverAbstractBaseScreen Constructor********");
	}

	public WebDriverBaseScreen getXpathWebElement(String xpath)throws Exception{
		log.info("Entering:********getXpathWebElement********");
		try{
			
			element=driver.findElement(By.xpath(xpath));
			System.out.println("------------------->"+element);
			
		}catch(Throwable t){
			log.info("Entering:*********Exception in getXpathWebElement()******");
			t.printStackTrace();
			
		}
		return new WebDriverBaseScreen();
	}
	public WebDriverBaseScreen getIdWebElement(String id)throws ScreenException{
		log.info("Entering:****getIdWebElement**********");
		try{
		element=driver.findElement(By.id(id));
		System.out.println("------------------->"+element);
		
		}catch(Throwable t){
			log.info("Entering:*********Exception in getIdWebElement()******");
			t.printStackTrace();
			
		}
		return new WebDriverBaseScreen();
	}
	public WebDriverBaseScreen getcssWebElement(String selector)throws ScreenException{
		log.info("Entering:****getIdWebElement**********");
		try{
		element=driver.findElement(By.cssSelector(selector));
		
		
		
		}catch(Throwable t){
			log.info("Entering:*********Exception in getIdWebElement()******");
			
			t.printStackTrace();
			
		}
		return new WebDriverBaseScreen();
	}
	public void deleteAllCookies()throws IOException, Exception{
		try{
			log.info("Entering:*********deleteAllCookies()******");
			driver.manage().deleteAllCookies();
		}catch(Exception e){
		e.printStackTrace();
		ScreenCapturer();
		}
	}
	
	public void waitForElementPresent(String locator, String methodName) throws IOException, Exception{
		try{
		log.info("Entering:*********waitForElementPresent()******");
	    By by= By.xpath(locator);
		WebDriverWait wait=new WebDriverWait(driver, TIMEOUT);
		log.info("Waiting:*************For element" +locator);
		wait.until(presenceOfElementLocated(by));
		}
		catch(Exception e){			
			File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
		    FileUtils.copyFile(scrFile, new File(GetCurrentDir.getCurrentDirectory()+"\\" + methodName + ".png"));
//		    e.printStackTrace();
		    throw new Exception("********Exception*******");
		    
		}
	
	}
	public void waitForTextPresent(String text) throws IOException, Exception{
		try{
		log.info("Entering:*********waitForElementPresent()******");
	    //By by= By.linkText(locator);
		WebDriverWait wait=new WebDriverWait(driver, TIMEOUT);
		driver.findElement(By.tagName(text)).getText();
		log.info("Waiting:*************One second***********");
		//wait.until(presenceOfElementLocated(by));
		}catch(Exception e){
			e.printStackTrace();
			ScreenCapturer();
			
		}
	
	}
	public void ScreenCapturer() throws IOException, Exception{
		try{
		File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
	    FileUtils.copyFile(scrFile, new File(GetCurrentDir.getCurrentDirectory()+"\\" + super.getClass().getSimpleName()+ ".png"));
		}catch(Exception e){
			e.printStackTrace();
			System.out.println("Screen was not captured");
		}
		
	}
	
	Function<WebDriver, WebElement> presenceOfElementLocated(final By locator) { 
		log.info("Entering:*********presenceOfElementLocated()******Start");
		  return new Function<WebDriver, WebElement>() { 
		    public WebElement apply(WebDriver driver) { 
		    	log.info("Entering:*********presenceOfElementLocated()******End");
		     return driver.findElement(locator);
		     
		    }
		    
		  };
		  
	
		 }
	
/*
	Function<WebDriver, WebElement> presenceOfTextLocated(final By locator) { 
		log.info("Entering:*********presenceOfElementLocated()******Start");
		 driver.findElement(By.tagName("body")).getText()
		     
		    }
		  
	}*/
	}