diff --git a/spring-restdocs-core/src/main/java/org/springframework/restdocs/RestDocumentationExtension.java b/spring-restdocs-core/src/main/java/org/springframework/restdocs/RestDocumentationExtension.java index f384255a..2ad54a20 100644 --- a/spring-restdocs-core/src/main/java/org/springframework/restdocs/RestDocumentationExtension.java +++ b/spring-restdocs-core/src/main/java/org/springframework/restdocs/RestDocumentationExtension.java @@ -47,8 +47,11 @@ public class RestDocumentationExtension @Override public boolean supportsParameter(ParameterContext parameterContext, ExtensionContext extensionContext) { - return RestDocumentationContextProvider.class - .isAssignableFrom(parameterContext.getParameter().getType()); + if (isTestMethodContext(extensionContext)) { + return RestDocumentationContextProvider.class + .isAssignableFrom(parameterContext.getParameter().getType()); + } + return false; } @Override @@ -58,6 +61,10 @@ public class RestDocumentationExtension .beforeOperation(); } + private boolean isTestMethodContext(ExtensionContext context) { + return context.getTestClass().isPresent() && context.getTestMethod().isPresent(); + } + private ManualRestDocumentation getDelegate(ExtensionContext context) { Namespace namespace = Namespace.create(getClass(), context.getUniqueId()); return context.getStore(namespace).getOrComputeIfAbsent(