diff --git a/java/execution/impl/src/com/intellij/execution/testframework/AbstractPatternBasedConfigurationProducer.java b/java/execution/impl/src/com/intellij/execution/testframework/AbstractPatternBasedConfigurationProducer.java index b1186ab0efc3..db9b42a863eb 100644 --- a/java/execution/impl/src/com/intellij/execution/testframework/AbstractPatternBasedConfigurationProducer.java +++ b/java/execution/impl/src/com/intellij/execution/testframework/AbstractPatternBasedConfigurationProducer.java @@ -21,6 +21,7 @@ import com.intellij.execution.actions.ConfigurationContext; import com.intellij.execution.configurations.ConfigurationType; import com.intellij.execution.configurations.ModuleBasedConfiguration; import com.intellij.execution.junit.JavaRunConfigurationProducerBase; +import com.intellij.execution.junit2.PsiMemberParameterizedLocation; import com.intellij.execution.junit2.info.MethodLocation; import com.intellij.openapi.actionSystem.CommonDataKeys; import com.intellij.openapi.actionSystem.DataContext; @@ -193,7 +194,9 @@ public abstract class AbstractPatternBasedConfigurationProducer(); } - if (location instanceof PsiMemberParameterizedLocation) { - final String paramSetName = ((PsiMemberParameterizedLocation)location).getParamSetName(); - if (paramSetName != null) { - strings.add(TestNGConfigurationProducer.getInvocationNumber(paramSetName)); - } - } + setupParameterName(location, strings); psiMethods.put(psiMethod, strings); } } } + + private static void setupParameterName(Location location, List strings) { + if (location instanceof PsiMemberParameterizedLocation) { + final String paramSetName = ((PsiMemberParameterizedLocation)location).getParamSetName(); + if (paramSetName != null) { + strings.add(TestNGConfigurationProducer.getInvocationNumber(paramSetName)); + } + } + } }