From 2291f2dedd3f76a1d8cc587cb187283b61b832d2 Mon Sep 17 00:00:00 2001
From: Andreas Gudian <agudian@apache.org>
Date: Sat, 7 Nov 2015 21:04:45 +0100
Subject: [PATCH] #667 Fix execution of single test methods in some cases.

---
 .../runner/AnnotationProcessorTestRunner.java | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/processor/src/test/java/org/mapstruct/ap/testutil/runner/AnnotationProcessorTestRunner.java b/processor/src/test/java/org/mapstruct/ap/testutil/runner/AnnotationProcessorTestRunner.java
index 400d31a72..744007ad3 100644
--- a/processor/src/test/java/org/mapstruct/ap/testutil/runner/AnnotationProcessorTestRunner.java
+++ b/processor/src/test/java/org/mapstruct/ap/testutil/runner/AnnotationProcessorTestRunner.java
@@ -18,11 +18,14 @@
  */
 package org.mapstruct.ap.testutil.runner;
 
+import java.lang.annotation.Annotation;
 import java.util.Arrays;
 import java.util.List;
 
 import org.junit.runner.Description;
 import org.junit.runner.Runner;
+import org.junit.runner.manipulation.Filter;
+import org.junit.runner.manipulation.NoTestsRemainException;
 import org.junit.runner.notification.RunNotifier;
 import org.junit.runners.BlockJUnit4ClassRunner;
 import org.junit.runners.ParentRunner;
@@ -76,4 +79,61 @@ public class AnnotationProcessorTestRunner extends ParentRunner<Runner> {
     protected void runChild(Runner child, RunNotifier notifier) {
         child.run( notifier );
     }
+
+    @Override
+    public void filter(Filter filter) throws NoTestsRemainException {
+        super.filter( new FilterDecorator( filter ) );
+    }
+
+    /**
+     * Allows to only execute selected methods, even if the executing framework is not aware of parameterized tests
+     * (e.g. some versions of IntelliJ, Netbeans, Eclipse).
+     */
+    private static final class FilterDecorator extends Filter {
+        private final Filter delegate;
+
+        private FilterDecorator(Filter delegate) {
+            this.delegate = delegate;
+        }
+
+        @Override
+        public boolean shouldRun(Description description) {
+            boolean shouldRun = delegate.shouldRun( description );
+            if ( !shouldRun ) {
+                return delegate.shouldRun( withoutParameterizedName( description ) );
+            }
+
+            return shouldRun;
+        }
+
+        @Override
+        public String describe() {
+            return delegate.describe();
+        }
+
+        private Description withoutParameterizedName(Description description) {
+            String cleanDispayName = removeParameter( description.getDisplayName() );
+            Description cleanDescription =
+                Description.createSuiteDescription(
+                    cleanDispayName,
+                    description.getAnnotations().toArray( new Annotation[description.getAnnotations().size()] ) );
+
+            for ( Description child : description.getChildren() ) {
+                cleanDescription.addChild( withoutParameterizedName( child ) );
+            }
+
+            return cleanDescription;
+        }
+
+        private String removeParameter(String name) {
+            if ( name.startsWith( "[" ) ) {
+                return name;
+            }
+
+            // remove "[compiler]" from "method[compiler](class)"
+            int open = name.indexOf( '[' );
+            int close = name.indexOf( ']' ) + 1;
+            return name.substring( 0, open ) + name.substring( close );
+        }
+    }
 }