Java Code Examples for org.junit.jupiter.api.extension.ExtensionContext#getRequiredTestMethod()

/**
 * Delegates to {@link TestContextManager#beforeTestExecution}.
 */
@Override
public void beforeTestExecution(ExtensionContext context) throws Exception {
	Object testInstance = context.getRequiredTestInstance();
	Method testMethod = context.getRequiredTestMethod();
	getTestContextManager(context).beforeTestExecution(testInstance, testMethod);
}
 

/**
 * Delegates to {@link TestContextManager#afterTestMethod}.
 */
@Override
public void afterEach(ExtensionContext context) throws Exception {
	Object testInstance = context.getRequiredTestInstance();
	Method testMethod = context.getRequiredTestMethod();
	Throwable testException = context.getExecutionException().orElse(null);
	getTestContextManager(context).afterTestMethod(testInstance, testMethod, testException);
}
 

@Override
protected Stream<TestTemplateInvocationContext> provideInvocationContexts(ExtensionContext context,
        TEST_ANNOTATION annotation) {
    Method testMethod = context.getRequiredTestMethod();

    DataProviderResolverContext dataProviderResolverContext = getDataProviderResolverContext(context, annotation);
    List<Method> dataProviderMethods = findDataProviderMethods(dataProviderResolverContext);

    checkArgument(dataProviderMethods.size() > 0,
            String.format("Could not find a dataprovider for test '%s' using resolvers '%s'.", testMethod,
                    dataProviderResolverContext.getResolverClasses()));

    return dataProviderMethods.stream().flatMap(dpm -> {
        DATAPROVIDER_ANNOTATION dataProviderAnnotation = dpm.getAnnotation(dataProviderAnnotationClass);
        boolean cacheDataProviderResult = cacheDataProviderResult(dataProviderAnnotation);

        Object data = invokeDataProviderMethodToRetrieveData(dpm, cacheDataProviderResult, context);

        return convertData(testMethod, data, getConverterContext(dataProviderAnnotation))
                .map(d -> (TestTemplateInvocationContext) new DataProviderInvocationContext(testMethod, d,
                        getDisplayNameContext(dataProviderAnnotation)));
    });
}
 

@Override
public void afterTestExecution(ExtensionContext context) throws Exception {
    Method testMethod = context.getRequiredTestMethod();
    long start = getStore(context).remove(testMethod, long.class);
    long duration = System.currentTimeMillis() - start;

    log.info("Method {} took {} ms", testMethod.getName(), duration);
}
 

@Override
protected DataProviderResolverContext getDataProviderResolverContext(ExtensionContext extensionContext,
        CustomResolverUseDataProvider annotation) {
    return new DataProviderResolverContext(extensionContext.getRequiredTestMethod(), asList(annotation.resolver()),
            annotation.resolveStrategy(),
            generateLocations(extensionContext.getRequiredTestClass(), annotation.location()),
            DataProvider.class, annotation.value());
}
 

@Override
public Stream<TestTemplateInvocationContext> provideTestTemplateInvocationContexts(ExtensionContext context) {
    Method testMethod = context.getRequiredTestMethod();

    return AnnotationUtils.findAnnotation(testMethod, testAnnotationClass)
            .map(annotation -> provideInvocationContexts(context, annotation))
            .orElseThrow(() -> new ExtensionConfigurationException(String.format(
                    "Could not find annotation '%s' on test method '%s'.", testAnnotationClass, testMethod)));
}
 

@Override
public void beforeEach(ExtensionContext extensionContext) {
    Method requiredTestMethod = extensionContext.getRequiredTestMethod();
    if (requiredTestMethod.getAnnotation(DataSet.class) != null) {
        setUp(extensionContext.getRequiredTestClass());
    }
}
 

@Override
public Stream<TestTemplateInvocationContext> provideTestTemplateInvocationContexts(ExtensionContext context) {
	final ResourceSetGlobalCompilationContext ctx = getOrCreateCompilationContext(context);
	final Method meth = context.getRequiredTestMethod();
	ctx.setCurrentMethod(meth.getName());
	return Stream.of(new DynamicContext(meth.getName()));
}
 

/**
 * Delegates to {@link TestContextManager#beforeTestMethod}.
 */
@Override
public void beforeEach(ExtensionContext context) throws Exception {
	Object testInstance = context.getRequiredTestInstance();
	Method testMethod = context.getRequiredTestMethod();
	getTestContextManager(context).beforeTestMethod(testInstance, testMethod);
}
 

@Override
public void beforeEach(ExtensionContext extensionContext) throws Exception {
    Method testMethod = extensionContext.getRequiredTestMethod();
    Class<?> testClass = extensionContext.getRequiredTestClass();
    String loadPropertiesFile = validateAndGetLoadPropertiesFile(testClass, testMethod);
    JupiterLoadProcessor loadProcessor = new JupiterLoadProcessor(loadPropertiesFile);

    //-------------------------------------------
    //       On/Off Extent-Report switch
    //-------------------------------------------
    // Load the key 'chart.dashboard.generation'
    // from 'loadPropertiesFile'
    //-------------------------------------------
    boolean chartAndDashBoardGenerationEnabled = false;


    TestMapping[] testMappingArray = testMethod.getAnnotationsByType(TestMapping.class);

    Arrays.stream(testMappingArray).forEach(thisMapping -> {
        loadProcessor.addJupiterTest(thisMapping.testClass(), thisMapping.testMethod());
    });

    boolean hasFailed = loadProcessor.processMultiLoad();

    reportGenerator.generateCsvReport();
    if (chartAndDashBoardGenerationEnabled) {
        reportGenerator.generateExtentReport();
    }

    if (hasFailed) {
        failTest(testMethod, testClass);
    } else {
        LOGGER.info("\nAll Passed \uD83D\uDC3C. \nSee the granular 'csv report' for individual test statistics.");
    }

}
 

private RetryExecutor retryExecutor(ExtensionContext context) {
    Method method = context.getRequiredTestMethod();
    ExtensionContext templateContext = context.getParent()
                                              .orElseThrow(() -> new IllegalStateException(
                                                  "Extension context \"" + context + "\" should have a parent context."));

    int maxRetries = maxRetries(context);
    return templateContext.getStore(NAMESPACE).getOrComputeIfAbsent(method.toString(), __ -> new RetryExecutor(maxRetries),
                                                                    RetryExecutor.class);


}
 

/**
 * Delegates to {@link TestContextManager#beforeTestMethod}.
 */
@Override
public void beforeEach(ExtensionContext context) throws Exception {
	Object testInstance = context.getRequiredTestInstance();
	Method testMethod = context.getRequiredTestMethod();
	getTestContextManager(context).beforeTestMethod(testInstance, testMethod);
}
 

@Override
public void afterTestExecution(ExtensionContext context) throws Exception {
    Method testMethod = context.getRequiredTestMethod();
    long start = getStore(context).remove(testMethod, long.class);
    long duration = System.currentTimeMillis() - start;

    log.info("Method {} took {} ms", testMethod.getName(), duration);
}
 

/**
 * Delegates to {@link TestContextManager#afterTestMethod}.
 */
@Override
public void afterEach(ExtensionContext context) throws Exception {
	Object testInstance = context.getRequiredTestInstance();
	Method testMethod = context.getRequiredTestMethod();
	Throwable testException = context.getExecutionException().orElse(null);
	getTestContextManager(context).afterTestMethod(testInstance, testMethod, testException);
}
 

@Override
public Stream<? extends Arguments> provideArguments(ExtensionContext context) {
    Method testMethod = context.getRequiredTestMethod();

    DataProviderResolverContext resolverContext = getDataProviderResolverContext(context, sourceAnnotation);
    return findDataProviderMethods(resolverContext).stream().flatMap(dpm -> {
        DATAPROVIDER_ANNOTATION dataProviderAnnotation = dpm.getAnnotation(dataProviderAnnotationClass);
        boolean cacheDataProviderResult = cacheDataProviderResult(dataProviderAnnotation);

        Object data = invokeDataProviderMethodToRetrieveData(dpm, cacheDataProviderResult, context);
        return convertData(testMethod, data, getConverterContext(dataProviderAnnotation));
    });
}
 

@Override
protected DataProviderResolverContext getDataProviderResolverContext(ExtensionContext extensionContext,
        CustomResolverDataProviderTest annotation) {
    return new DataProviderResolverContext(extensionContext.getRequiredTestMethod(), asList(annotation.resolver()),
            annotation.resolveStrategy(),
            generateLocations(extensionContext.getRequiredTestClass(), annotation.location()),
            DataProvider.class, annotation.value());
}
 

@Override
protected Stream<TestTemplateInvocationContext> provideInvocationContexts(ExtensionContext extensionContext,
        TEST_ANNOTATION testAnnotation) {
    Method testMethod = extensionContext.getRequiredTestMethod();
    return convertData(testMethod, getData(testAnnotation), getConverterContext(testAnnotation))
            .map(d -> new DataProviderInvocationContext(testMethod, d, getDisplayNameContext(testAnnotation)));
}
 

/**
 * Delegates to {@link TestContextManager#beforeTestExecution}.
 */
@Override
public void beforeTestExecution(ExtensionContext context) throws Exception {
	Object testInstance = context.getRequiredTestInstance();
	Method testMethod = context.getRequiredTestMethod();
	getTestContextManager(context).beforeTestExecution(testInstance, testMethod);
}
 

@Override
public Stream<? extends Arguments> provideArguments(ExtensionContext extensionContext) {
    Method testMethod = extensionContext.getRequiredTestMethod();
    return convertData(testMethod, getData(sourceAnnotation), getConverterContext(sourceAnnotation));
}
 

public PerfConfigContext(ExtensionContext context) {
    this.method = context.getRequiredTestMethod();
    this.perfConfig = method.getAnnotation(JunitPerfConfig.class);
    this.perfRequire = method.getAnnotation(JunitPerfRequire.class);
}