de.dagere.peass.testtransformation.TestMethodFinder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dependency Show documentation
The newest version!
package de.dagere.peass.testtransformation;

import java.util.LinkedList;
import java.util.List;

import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
import com.github.javaparser.ast.body.MethodDeclaration;
import com.github.javaparser.ast.expr.AnnotationExpr;

import de.dagere.kopeme.parsing.JUnitParseUtil;

public class TestMethodFinder {

   public static List findJUnit5TestMethods(final ClassOrInterfaceDeclaration clazz) {
      List testMethods = new LinkedList<>();

      if (JUnitParseUtil.isDeactivated(clazz)) {
         return testMethods;
      }

      for (final MethodDeclaration method : clazz.getMethods()) {
         boolean performanceTestFound = false;
         boolean testFound = false;
         for (final AnnotationExpr annotation : method.getAnnotations()) {
            final String currentName = annotation.getNameAsString();
            if (currentName.equals("de.dagere.kopeme.annotations.PerformanceTest") || currentName.equals("PerformanceTest")) {
               performanceTestFound = true;
            }
            if (currentName.equals("org.junit.Test") || currentName.equals("org.junit.jupiter.api.Test") || currentName.equals("Test")
                  || currentName.equals("ParameterizedTest") || currentName.equals("org.junit.jupiter.params.ParameterizedTest")) {
               testFound = true;
            }
         }
         boolean testIsDeactivated = JUnitParseUtil.isDeactivated(method);
         if (testFound && !performanceTestFound && !testIsDeactivated) {
            testMethods.add(method);
         }
      }
      return testMethods;
   }

   public static List findBeforeEachMethods(final ClassOrInterfaceDeclaration clazz) {
      String[] annotations = new String[] { "org.junit.Before", "Before",
            "org.junit.jupiter.api.BeforeEach", "BeforeEach", };
      List beforeMethods = findAnnotation(clazz, annotations);
      return beforeMethods;
   }

   public static List findBeforeAllMethods(final ClassOrInterfaceDeclaration clazz) {
      String[] annotations = new String[] { "org.junit.jupiter.api.BeforeClass", "BeforeClass",
            "org.junit.jupiter.api.BeforeAll", "BeforeAll" };
      List beforeMethods = findAnnotation(clazz, annotations);
      return beforeMethods;
   }

   public static List findAfterEachMethods(final ClassOrInterfaceDeclaration clazz) {
      String[] annotations = new String[] { "org.junit.After", "After",
            "org.junit.jupiter.api.AfterEach", "AfterEach"};
      List beforeMethods = findAnnotation(clazz, annotations);
      return beforeMethods;
   }

   public static List findAfterAllMethods(final ClassOrInterfaceDeclaration clazz) {
      String[] annotations = new String[] { "org.junit.jupiter.api.AfterAll", "AfterAll",
            "org.junit.jupiter.api.AfterAll", "AfterAll" };
      List beforeMethods = findAnnotation(clazz, annotations);
      return beforeMethods;
   }

   private static List findAnnotation(final ClassOrInterfaceDeclaration clazz, final String[] annotations) {
      List annotatedMethods = new LinkedList<>();
      for (final MethodDeclaration method : clazz.getMethods()) {
         AnnotationExpr foundAnnotation = null;
         for (final AnnotationExpr annotation : method.getAnnotations()) {
            final String currentName = annotation.getNameAsString();
            for (String searchedAnnotation : annotations) {
               if (currentName.equals(searchedAnnotation)) {
                  foundAnnotation = annotation;
               }
            }
         }

         if (foundAnnotation != null) {
            method.getAnnotations().remove(foundAnnotation);
            annotatedMethods.add(method);
         }
      }
      return annotatedMethods;
   }

   public static List findJUnit4TestMethods(final ClassOrInterfaceDeclaration clazz) {
      List testMethods = new LinkedList<>();

      if (JUnitParseUtil.isDeactivated(clazz)) {
         return testMethods;
      }

      for (final MethodDeclaration method : clazz.getMethods()) {
         boolean performanceTestFound = false;
         boolean testFound = false;
         for (final AnnotationExpr annotation : method.getAnnotations()) {
            final String currentName = annotation.getNameAsString();
            if (currentName.equals("de.dagere.kopeme.annotations.PerformanceTest") || currentName.equals("PerformanceTest")) {
               performanceTestFound = true;
            }
            if (currentName.equals("org.junit.Test") || currentName.equals("org.junit.jupiter.api.Test") || currentName.equals("Test")) {
               testFound = true;
            }
         }
         
         boolean testIsDeactivated = JUnitParseUtil.isDeactivated(method);

         if (testFound && !performanceTestFound && !testIsDeactivated) {
            testMethods.add(method);
         }
      }
      return testMethods;
   }
}