sparkmuse / wiremock-junit-jupiter Goto Github PK

@Override
 public void afterEach(ExtensionContext extensionContext) {
     // because we are in afterEach, ExtensionContext actually is a MethodExtensionContext which always has non-empty parent
     @SuppressWarnings("OptionalGetWithoutIsPresent")
     ExtensionContext classContext = extensionContext.getParent().get();
     List<Field> serverFields = retrieveAnnotatedFields(classContext, Wiremock.class, WireMockServer.class);
     ExtensionContext.Store store = getStore(classContext);

     serverFields.stream()
             .map(field -> store.get(field.getName(), WireMockServer.class))
             .forEach(WireMockServer::resetAll);
 }

MethodExtensionContext
TestTemplateExtensionContext
ClassExtensionContext
JupiterEngineExtensionContext

	private static List<Field> retrieveAnnotatedFields(ExtensionContext context,
                                                       Class<? extends Annotation> annotationType,
                                                       Class<?> fieldType) {
        return context.getElement()
                .map(Class.class::cast)
                .map(clazz -> findAnnotatedFields(clazz, annotationType, getFieldPredicate(fieldType)))
                .orElse(Collections.emptyList());
    }

@Override
public void afterEach(ExtensionContext extensionContext) {
    List<Field> serverFields = retrieveAnnotatedFields(extensionContext.getRoot(), Wiremock.class, WireMockServer.class);
    ExtensionContext.Store store = getStore(extensionContext);

    serverFields.stream()
                .map(field -> store.get(field.getName(), WireMockServer.class))
                .forEach(WireMockServer::resetAll);
}

@Override
public void afterEach(ExtensionContext extensionContext) {
    List<Field> serverFields = retrieveAnnotatedFields(extensionContext.getParent().get(), Wiremock.class, WireMockServer.class);
    ExtensionContext.Store store = getStore(extensionContext.getParent().get());

    serverFields.stream()
                .map(field -> store.get(field.getName(), WireMockServer.class))
                .forEach(WireMockServer::resetAll);
}