67 lines
2.5 KiB
Java
67 lines
2.5 KiB
Java
// SPDX-FileCopyrightText: 2021 Paul Schaub <vanitasvitae@fsfe.org>
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package org.pgpainless.util;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
import java.util.stream.Stream;
|
|
|
|
import org.junit.jupiter.api.extension.BeforeTestExecutionCallback;
|
|
import org.junit.jupiter.api.extension.Extension;
|
|
import org.junit.jupiter.api.extension.ExtensionContext;
|
|
import org.junit.jupiter.api.extension.TestTemplateInvocationContext;
|
|
import org.junit.jupiter.api.extension.TestTemplateInvocationContextProvider;
|
|
import org.pgpainless.implementation.BcImplementationFactory;
|
|
import org.pgpainless.implementation.ImplementationFactory;
|
|
import org.pgpainless.implementation.JceImplementationFactory;
|
|
|
|
/**
|
|
* InvocationContextProvider that sets different {@link ImplementationFactory} implementations before running annotated
|
|
* tests.
|
|
*
|
|
* Example test annotation:
|
|
* {@code
|
|
* @TestTemplate
|
|
* @ExtendWith(ImplementationFactoryTestInvocationContextProvider.class)
|
|
* public void testAllImplementationFactories() {
|
|
* ...
|
|
* }
|
|
* }
|
|
*
|
|
* @see <a href="https://www.baeldung.com/junit5-test-templates">Baeldung: Writing Templates for Test Cases Using JUnit 5</a>
|
|
*/
|
|
public class ImplementationFactoryTestInvocationContextProvider implements TestTemplateInvocationContextProvider {
|
|
|
|
private static final List<ImplementationFactory> IMPLEMENTATIONS = Arrays.asList(
|
|
new BcImplementationFactory(),
|
|
new JceImplementationFactory()
|
|
);
|
|
|
|
@Override
|
|
public boolean supportsTestTemplate(ExtensionContext context) {
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public Stream<TestTemplateInvocationContext> provideTestTemplateInvocationContexts(ExtensionContext context) {
|
|
|
|
return IMPLEMENTATIONS.stream()
|
|
.map(implementationFactory -> new TestTemplateInvocationContext() {
|
|
@Override
|
|
public String getDisplayName(int invocationIndex) {
|
|
return context.getDisplayName() + " with " + implementationFactory.getClass().getSimpleName();
|
|
}
|
|
|
|
@Override
|
|
public List<Extension> getAdditionalExtensions() {
|
|
return Collections.singletonList(
|
|
(BeforeTestExecutionCallback) ctx -> ImplementationFactory.setFactoryImplementation(implementationFactory)
|
|
);
|
|
}
|
|
});
|
|
}
|
|
}
|