testDiscovery for TestNG: apply chosen state

This commit is contained in:
Anna Kozlova
2016-07-04 10:21:00 +03:00
parent 7157c0858d
commit 48954bbd29
@@ -112,11 +112,11 @@ public class TestNGConfigurationModel
data.MAIN_CLASS_NAME = "";
data.METHOD_NAME = "";
data.SUITE_NAME = "";
} else if (TestType.METHOD == type || TestType.CLASS == type) {
} else if (TestType.METHOD == type || TestType.CLASS == type || TestType.SOURCE == type) {
String className = getText(TestType.CLASS);
data.GROUP_NAME = "";
data.SUITE_NAME = "";
if (TestType.METHOD == type)
if (TestType.METHOD == type || TestType.SOURCE == type)
data.METHOD_NAME = getText(TestType.METHOD);
PsiClass psiClass = !getProject().isDefault() && !StringUtil.isEmptyOrSpaces(className) ? JUnitUtil.findPsiClass(className, module, getProject()) : null;