Java ITestResult.setAttribute方法代码示例

import org.testng.ITestResult; //导入方法依赖的package包/类
/**
 * Store the specified object in the attributes collection.
 * 
 * @param obj object to be stored; 'null' to discard value
 * @return (optional) specified object
 */
private <T> Optional<T> set(T obj) {
    ITestResult result = Reporter.getCurrentTestResult();
    Optional<T> val = TestBase.optionalOf(obj);
    result.setAttribute(key, val);
    return val;
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
@Test
public void testAttributeExtraction() {
    
    ITestResult testResult = Reporter.getCurrentTestResult();
    
    testResult.setAttribute(BOOLEAN_KEY, BOOLEAN_VAL);
    testResult.setAttribute(STRING_KEY, STRING_VAL);
    testResult.setAttribute(OBJECT_KEY, OBJECT_VAL);
    Map<String, Object> extractedMap = PropertyManager.extractAttributes(testResult);
    
    assertEquals(extractedMap, attributeMap, "Extracted map differs from expected map");
    
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
@Override
public boolean retryMethod(ITestResult result) {
    boolean willRetry = !result.isSuccess();
    if (willRetry) {
        result.setAttribute("retry", true);
    }
    return willRetry;
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
/**
 * Record the path at which the specified artifact was store in the indicated test result.
 * @param artifactPath path at which the captured artifact was stored 
 * @param result TestNG test result object
 */
private static void recordArtifactPath(Path artifactPath, ITestResult result) {
    @SuppressWarnings("unchecked")
    List<Path> artifactPaths = (List<Path>) result.getAttribute(ARTIFACT_PATHS);
    if (artifactPaths == null) {
        artifactPaths = new ArrayList<>();
        result.setAttribute(ARTIFACT_PATHS, artifactPaths);
    }
    artifactPaths.add(artifactPath);
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
/**
 * Get the TestNG configuration object for the specified context.
 * 
 * @param testResult configuration context (TestNG test result object)
 * @return TestNG configuration object
 */
public static TestNGConfig getConfig(ITestResult testResult) {
    if (testResult == null) {
        return testNGConfig.get();
    }
    
    TestNGConfig config = (TestNGConfig) testResult.getAttribute(TESTNG_CONFIG);
    
    if (config == null) {
        config = testNGConfig.get();
        testResult.setAttribute(TESTNG_CONFIG, config);
    }
    
    return config;
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
/**
 * Inject the entries of the specified {@link Map} into the attribute collection of the specified test result.
 * 
 * @param attributes map of test result attributes
 * @param testResult test result object
 */
public static void injectAttributes(Map<String, Object> attributes, ITestResult testResult) {
    if (attributes != null) {
        for (Entry<String, Object> thisEntry : attributes.entrySet()) {
            testResult.setAttribute(thisEntry.getKey(), thisEntry.getValue());
        }
    }
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
private void setLog(ITestResult tr) {
	StringWriter exception = new StringWriter();
	if (tr.getThrowable() != null) {
		tr.getThrowable().printStackTrace(new PrintWriter(exception));
	}
	tr.setAttribute("log", MTLogBuffer.getStringLog() + exception);
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
private void takeScreenshot(ITestResult tr) {
	if (Config.driver() != null) {
		String file = SeleniumUtils.takeScreenShot();
		String imageFile = SeleniumUtils.SCREENSHOT_PATH + file;
		String thumbFile = SeleniumUtils.SCREENSHOT_PATH + "thumb_" + file;
		buildThumbImage(imageFile, thumbFile);
		tr.setAttribute("screenshot", imageFile);
		tr.setAttribute("thumbScreenshot", thumbFile);
	}
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
@AfterMethod(alwaysRun = true)
public void baseCleanup(ITestContext context, ITestResult result, Object[] params) throws IOException {
	if (params.length > 0 && params[0] instanceof WebDriverInstance){
		WebDriverInstance webDriverInstance = (WebDriverInstance)params[0];
		webDriverInstance.cleanup();
	}
	logTestResult(result);
	String tag = ThreadContext.get(THREAD_TAG);
	logger.info("Test is finished, removing the tag [{}]...", tag);
	ThreadContext.remove(THREAD_TAG);
	result.setAttribute(THREAD_TAG, tag);
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
static void disableCapture(ITestResult result) {
    result.setAttribute(CAPTURE_DISABLED, Boolean.TRUE);
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
static void crippleCapture(ITestResult result) {
    result.setAttribute(CAPTURE_CRIPPLED, Boolean.TRUE);
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
static void setCaptureState(ITestResult result, CaptureState state) {
    result.setAttribute(CAPTURE_STATE, state);
}
 

import org.testng.ITestResult; //导入方法依赖的package包/类
/**
 * Capture artifact from the current test result context.
 * <br><br>
 * <b>NOTE</b>: This override is here solely to record the artifact path for the benefit of the unit tests,
 * as verification meta-data. It makes no contribution to the actual process of artifact capture
 * 
 * @param testResult TestNG test result object
 * @return path at which the captured artifact was stored
 */
@Override
public Optional<Path> captureArtifact(ITestResult testResult) {
    Optional<Path> artifactPath = super.captureArtifact(testResult);
    testResult.setAttribute(ARTIFACT_PATH, artifactPath);
    return artifactPath;
}