diff --git a/plugins/junit/src/com/intellij/execution/junit/JUnit3Framework.java b/plugins/junit/src/com/intellij/execution/junit/JUnit3Framework.java index 03e20d55d74a..a965aedcddf5 100644 --- a/plugins/junit/src/com/intellij/execution/junit/JUnit3Framework.java +++ b/plugins/junit/src/com/intellij/execution/junit/JUnit3Framework.java @@ -15,6 +15,7 @@ */ package com.intellij.execution.junit; +import com.intellij.execution.junit2.info.MethodLocation; import com.intellij.icons.AllIcons; import com.intellij.ide.fileTemplates.FileTemplateDescriptor; import com.intellij.openapi.roots.ExternalLibraryDescriptor; @@ -48,6 +49,11 @@ public class JUnit3Framework extends JavaTestFramework { return true; } + @Override + public boolean isTestMethod(PsiMethod method, PsiClass myClass) { + return JUnitUtil.isTestMethod(MethodLocation.elementInClass(method, myClass)); + } + @NotNull @Override public Icon getIcon() {