Add filter for methods annotated with @groovy.transform.Generated (#610)

4 files changed, 114 insertions, 19 deletions

diff --git a/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/AbstractAnnotatedMethodFilter.java b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/AbstractAnnotatedMethodFilter.java
new file mode 100644
index 00000000..75e33093
--- /dev/null
+++ b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/AbstractAnnotatedMethodFilter.java
@@ -0,0 +1,66 @@
+/*******************************************************************************
+ * Copyright (c) 2009, 2017 Mountainminds GmbH & Co. KG and Contributors
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *    Marc R. Hoffmann - initial API and implementation
+ *
+ *******************************************************************************/
+package org.jacoco.core.internal.analysis.filter;
+
+import java.util.List;
+
+import org.objectweb.asm.tree.AnnotationNode;
+import org.objectweb.asm.tree.MethodNode;
+
+/**
+ * Filters annotated methods.
+ */
+abstract class AbstractAnnotatedMethodFilter implements IFilter {
+	private final String descType;
+
+	/**
+	 * Configures a new filter instance.
+	 * 
+	 * @param annotationType
+	 *            VM type of the annotation
+	 */
+	protected AbstractAnnotatedMethodFilter(final String annotationType) {
+		this.descType = "L" + annotationType + ";";
+	}
+
+	public void filter(final String className, final String superClassName,
+			final MethodNode methodNode, final IFilterOutput output) {
+		if (hasAnnotation(methodNode)) {
+			output.ignore(methodNode.instructions.getFirst(),
+					methodNode.instructions.getLast());
+		}
+	}
+
+	private boolean hasAnnotation(final MethodNode methodNode) {
+		final List<AnnotationNode> annotations = getAnnotations(methodNode);
+		if (annotations != null) {
+			for (final AnnotationNode annotation : annotations) {
+				if (descType.equals(annotation.desc)) {
+					return true;
+				}
+			}
+		}
+		return false;
+	}
+
+	/**
+	 * Retrieves the annotations to search from a method. Depending on the
+	 * retention of the annotation this is either
+	 * <code>visibleAnnotations</code> or <code>invisibleAnnotations</code>.
+	 * 
+	 * @param methodNode
+	 *            method to retrieve annotations from
+	 * @return list of annotations
+	 */
+	abstract List<AnnotationNode> getAnnotations(final MethodNode methodNode);
+
+}
diff --git a/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/Filters.java b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/Filters.java
index e2c26d92..4b2a64a9 100644
--- a/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/Filters.java
+++ b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/Filters.java
@@ -30,11 +30,12 @@ public final class Filters implements IFilter {
 			new SyntheticFilter(), new SynchronizedFilter(),
 			new TryWithResourcesJavacFilter(), new TryWithResourcesEcjFilter(),
 			new PrivateEmptyNoArgConstructorFilter(),
-			new StringSwitchJavacFilter(), new LombokGeneratedFilter());
+			new StringSwitchJavacFilter(), new LombokGeneratedFilter(),
+			new GroovyGeneratedFilter());
 
 	private final IFilter[] filters;
 
-	private Filters(IFilter... filters) {
+	private Filters(final IFilter... filters) {
 		this.filters = filters;
 	}
 
diff --git a/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/GroovyGeneratedFilter.java b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/GroovyGeneratedFilter.java
new file mode 100644
index 00000000..353868d5
--- /dev/null
+++ b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/GroovyGeneratedFilter.java
@@ -0,0 +1,36 @@
+/*******************************************************************************
+ * Copyright (c) 2009, 2017 Mountainminds GmbH & Co. KG and Contributors
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *    Marc R. Hoffmann - initial API and implementation
+ *
+ *******************************************************************************/
+package org.jacoco.core.internal.analysis.filter;
+
+import java.util.List;
+
+import org.objectweb.asm.tree.AnnotationNode;
+import org.objectweb.asm.tree.MethodNode;
+
+/**
+ * Filters methods annotated with <code>@groovy.transform.Generated</code>.
+ */
+public final class GroovyGeneratedFilter extends AbstractAnnotatedMethodFilter {
+
+	/**
+	 * New filter.
+	 */
+	public GroovyGeneratedFilter() {
+		super("groovy/transform/Generated");
+	}
+
+	@Override
+	List<AnnotationNode> getAnnotations(final MethodNode methodNode) {
+		return methodNode.visibleAnnotations;
+	}
+
+}
diff --git a/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/LombokGeneratedFilter.java b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/LombokGeneratedFilter.java
index 98575e80..62021ae7 100644
--- a/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/LombokGeneratedFilter.java
+++ b/org.jacoco.core/src/org/jacoco/core/internal/analysis/filter/LombokGeneratedFilter.java
@@ -19,26 +19,18 @@ import org.objectweb.asm.tree.MethodNode;
 /**
  * Filters methods annotated with <code>@lombok.Generated</code>.
  */
-public final class LombokGeneratedFilter implements IFilter {
+public final class LombokGeneratedFilter extends AbstractAnnotatedMethodFilter {
 
-	public void filter(final String className, final String superClassName,
-			final MethodNode methodNode, final IFilterOutput output) {
-		if (hasLombokGeneratedAnnotation(methodNode)) {
-			output.ignore(methodNode.instructions.getFirst(),
-					methodNode.instructions.getLast());
-		}
+	/**
+	 * New filter.
+	 */
+	public LombokGeneratedFilter() {
+		super("lombok/Generated");
 	}
 
-	private boolean hasLombokGeneratedAnnotation(final MethodNode methodNode) {
-		final List<AnnotationNode> runtimeInvisibleAnnotations = methodNode.invisibleAnnotations;
-		if (runtimeInvisibleAnnotations != null) {
-			for (final AnnotationNode annotation : runtimeInvisibleAnnotations) {
-				if ("Llombok/Generated;".equals(annotation.desc)) {
-					return true;
-				}
-			}
-		}
-		return false;
+	@Override
+	List<AnnotationNode> getAnnotations(final MethodNode methodNode) {
+		return methodNode.invisibleAnnotations;
 	}
 
 }