Skip to content

Commit f002c40

Browse files
committed
Use more correct stubs for RunnerToExtension
1 parent e90b474 commit f002c40

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

src/main/java/org/openrewrite/java/testing/junit5/RunnerToExtension.java

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,7 @@
1818
import com.fasterxml.jackson.annotation.JsonCreator;
1919
import lombok.EqualsAndHashCode;
2020
import lombok.Value;
21-
import org.openrewrite.ExecutionContext;
22-
import org.openrewrite.Option;
23-
import org.openrewrite.Recipe;
24-
import org.openrewrite.TreeVisitor;
21+
import org.openrewrite.*;
2522
import org.openrewrite.internal.lang.Nullable;
2623
import org.openrewrite.java.JavaIsoVisitor;
2724
import org.openrewrite.java.JavaParser;
@@ -96,14 +93,20 @@ private JavaTemplate getExtendsWithTemplate() {
9693
if (extendsWithTemplate == null) {
9794
extendsWithTemplate = JavaTemplate.builder(this::getCursor, "@ExtendWith(#{}.class)")
9895
.javaParser(() -> JavaParser.fromJavaVersion().dependsOn(Arrays.asList(
96+
fromString("package org.junit.jupiter.api.extension;\n" +
97+
"public interface Extension {}"),
9998
fromString("package org.junit.jupiter.api.extension;\n" +
10099
"public @interface ExtendWith {\n" +
101100
" Class<? extends Extension>[] value();\n" +
102101
"}"),
103102
fromString("package " + extensionType.getPackageName() + ";\n" +
104-
"public class " + extensionType.getClassName() + " {}"
105-
))).build())
106-
.imports("org.junit.jupiter.api.extension.ExtendWith", extension)
103+
"import org.junit.jupiter.api.extension.Extension;\n" +
104+
"public class " + extensionType.getClassName() + " implements Extension {}"
105+
)))
106+
.classpath("junit")
107+
.build())
108+
.imports("org.junit.jupiter.api.extension.ExtendWith",
109+
"org.junit.jupiter.api.extension.Extension", extension)
107110
.build();
108111
}
109112
return extendsWithTemplate;

0 commit comments

Comments
 (0)