BAEL-20537: Migrate custom-pmd module to the com.baeldung package

2019-12-24 10:44:49 +01:00
parent e86eeeb593
commit 736b684c24
2 changed files with 2 additions and 2 deletions
@@ -0,0 +1,36 @@
+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;
+    }
+}