37 lines
995 B
Java
37 lines
995 B
Java
package com.baeldung.pmd;
|
|
|
|
import net.sourceforge.pmd.lang.java.ast.ASTClassOrInterfaceDeclaration;
|
|
import net.sourceforge.pmd.lang.java.rule.AbstractJavaRule;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.List;
|
|
import java.util.Objects;
|
|
|
|
public class UnitTestNamingConventionRule extends AbstractJavaRule {
|
|
|
|
private static List<String> allowedEndings = Arrays.asList(
|
|
"IntegrationTest",
|
|
"IntTest",
|
|
"ManualTest",
|
|
"JdbcTest",
|
|
"LiveTest",
|
|
"UnitTest",
|
|
"jmhTest");
|
|
|
|
public Object visit(ASTClassOrInterfaceDeclaration node, Object data) {
|
|
String className = node.getImage();
|
|
Objects.requireNonNull(className);
|
|
|
|
if (className.endsWith("SpringContextTest")) {
|
|
return data;
|
|
}
|
|
|
|
if (className.endsWith("Tests")
|
|
|| (className.endsWith("Test") && allowedEndings.stream().noneMatch(className::endsWith))) {
|
|
addViolation(data, node);
|
|
}
|
|
|
|
return data;
|
|
}
|
|
}
|