TestNg的IReporter接口的使用

IReporter接口是干嘛的？就是让用户自定义报告的，很多人想要自定义报告，于是乎找各种插件，比如什么testng-xslt啊，reportng啊，各种配置，最后出来的结果，还不能定制化，但为什么不能自已定制一份呢？testng的IReporter接口就接供了这样的功能，我们只需要实现这个接口，并加上监听，就能拿到所有的信息了，你可以把这些信息存数据库，或者自已生成一个html，亦或者写在EXCEL里啊等等，都可以，下面给出IReporter接口的实现方式：

package com.demo;



import java.io.BufferedWriter;

import java.io.File;

import java.io.FileWriter;

import java.io.IOException;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Comparator;

import java.util.List;

import java.util.Map;

import java.util.Set;



import org.testng.IReporter;

import org.testng.IResultMap;

import org.testng.ISuite;

import org.testng.ISuiteResult;

import org.testng.ITestContext;

import org.testng.ITestResult;

import org.testng.xml.XmlSuite;



public class NewReport implements IReporter{

	

	@Override

	public void generateReport(List<XmlSuite> xmlSuites, List<ISuite> suites, String outputDirectory) {

		List<ITestResult> list = new ArrayList<ITestResult>();

		for (ISuite suite : suites) {			

			Map<String, ISuiteResult> suiteResults = suite.getResults();

			for (ISuiteResult suiteResult : suiteResults.values()) {

				ITestContext testContext = suiteResult.getTestContext();			

				IResultMap passedTests = testContext.getPassedTests();

				IResultMap failedTests = testContext.getFailedTests();

				IResultMap skippedTests = testContext.getSkippedTests();

				IResultMap failedConfig = testContext.getFailedConfigurations();	

				list.addAll(this.listTestResult(passedTests));

				list.addAll(this.listTestResult(failedTests));

				list.addAll(this.listTestResult(skippedTests));

				list.addAll(this.listTestResult(failedConfig));

			}			

		}

		this.sort(list);

		this.outputResult(list, outputDirectory+"/test.txt");

	}

	

	private ArrayList<ITestResult> listTestResult(IResultMap resultMap){

		Set<ITestResult> results = resultMap.getAllResults();		

		return new ArrayList<ITestResult>(results);

	}

	

	private void sort(List<ITestResult> list){

		Collections.sort(list, new Comparator<ITestResult>() {

			@Override

			public int compare(ITestResult r1, ITestResult r2) {

				if(r1.getStartMillis()>r2.getStartMillis()){

					return 1;

				}else{

					return -1;

				}				

			}

		});

	}

	

	private void outputResult(List<ITestResult> list, String path){		

		try {

			BufferedWriter output = new BufferedWriter(new FileWriter(new File(path)));	

			StringBuffer sb = new StringBuffer();

			for (ITestResult result : list) {

				if(sb.length()!=0){

					sb.append("\r\n");

				}

				sb.append(result.getTestClass().getRealClass().getName())

				  .append(" ")

				  .append(result.getMethod().getMethodName())

				  .append(" ")

				  .append(this.formatDate(result.getStartMillis()))

				  .append(" ")

				  .append(result.getEndMillis()-result.getStartMillis())

				  .append("毫秒 ")

				  .append(this.getStatus(result.getStatus()));

			}

			output.write(sb.toString());

			output.flush();

			output.close();

		} catch (IOException e) {			

			e.printStackTrace();

		}

		

	}

	

	private String getStatus(int status){

		String statusString = null;

		switch (status) {		

		case 1:

			statusString = "SUCCESS";

			break;

		case 2:

			statusString = "FAILURE";

			break;

		case 3:

			statusString = "SKIP";

			break;

		default:			

			break;

		}

		return statusString;

	}

	

	private String formatDate(long date){      

       SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");

       return formatter.format(date);

	}



}

 代码简单易懂，下面给一个测试类：

package com.demo;



import org.testng.Assert;

import org.testng.annotations.DataProvider;

import org.testng.annotations.Listeners;

import org.testng.annotations.Test;



@Listeners({com.demo.NewReport.class})

public class Test15 {

	

	@DataProvider

	public Object[][] dataProvider(){

		return new Object[][]{{1},{2}};

	}

	

	@Test(dataProvider="dataProvider")

	public void testAssert1(int a){	

		Assert.assertEquals(1, a);

	}	

	

	@Test

	public void testAssert2(){	

		Assert.assertEquals("2", "2");	

	}



}

 运行后，大家打开test-output文件夹，看是不是多了个test.txt的文件夹。截图如下：