diff --git a/java/java-impl/src/com/intellij/testIntegration/createTest/CreateTestAction.java b/java/java-impl/src/com/intellij/testIntegration/createTest/CreateTestAction.java index 5ee52555c459..0f92d485b4ed 100644 --- a/java/java-impl/src/com/intellij/testIntegration/createTest/CreateTestAction.java +++ b/java/java-impl/src/com/intellij/testIntegration/createTest/CreateTestAction.java @@ -129,20 +129,15 @@ public class CreateTestAction extends PsiElementBaseIntentionAction { } protected static void checkForTestRoots(Module srcModule, Set testFolders) { - checkForTestRoots(srcModule, testFolders, new HashSet()); - } - - private static void checkForTestRoots(final Module srcModule, final Set testFolders, final Set processed) { - final boolean isFirst = processed.isEmpty(); - if (!processed.add(srcModule)) return; - testFolders.addAll(ModuleRootManager.getInstance(srcModule).getSourceRoots(JavaSourceRootType.TEST_SOURCE)); - if (isFirst && !testFolders.isEmpty()) return; + //create test in the same module + if (!testFolders.isEmpty()) return; + //suggest to choose from all dependencies modules final HashSet modules = new HashSet(); ModuleUtilCore.collectModulesDependsOn(srcModule, modules); for (Module module : modules) { - checkForTestRoots(module, testFolders, processed); + testFolders.addAll(ModuleRootManager.getInstance(module).getSourceRoots(JavaSourceRootType.TEST_SOURCE)); } }