Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- package com.yourpackage.tests;
- /**
- * Created by wcn on 4/2/2014.
- */
- import org.apache.commons.io.FileUtils;
- import org.openqa.selenium.OutputType;
- import org.openqa.selenium.WebDriver;
- import org.openqa.selenium.chrome.ChromeDriver;
- import org.openqa.selenium.firefox.FirefoxDriver;
- import org.openqa.selenium.firefox.FirefoxProfile;
- import org.openqa.selenium.ie.InternetExplorerDriver;
- import org.openqa.selenium.remote.CapabilityType;
- import org.openqa.selenium.remote.DesiredCapabilities;
- import org.testng.annotations.AfterMethod;
- import org.testng.annotations.BeforeMethod;
- import org.testng.annotations.Parameters;
- import java.io.File;
- import java.io.IOException;
- import java.util.concurrent.TimeUnit;
- public class YourTest {
- protected WebDriver driver;
- protected static String startUrl;
- @BeforeMethod
- @Parameters("browser")
- public void prepareTest(String browser){
- startUrl = "https://yourUrl.com";
- if(browser.equalsIgnoreCase("FF"))
- {
- System.out.println("Firefox driver would be used");
- final String firebugPath = "C:\\firebug.xpi";
- final String firepathPath = "C:\\firepath.xpi";
- FirefoxProfile profile = new FirefoxProfile();
- try {
- profile.addExtension(new File(firebugPath));
- profile.addExtension(new File(firepathPath));
- profile.setPreference("dom.max_script_run_time", 60);
- } catch (IOException e) {
- e.printStackTrace();
- }
- driver = new FirefoxDriver(profile);
- }
- else if(browser.equalsIgnoreCase("IE"))
- {
- System.out.println("Ie webdriver would be used");
- System.setProperty("webdriver.ie.driver", "c:\\IEDriverServer.exe");
- DesiredCapabilities capab = DesiredCapabilities.internetExplorer();
- capab.setCapability(CapabilityType.ACCEPT_SSL_CERTS, true);
- driver = new InternetExplorerDriver(capab);
- }
- else if(browser.equalsIgnoreCase("CH"))
- {
- System.out.println("Chrome webdriver would be used");
- System.setProperty("webdriver.chrome.driver", "c:\\chromedriver.exe");
- driver = new ChromeDriver();
- }
- driver.manage().timeouts().implicitlyWait(20, TimeUnit.SECONDS);
- driver.manage().timeouts().pageLoadTimeout(60, TimeUnit.SECONDS);
- driver.manage().timeouts().setScriptTimeout(60, TimeUnit.SECONDS);
- }
- @AfterMethod
- public void closeBrowser(){
- takeScreenshot();
- driver.close();
- driver.quit();
- }
- private void takeScreenshot() {
- File scrFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
- try {
- FileUtils.copyFile(scrFile, new File("target\\screenshot_" + this.getClass().getName() + ".jpg"));
- } catch (IOException e) {
- e.printStackTrace();
- }
- }
- protected void setImplicitlyWaitTimeout(long seconds){
- driver.manage().timeouts().implicitlyWait(seconds, TimeUnit.SECONDS);
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement