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;
    }
}