de.dagere.kopeme.parsing.JUnitParseUtil Maven / Gradle / Ivy

Go to download
package de.dagere.kopeme.parsing;

import java.io.File;
import java.io.FileNotFoundException;
import java.util.LinkedList;
import java.util.List;
import java.util.Optional;

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

/**
 * Helps parsing JUnit classes for test execution
 * 
 * @author reichelt
 *
 */
public class JUnitParseUtil {

   private final static ThreadLocal JAVA_PARSER = new ThreadLocal() {
      protected JavaParser initialValue() {
         return new JavaParser();
      };
   };

   public synchronized static CompilationUnit parse(final File file) throws FileNotFoundException {
      final JavaParser parser = JAVA_PARSER.get();
      final Optional result = parser.parse(file).getResult();
      return result.get();
   }

   public List getAnnotatedMethods(File clazzFile, final String fqnAnnotationName, String annotationName) throws FileNotFoundException {
      CompilationUnit unit = parse(clazzFile);
      TypeDeclaration typeDeclaration = unit.getPrimaryType().get();
      if (typeDeclaration instanceof ClassOrInterfaceDeclaration) {
         return getAnnotatedMethods((ClassOrInterfaceDeclaration) typeDeclaration, fqnAnnotationName, annotationName);
      } else {
         throw new RuntimeException("Classfile must contain class!");
      }
   }

   /**
    * Identifies all methods which are tests
    * 
    * @param clazz Class which should be analyzed
    * @param fqnAnnotationName Full qualified name of the annotation, e.g. de.dagere.kopeme.annotations.PerformanceTest
    * @param annotationName Simple name, e.g. PerformanceTest
    * @return List of annotated methods
    */
   public static List getAnnotatedMethods(final ClassOrInterfaceDeclaration clazz, final String fqnAnnotationName, String annotationName) {
      List methods = new LinkedList<>();
      for (final MethodDeclaration method : clazz.getMethods()) {
         boolean found = false;
         for (final AnnotationExpr annotation : method.getAnnotations()) {
            final String currentName = annotation.getNameAsString();

            if (currentName.equals(fqnAnnotationName) || currentName.equals(annotationName)) {
               found = true;
            }
         }
         if (found) {
            methods.add(method.getNameAsString());
         }
      }
      return methods;
   }
}