selenium 利用testNG对异常进行自动截图

　　哈哈哈，很久没写博客了，懒了。

　　因为一些原因最近需要把监听事件重新整理一下，开始没细想，直接copy网上的，其实结果发现报错很多，或者是达不到效果，然后把之前的代码翻出来，仔细看了一下。下面给一些需要的小伙伴整理一下思路：

　　1、首先我们用到的是testng里面的监听，所以这个毋庸置疑

　　2、我们需要重新他的监听事件

　　3、用例中肯定需要加入监听事件

这三点是网上公认的，但是怎么做确没有说明白。

　　首先看一下我的目录结构：

图片中我有4个类，

1、driverBase是一个基类，里面是driver的初始化，截图的方法也在里面。

2、selectDriver是根据浏览器不同返回不同的driver

3、testngListenerScreen是testng的监听类，他继承了TestListenerAdapter类，把里面的方法进行了重写。主要他会去监听你测试类是否报错，如果报错他就会去调用我们driverbase下面的截图方法。

4、testlogin 不用说，我们的测试类，他需要继承我们的driverBase基类，去拿里面的driver。

这个说完了先看代码，首先从我们基类开始：

package com.mushishi.selenium.base;

import java.io.File;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;

/**
 * @author mushishi
 * */
public class driverBase {
	public WebDriver driver;

	/**
	 * 获取driver
	 * */
	public WebDriver getDriver() {
        return driver;
    }

	/**
	 * 自动截图
	 * */
    public void takeScreenShot() {
        SimpleDateFormat sf = new SimpleDateFormat("yyyy_MM_dd_HH_mm_ss");
        Calendar cal = Calendar.getInstance();
        Date date = cal.getTime();
        String dateStr = sf.format(date);
　　　　//上面几行代码的意思都是获取时间，并且格式化，用来作为图片的名称
　　　　//下面这个是获取当前运行的类名称和时间的组合一起命名图片
        String path = this.getClass().getSimpleName() + "_" + dateStr + ".png";
　　　　//因为我们截图是需要用到driver的，所以这里需要获取driver，这个driver是获取的当前对象的driver
        takeScreenShot((TakesScreenshot) this.getDriver(), path);

    }

    /**
     * 传入参数截图
     * */
    public void takeScreenShot(TakesScreenshot drivername, String path) {
        String currentPath = System.getProperty("user.dir"); // get current work
        File scrFile = drivername.getScreenshotAs(OutputType.FILE);
        try {
            FileUtils.copyFile(scrFile, new File(currentPath + "\\" + path));
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
        	System.out.println("截图成功");
        }
    }
}

　　上面的代码，可能有那么一两个地方看不懂，都没关系，我写备注了。主要是以类名字和时间组合起来命名，并且把当前运行的对象的driver获取到，用来截图操作。

下面看selectDriver的类：

package com.mushishi.selenium.base;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.firefox.FirefoxDriver;

public class SelectDriver {
	public WebDriver driverName(String browser){
		if(browser.equalsIgnoreCase("fireFox")){
　　　　　　　　　　　　　　//设置环境变量，并且返回driver
			System.setProperty("webdriver.firefox.marionette","D:\\java\\geckodriver\\geckodriver-v0.14.0-win64\\geckodriver.exe");
			return new FirefoxDriver();
		}else{
			System.setProperty("webdriver.chrome.driver", "D:\\java\\chromedriver_win32\\chromedriver.exe");
			return new ChromeDriver();
		}
	}
}

　　上面的没什么可看的，就是根据传入的对象生成一个driver，这个我用的3.0的selenium，所以需要这么设置。这个根据自己情况。

下面看监听类的： 1 package com.mushishi.selenium.base;

 import org.testng.ITestContext;
 import org.testng.ITestResult;
 import org.testng.TestListenerAdapter;

 public class TestNGListenerScreen extends TestListenerAdapter{

     @Override
     public void onTestSuccess(ITestResult tr) {
         super.onTestSuccess(tr);
     }
 
　　　　//主要是用到这个方法了，当你报错时他会监听到，然后就会执行截图操作，这里的 ITestresult tr是testng里的，获取到的是当前运行对象，你这样理解，他就是
//我当前运行的类这个对象
     @Override
     public void onTestFailure(ITestResult tr) {
         super.onTestFailure(tr);
         System.out.println(tr);
         takeScreenShot(tr);
     }
 //所以这里我执行截图的时候就获取到了我运行过程中实力对象的driver了，懂了吧。所以当我再回去调用我基类的对象时，那么driver就是同一个了
     private void takeScreenShot(ITestResult tr) {
         driverBase b = (driverBase) tr.getInstance();
         // driver = b.driver;
         b.takeScreenShot();

     }

     @Override
     public void onTestSkipped(ITestResult tr) {
         super.onTestSkipped(tr);
     }

     @Override
     public void onTestStart(ITestResult result) {
         super.onTestStart(result);
     }

     @Override
     public void onStart(ITestContext testContext) {
         super.onStart(testContext);
     }

     @Override
     public void onFinish(ITestContext testContext) {
         super.onFinish(testContext);
     }
 }

最后一个类，测试类：

 package com.mushishi.selenium.testCase;

 import org.openqa.selenium.By;
 import org.openqa.selenium.WebElement;
 import org.openqa.selenium.interactions.Actions;
 import org.testng.annotations.Listeners;
 import org.testng.annotations.Test;

 import com.mushishi.selenium.ProUtil;
 import com.mushishi.selenium.base.SelectDriver;
 import com.mushishi.selenium.base.TestNGListenerScreen;
 import com.mushishi.selenium.base.driverBase;

 @Listeners({ TestNGListenerScreen.class })
 public class testLogin extends driverBase {
     SelectDriver select = new SelectDriver();
     public testLogin(){
         if(driver == null){
         driver = select.driverName("chrome");
         }
     }
     public void loginScript(String username,String userpass) throws Exception{
         String url = "http://www.imooc.com";
         driver.get(url);
         driver.manage().window().maximize();
         driver.findElement(By.id("js-signin-btn")).click();
         Thread.sleep(2000);
         WebElement user = this.element(this.byStr("username"));
         user.isDisplayed();
         WebElement password = this.element(this.byStr("userpass"));
         password.isDisplayed();
         WebElement loginButton = this.element(this.byStr("loginbutton"));
         loginButton.isDisplayed();
         user.sendKeys(username);
         password.sendKeys(userpass);
         loginButton.click();
         Thread.sleep(3000);
         WebElement header = this.element(this.byStr("header"));
         header.isDisplayed();
         Actions actions = new Actions(driver);
         actions.moveToElement(header).perform();
         String userInfo = this.element(this.byStr("nameInfo")).getText();
         System.out.println(userInfo);
         if(userInfo.equals("xxxx")){
             System.out.println("登陆成功");
         }else{
             System.out.println("登陆失败");

         }
         driver.close();
     }

     /**
      * 封装By by
      * */
     public By byStr(String username){
         ProUtil properties = new ProUtil("element.properties");
         String locator = properties.getPro(username);
         String locatorType = locator.split(">")[0];
         String locatorValue = locator.split(">")[1];
         if(locatorType.equals("id")){
             return By.id(locatorValue);
         }else if(locatorType.equals("name")){
             return By.name(locatorValue);
         }else if(locatorType.equals("className")){
             return By.className(locatorValue);
         }else{
             return By.xpath(locatorValue);
         }
     }

     /**
      * 封装Element
      * */
     public WebElement element(By by){
         WebElement ele = driver.findElement(by);
         return ele;
     }

     @Test
     public void loginpage() throws Exception{
         this.loginScript("xxxx", "xxxx");
     }
 }

这个是测试类，不知道你懂了么？因为我很多都封装了的，所以看不到，但是不影响大家理解testng自动截图思想。动手操作吧。