List<TestFramework> (PsiClass targetClass) { List<TestFramework> frameworks = ContainerUtil.filter(TestFramework.EXTENSION_NAME.getExtensionList(), framework -> TestFrameworks.isSuitableByLanguage(targetClass, framework) ); Project project = targetClass.getProject(); List<TestFramework> result = new SmartList<>(); for (TestFramework framework : frameworks) { if (isAvailableFor(project, framework)) { if (framework.isTestClass(targetClass)) { return Collections.singletonList(framework); } if (framework.isPotentialTestClass(targetClass)) { result.add(framework); } } } return result; }