selenium自动化-java-封断言类2

封装断言类

 package streamax.com;

 import java.util.ArrayList;
 import java.util.List;

 import org.testng.Assert;
  /*
   * 断言类
   * 被調用判斷
   *
   */
 public class Assertion {

     public static boolean flag = true;

     public static List<Error> errors = new ArrayList<Error>();

     public static void verifyEquals(Object actual, Object expected){
         try{
             Assert.assertEquals(actual, expected);
         }catch(Error e){
             errors.add(e);
             flag = false;
         }
     }

     public static void verifyEquals(Object actual, Object expected, String message){
         try{
             Assert.assertEquals(actual, expected, message);
         }catch(Error e){
             errors.add(e);
             flag = false;
         }
     }

 }

2 监听类

 package streamax.com;

 import java.util.ArrayList;
 import java.util.List;

 import org.testng.ITestResult;
 import org.testng.TestListenerAdapter;
  /*
   * 监听类
   */
 public class AssertionListener extends TestListenerAdapter {

     @Override
     public void onTestStart(ITestResult result) {
         Assertion.flag = true;
         Assertion.errors.clear();
     }

     @Override
     public void onTestFailure(ITestResult tr) {
         this.handleAssertion(tr);
     }

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

     @Override
     public void onTestSuccess(ITestResult tr) {
         this.handleAssertion(tr);
     }

     private int index = 0;

     private void handleAssertion(ITestResult tr){
         if(!Assertion.flag){
             Throwable throwable = tr.getThrowable();
             if(throwable==null){
                 throwable = new Throwable();
             }
             StackTraceElement[] traces = throwable.getStackTrace();
             StackTraceElement[] alltrace = new StackTraceElement[0];
             for (Error e : Assertion.errors) {
                 StackTraceElement[] errorTraces = e.getStackTrace();
                 StackTraceElement[] et = this.getKeyStackTrace(tr, errorTraces);
                 StackTraceElement[] message = new StackTraceElement[]{new StackTraceElement("message : "+e.getMessage()+" in method : ", tr.getMethod().getMethodName(), tr.getTestClass().getRealClass().getSimpleName(), index)};
                 index = 0;
                 alltrace = this.merge(alltrace, message);
                 alltrace = this.merge(alltrace, et);
             }
             if(traces!=null){
                 traces = this.getKeyStackTrace(tr, traces);
                 alltrace = this.merge(alltrace, traces);
             }
             throwable.setStackTrace(alltrace);
             tr.setThrowable(throwable);
             Assertion.flag = true;
             Assertion.errors.clear();
             tr.setStatus(ITestResult.FAILURE);
         }
     }

     private StackTraceElement[] getKeyStackTrace(ITestResult tr, StackTraceElement[] stackTraceElements){
         List<StackTraceElement> ets = new ArrayList<StackTraceElement>();
         for (StackTraceElement stackTraceElement : stackTraceElements) {
             if(stackTraceElement.getClassName().equals(tr.getTestClass().getName())){
                 ets.add(stackTraceElement);
                 index = stackTraceElement.getLineNumber();
             }
         }
         StackTraceElement[] et = new StackTraceElement[ets.size()];
         for (int i = 0; i < et.length; i++) {
             et[i] = ets.get(i);
         }
         return et;
     }

     private StackTraceElement[] merge(StackTraceElement[] traces1, StackTraceElement[] traces2){
         StackTraceElement[] ste = new StackTraceElement[traces1.length+traces2.length];
         for (int i = 0; i < traces1.length; i++) {
             ste[i] = traces1[i];
         }
         for (int i = 0; i < traces2.length; i++) {
             ste[traces1.length+i] = traces2[i];
         }
         return ste;
     }
 }