extend test framework: isTestMethodOrConfig

java/openapi/src/com/intellij/codeInsight/TestFramework.java

@@ -27,4 +27,6 @@ import com.intellij.util.IncorrectOperationException;
 public interface TestFramework {
   boolean isTestKlass(PsiClass psiClass);
   PsiMethod findSetUpMethod(PsiClass psiClass) throws IncorrectOperationException;
+
+  boolean isTestMethodOrConfig(PsiMethod psiMethod);
 }

plugins/junit/src/com/intellij/execution/junit/JUnitTestFramework.java

@@ -66,4 +66,8 @@ public class JUnitTestFramework implements TestFramework {
     }
     return inClass;
   }
+
+  public boolean isTestMethodOrConfig(PsiMethod psiMethod) {
+    return JUnitUtil.isTestMethodOrConfig(psiMethod);
+  }
 }

plugins/testng/src/com/theoryinpractice/testng/util/TestNGUtil.java

@@ -477,6 +477,10 @@ public class TestNGUtil implements TestFramework
     return inClass;
   }
 
+  public boolean isTestMethodOrConfig(PsiMethod psiMethod) {
+    return hasTest(psiMethod) || hasConfig(psiMethod);
+  }
+
   public static boolean checkTestNGInClasspath(PsiElement psiElement) {
     final Project project = psiElement.getProject();
     final PsiManager manager = PsiManager.getInstance(project);