org.openrewrite.java.testing.junit5.UseWiremockExtension Maven / Gradle / Ivy
Show all versions of rewrite-testing-frameworks Show documentation
/*
* Copyright 2024 the original author or authors.
*
* Licensed under the Moderne Source Available License (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://docs.moderne.io/licensing/moderne-source-available-license
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.openrewrite.java.testing.junit5;
import org.openrewrite.ExecutionContext;
import org.openrewrite.Preconditions;
import org.openrewrite.Recipe;
import org.openrewrite.TreeVisitor;
import org.openrewrite.java.*;
import org.openrewrite.java.dependencies.UpgradeDependencyVersion;
import org.openrewrite.java.search.UsesType;
import org.openrewrite.java.tree.*;
import java.util.List;
import static java.util.Collections.emptyList;
import static java.util.Collections.singletonList;
public class UseWiremockExtension extends Recipe {
private static final MethodMatcher newWiremockRule = new MethodMatcher("com.github.tomakehurst.wiremock.junit.WireMockRule (..)");
@Override
public String getDisplayName() {
return "Use wiremock extension";
}
@Override
public String getDescription() {
return "As of 2.31.0, wiremock [supports JUnit 5](https://wiremock.org/docs/junit-jupiter/) via an extension.";
}
@Override
public TreeVisitor getVisitor() {
return Preconditions.check(new UsesType<>("com.github.tomakehurst.wiremock.junit.WireMockRule", false), new JavaVisitor() {
@Override
public J preVisit(J tree, ExecutionContext ctx) {
if (tree instanceof JavaSourceFile) {
doAfterVisit(new ChangeType("com.github.tomakehurst.wiremock.junit.WireMockRule",
"com.github.tomakehurst.wiremock.junit5.WireMockExtension", true).getVisitor());
}
return tree;
}
@SuppressWarnings("ConcatenationWithEmptyString")
@Override
public J visitNewClass(J.NewClass newClass, ExecutionContext ctx) {
J.NewClass n = (J.NewClass) super.visitNewClass(newClass, ctx);
if (newWiremockRule.matches(n)) {
maybeAddImport("com.github.tomakehurst.wiremock.junit5.WireMockExtension");
doAfterVisit(new ChangeType("org.junit.Rule", "org.junit.jupiter.api.extension.RegisterExtension", true)
.getVisitor());
Expression arg = n.getArguments().get(0);
JavaParser.Builder wiremockParser = JavaParser.fromJavaVersion()
.dependsOn(
//language=java
"" +
"package com.github.tomakehurst.wiremock.junit5;" +
"import com.github.tomakehurst.wiremock.core.Options;" +
"public class WireMockExtension {" +
" public native static Builder newInstance();" +
" public static class Builder {" +
" public native Builder options(Options options);" +
" public native Builder failOnUnmatchedRequests(boolean failOnUnmatched);" +
" public native WireMockExtension build();" +
" }" +
"}",
//language=java
"" +
"package com.github.tomakehurst.wiremock.core;" +
"public class WireMockConfiguration implements Options {" +
" public static native WireMockConfiguration options();" +
" public native WireMockConfiguration port(int portNumber);" +
" public native WireMockConfiguration dynamicPort();" +
" public native WireMockConfiguration httpsPort(Integer httpsPort);" +
" public native WireMockConfiguration dynamicHttpsPort();" +
"}",
//language=java
"" +
"package com.github.tomakehurst.wiremock.core;" +
"public interface Options {}");
if (arg instanceof J.Empty) {
String newWiremockExtension = "WireMockExtension.newInstance().build()";
return JavaTemplate.builder(newWiremockExtension)
.imports("com.github.tomakehurst.wiremock.junit5.WireMockExtension")
.javaParser(wiremockParser)
.build()
.apply(getCursor(), n.getCoordinates().replace());
} else {
JavaType.Class optsType = JavaType.ShallowClass.build("com.github.tomakehurst.wiremock.core.Options");
if (TypeUtils.isAssignableTo(optsType, arg.getType())) {
String newWiremockExtension = "WireMockExtension.newInstance()" +
".options(#{any(com.github.tomakehurst.wiremock.core.Options)})";
if (n.getArguments().size() > 1) {
newWiremockExtension += ".failOnUnmatchedRequests(#{any(boolean)})";
return JavaTemplate.builder(newWiremockExtension + ".build()")
.imports("com.github.tomakehurst.wiremock.junit5.WireMockExtension")
.javaParser(wiremockParser)
.build()
.apply(
updateCursor(n),
n.getCoordinates().replace(),
arg,
n.getArguments().get(1)
);
} else {
return JavaTemplate.builder(newWiremockExtension + ".build()")
.imports("com.github.tomakehurst.wiremock.junit5.WireMockExtension")
.javaParser(wiremockParser)
.build()
.apply(updateCursor(n), n.getCoordinates().replace(), arg);
}
} else {
maybeAddImport("com.github.tomakehurst.wiremock.core.WireMockConfiguration");
String newWiremockExtension = "WireMockExtension.newInstance().options(WireMockConfiguration.options().port(#{any(int)})";
if (n.getArguments().size() > 1) {
newWiremockExtension += ".httpsPort(#{any(java.lang.Integer)})";
return JavaTemplate.builder(newWiremockExtension + ").build()")
.imports("com.github.tomakehurst.wiremock.core.WireMockConfiguration")
.imports("com.github.tomakehurst.wiremock.junit5.WireMockExtension")
.javaParser(wiremockParser)
.build()
.apply(
updateCursor(n),
n.getCoordinates().replace(),
arg,
n.getArguments().get(1)
);
} else {
return JavaTemplate.builder(newWiremockExtension + ").build()")
.imports("com.github.tomakehurst.wiremock.core.WireMockConfiguration")
.imports("com.github.tomakehurst.wiremock.junit5.WireMockExtension")
.javaParser(wiremockParser)
.build()
.apply(updateCursor(n), n.getCoordinates().replace(), arg);
}
}
}
}
return n;
}
});
}
@Override
public List getRecipeList() {
return singletonList(new UpgradeDependencyVersion("com.github.tomakehurst", "wiremock*",
"2.x", null, true, emptyList()));
}
}