diff --git a/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java b/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java index e4de6db9a873..d7c34ea475ba 100644 --- a/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java +++ b/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java @@ -34,6 +34,7 @@ import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import java.util.Arrays; +import java.util.Collection; import java.util.Collections; import java.util.Iterator; @@ -52,6 +53,9 @@ public class JUnitUtil { public static final String PARAMETRIZED_PARAMETERS_ANNOTATION_NAME = "org.junit.runners.Parameterized.Parameters"; private static final String AFTER_CLASS_ANNOTATION_NAME = "org.junit.AfterClass"; private static final String BEFORE_CLASS_ANNOTATION_NAME = "org.junit.BeforeClass"; + private static final Collection CONFIGURATIONS_ANNOTATION_NAME = Collections.unmodifiableList( + Arrays.asList(DATA_POINT, AFTER_ANNOTATION_NAME, BEFORE_ANNOTATION_NAME, AFTER_CLASS_ANNOTATION_NAME, BEFORE_CLASS_ANNOTATION_NAME)); + public static final String PARAMETERIZED_CLASS_NAME = "org.junit.runners.Parameterized"; public static boolean isSuiteMethod(@NotNull PsiMethod psiMethod) { @@ -78,7 +82,7 @@ public class JUnitUtil { if (psiMethod.isConstructor()) return false; if (!psiMethod.hasModifierProperty(PsiModifier.PUBLIC)) return false; if (psiMethod.hasModifierProperty(PsiModifier.ABSTRACT)) return false; - if (AnnotationUtil.isAnnotated(psiMethod, DATA_POINT, false)) return false; + if (AnnotationUtil.isAnnotated(psiMethod, CONFIGURATIONS_ANNOTATION_NAME, false)) return false; if (AnnotationUtil.isAnnotated(aClass, RUN_WITH, true)) return true; if (psiMethod.getParameterList().getParametersCount() > 0) return false; if (psiMethod.hasModifierProperty(PsiModifier.STATIC) && SUITE_METHOD_NAME.equals(psiMethod.getName())) return false;