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
There is a newer version: 0.4.0
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;

public class TestMethodFinder {
   public static List findJUnit5TestMethods(final ClassOrInterfaceDeclaration clazz) {
      List testMethods = new LinkedList<>();
      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;
            }
         }
         if (testFound && !performanceTestFound) {
            testMethods.add(method);
         }
      }
      return testMethods;
   }
   
   public static List findBeforeMethods(final ClassOrInterfaceDeclaration clazz){
      String[] annotations = new String[] {"org.junit.Before", "Before", 
            "org.junit.jupiter.api.BeforeEach", "BeforeEach",
            "org.junit.jupiter.api.BeforeAll", "BeforeAll"};
      List beforeMethods = findAnnotation(clazz, annotations);
      return beforeMethods;
   }
   
   public static List findAfterMethods(final ClassOrInterfaceDeclaration clazz){
      String[] annotations = new String[] {"org.junit.After", "After", 
            "org.junit.jupiter.api.AfterEach", "AfterEach",
            "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()) {
         boolean beforeFound = false;
         for (final AnnotationExpr annotation : method.getAnnotations()) {
            final String currentName = annotation.getNameAsString();
            for (String searchedAnnotation : annotations) {
               if (currentName.equals(searchedAnnotation)) {
                  beforeFound = true;
               }
            }
         }
         
         if (beforeFound) {
            annotatedMethods.add(method);
         }
      }
      return annotatedMethods;
   }
   
   public static List findJUnit4TestMethods(final ClassOrInterfaceDeclaration clazz) {
      List testMethods = new LinkedList<>();
      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;
            }
         }
         
         if (testFound && !performanceTestFound) {
            testMethods.add(method);
         }
      }
      return testMethods;
   }
}