【selenium】实现失败用例重跑

ゝ一世哀愁。 2022-03-26 05:36 287阅读 0赞

1、新增Retry类实现IRetryAnalyzer接口的retry方法

import org.testng.IRetryAnalyzer;
    import org.testng.ITestResult;
    
    /**
     * Author: 灵枢
     * Date: 2018/12/25
     * Time: 17:40
     * Description:
     */
    public class Retry implements IRetryAnalyzer {
        private int retryCount = 0;
        private int maxRetryCount = 1; //设置重跑次数
        @Override
        public boolean retry(ITestResult iTestResult) {
            if(retryCount<maxRetryCount){
                retryCount++;
                return true;
            }
            return false;
        }
    }

2、新增RetryListener实现IAnnotationTransformer接口的transform方法

import org.testng.IAnnotationTransformer;
    import org.testng.IRetryAnalyzer;
    import org.testng.annotations.ITestAnnotation;
    
    import java.lang.reflect.Constructor;
    import java.lang.reflect.Method;
    
    /**
     * Author: 灵枢
     * Date: 2018/12/25
     * Time: 17:40
     * Description:
     */
    public class RetryListener implements IAnnotationTransformer {
    
        @Override
        public void transform(ITestAnnotation annotation, Class testClass, Constructor testConstructor, Method testMethod) {
            IRetryAnalyzer retryAnalyzer = annotation.getRetryAnalyzer();
            if(retryAnalyzer==null){
                annotation.setRetryAnalyzer(Retry.class);
            }
    
        }
    }

3、在TestngListener里重写onFinish方法

import org.apache.commons.io.FileUtils;
    import org.openqa.selenium.OutputType;
    import org.openqa.selenium.TakesScreenshot;
    import org.openqa.selenium.WebDriver;
    import org.testng.ITestContext;
    import org.testng.ITestResult;
    import org.testng.Reporter;
    import org.testng.TestListenerAdapter;
    
    import java.io.File;
    import java.io.IOException;
    import java.util.*;
    //import org.apache.poi.ss.usermodel.DataFormat;
    
    public class TestngListener extends TestListenerAdapter  {
    	
    	public static WebDriver driver;
    	
    	public void onTestFailure(ITestResult tr) {
    		
    		super.onTestFailure(tr);
    		try {
    			takeScreenShot(tr);
    		} catch (IOException e) {
    			// TODO 自动生成的 catch 块
    			e.printStackTrace();
    		}
    	}
    		
    	/**
    	 * 获取用例失败截图
    	 * @param tr
    	 * @throws IOException
    	 */
    	public void takeScreenShot(ITestResult tr) throws IOException{
    		String fileName = tr.getName()+".png";
    		File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
    		FileUtils.copyFile(srcFile, new File("FailureScreenShot/"+fileName));
    	}
    	
    	//失败用例重跑实现
    	public void onFinish(ITestContext testContext) {
    		super.onFinish(testContext);
    
    		// List of test results which we will delete later
    		ArrayList<ITestResult> testsToBeRemoved = new ArrayList<ITestResult>();
    		// collect all id's from passed test
    		Set<Integer> passedTestIds = new HashSet<Integer>();
    		for (ITestResult passedTest : testContext.getPassedTests()
    				.getAllResults()) {
    //			logger.info("PassedTests = " + passedTest.getName());
    			passedTestIds.add(getId(passedTest));
    		}
    
    		Set<Integer> failedTestIds = new HashSet<Integer>();
    		for (ITestResult failedTest : testContext.getFailedTests()
    				.getAllResults()) {
    //			logger.info("failedTest = " + failedTest.getName());
    			int failedTestId = getId(failedTest);
    
    			// if we saw this test as a failed test before we mark as to be
    			// deleted
    			// or delete this failed test if there is at least one passed
    			// version
    			if (failedTestIds.contains(failedTestId)
    					|| passedTestIds.contains(failedTestId)) {
    				testsToBeRemoved.add(failedTest);
    			} else {
    				failedTestIds.add(failedTestId);
    			}
    		}
    
    		// finally delete all tests that are marked
    		for (Iterator<ITestResult> iterator = testContext.getFailedTests()
    				.getAllResults().iterator(); iterator.hasNext();) {
    			ITestResult testResult = iterator.next();
    			if (testsToBeRemoved.contains(testResult)) {
    //				logger.info("Remove repeat Fail Test: " + testResult.getName());
    				iterator.remove();
    			}
    		}
    
    	}
    
    	private int getId(ITestResult result) {
    		int id = result.getTestClass().getName().hashCode();
    		id = id + result.getMethod().getMethodName().hashCode();
    		id = id
    				+ (result.getParameters() != null ? Arrays.hashCode(result
    				.getParameters()) : 0);
    		return id;
    	}
    }

4、在testng.xml新增RetryListener监听