From be01c64a9e01f58712d75d7ae19db09f9faaf69c Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Wed, 10 May 2017 13:07:06 +0300
Subject: [PATCH] junit: ensure isTest check is performed (IDEA-172627)

---
 .../impl/src/com/intellij/execution/junit/JUnitUtil.java        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 91bcef935b00..4f1846b82b4e 100644
--- a/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java
+++ b/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java
@@ -434,7 +434,7 @@ public class JUnitUtil {
     PsiElement element = location.getPsiElement();
     if (element instanceof PsiClassOwner) {
       PsiClass[] classes = ((PsiClassOwner)element).getClasses();
-      if (classes.length == 1) return classes[0];
+      if (classes.length == 1 && isTestClass(classes[0], false, true)) return classes[0];
     }
     return null;
   }