Java 类org.testng.IHookCallBack 实例源码

@Override
public void run(IHookCallBack callBack, ITestResult testResult) {
    super.run(callBack, testResult);
    if (testResult.getThrowable() != null) {
        try {
            final Throwable testResultThrowable = testResult.getThrowable();
            String message = testResultThrowable.getMessage() != null ? testResultThrowable.getMessage() :
                    testResultThrowable.getCause().getMessage();

            if (message == null) {
                message = "Test failed";
            }

            new Screenshoter().takeScreenshot(message, TestUtils.getTestName(testResult));
        } catch (Exception e) {
            log("Couldn't take screenshot. Error: " + e.getMessage());
        }
    }
}

/**
 * 测试方法拦截器、所有测试方法在执行开始、执行中、执行完成都会在此方法中完成
 * @param callBack
 * @param testResult
 */
@Override
public void run(IHookCallBack callBack, ITestResult testResult) {
    // 如果测试方法上有@Ignore注解，则跳过测试框架直接执行测试方法
    if (isIgnoreAnnotation(getTestMethod(testResult))) {
        callBack.runTestMethod(testResult);
    }
}

@SuppressWarnings("unchecked")
private void prepareTest(IHookCallBack callBack, ITestResult testResult) {
    Optional<Map<String, Object>> paramOptional = ofNullable(getParameters(testResult));
    Map<String, Object> param;
    if (paramOptional.isPresent()) {
        param = paramOptional.get();
    } else {
        throw new ParameterException("---------------获取测试入参失败！---------------");
    }
    Class<T> clazz = (Class<T>) getInterfaceClass(this);
}

/**
 * Delegates to the {@link IHookCallBack#runTestMethod(ITestResult) test
 * method} in the supplied {@code callback} to execute the actual test
 * and then tracks the exception thrown during test execution, if any.
 *
 * @see org.testng.IHookable#run(org.testng.IHookCallBack,
 * org.testng.ITestResult)
 */
@Override
public void run(IHookCallBack callBack, ITestResult testResult) {
    callBack.runTestMethod(testResult);

    Throwable testResultException = testResult.getThrowable();
    if (testResultException instanceof InvocationTargetException) {
        testResultException = ((InvocationTargetException) testResultException).getCause();
    }
    this.testException = testResultException;
}

@Override
public void run(IHookCallBack callBack, ITestResult testResult) {
    long time = System.currentTimeMillis();
    LOGGER.info("###>>> run start");
    super.run(callBack, testResult);
    LOGGER.info("###>>> run end ({}ms)", new Object[]{(System.currentTimeMillis() - time)});
}

@Override
public void run(final IHookCallBack callBack,
                final ITestResult testResult) {

    try {
        injectTestResult(testResult);
    } catch (final ReflectiveOperationException roe) {
        testResult.setThrowable(roe);
        return;
    }

    callBack.runTestMethod(testResult);
}

@Override
public void run(IHookCallBack callBack, ITestResult testResult) {
    long time = System.currentTimeMillis();
    LOGGER.info("###>>> run start");
    super.run(callBack, testResult);
    LOGGER.info("###>>> run end ({}ms)", new Object[]{(System.currentTimeMillis() - time)});
}

/**
 * Delegates to the {@link IHookCallBack#runTestMethod(ITestResult) test
 * method} in the supplied {@code callback} to execute the actual test
 * and then tracks the exception thrown during test execution, if any.
 *
 * @see org.testng.IHookable#run(org.testng.IHookCallBack,
 * org.testng.ITestResult)
 */
public void run(IHookCallBack callBack, ITestResult testResult) {
    callBack.runTestMethod(testResult);

    Throwable testResultException = testResult.getThrowable();
    if (testResultException instanceof InvocationTargetException) {
        testResultException = ((InvocationTargetException) testResultException).getCause();
    }
    this.testException = testResultException;
}

@Override
public void run(IHookCallBack callback, ITestResult res) {
  System.out.println("Starting " +  res.getName());
  callback.runTestMethod(res);
  if (res.getThrowable() != null) {
    res.setThrowable(null);
    System.out.println("Second attempt " +  res.getName());
    callback.runTestMethod(res);
  }
}

@Override
public void run(IHookCallBack callBack, ITestResult testResult) {
    long time = System.currentTimeMillis();
    LOGGER.info("###>>> run start");
    super.run(callBack, testResult);
    LOGGER.info("###>>> run end ({}ms)", new Object[]{(System.currentTimeMillis() - time)});
}

public void run(IHookCallBack hookCallBack) {
}

public void run(IHookCallBack hookCallBack) {
}

@Override
public void run(IHookCallBack callBack, ITestResult testResult) {
    this.method = testResult.getMethod().getMethodName();
    super.run(callBack, testResult);
}

@Override
public void run(IHookCallBack iHookCallBack,
        ITestResult iTestResult) {
    super.run(iHookCallBack, iTestResult);
}

@Override
public void run(IHookCallBack iHookCallBack, ITestResult iTestResult) {
  iHookCallBack.runTestMethod(iTestResult);
  this.tearDownStack.runTearDown();
}

@Override
public void run(IHookCallBack callBack, ITestResult testResult) {

}