Produce test for subpackages with tests

Author: zolotov

src/com/goide/runconfig/testing/GoTestFramework.java

@@ -40,7 +40,7 @@ public static GoTestFramework fromName(@Nullable String name) {
 
   public abstract boolean isAvailable(@Nullable Module module);
 
-  public abstract boolean isAvailableOnFile(@NotNull PsiFile file);
+  public abstract boolean isAvailableOnFile(@Nullable PsiFile file);
 
   @NotNull
   protected abstract GoTestRunningState newRunningState(@NotNull ExecutionEnvironment env,

src/com/goide/runconfig/testing/GoTestRunConfigurationProducerBase.java

@@ -24,14 +24,19 @@
 import com.intellij.execution.actions.RunConfigurationProducer;
 import com.intellij.openapi.module.Module;
 import com.intellij.openapi.module.ModuleUtilCore;
+import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.Comparing;
 import com.intellij.openapi.util.Ref;
 import com.intellij.openapi.util.io.FileUtil;
 import com.intellij.openapi.util.text.StringUtil;
+import com.intellij.openapi.vfs.VfsUtilCore;
+import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.PsiDirectory;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.PsiFile;
+import com.intellij.psi.PsiManager;
 import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.util.Processor;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -45,32 +50,38 @@ protected GoTestRunConfigurationProducerBase(@NotNull GoTestFramework framework)
   }
 
   @Override
-  protected boolean setupConfigurationFromContext(@NotNull GoTestRunConfiguration configuration,
+  protected boolean setupConfigurationFromContext(@NotNull final GoTestRunConfiguration configuration,
                                                   ConfigurationContext context,
                                                   Ref sourceElement) {
-    PsiElement contextElement = GoRunUtil.getContextElement(context);
+    final PsiElement contextElement = GoRunUtil.getContextElement(context);
     if (contextElement == null) {
       return false;
     }
 
     Module module = ModuleUtilCore.findModuleForPsiElement(contextElement);
-    if (module == null || !GoSdkService.getInstance(module.getProject()).isGoModule(module)) return false;
+    final Project project = contextElement.getProject();
+    if (module == null || !GoSdkService.getInstance(project).isGoModule(module)) return false;
     if (!myFramework.isAvailable(module)) return false;
 
     configuration.setModule(module);
     configuration.setTestFramework(myFramework);
     if (contextElement instanceof PsiDirectory) {
-      for (PsiFile file : ((PsiDirectory)contextElement).getFiles()) {
-        if (myFramework.isAvailableOnFile(file)) {
-          configuration.setName(getPackageConfigurationName(((PsiDirectory)contextElement).getName()));
-          configuration.setKind(GoTestRunConfiguration.Kind.DIRECTORY);
-          String directoryPath = ((PsiDirectory)contextElement).getVirtualFile().getPath();
-          configuration.setDirectoryPath(directoryPath);
-          configuration.setWorkingDirectory(directoryPath);
+      VirtualFile virtualFile = ((PsiDirectory)contextElement).getVirtualFile();
+      return !VfsUtilCore.processFilesRecursively(virtualFile, new Processor<VirtualFile>() {
+        @Override
+        public boolean process(VirtualFile file) {
+          PsiFile psiFile = PsiManager.getInstance(project).findFile(file);
+          if (myFramework.isAvailableOnFile(psiFile)) {
+            configuration.setName(getPackageConfigurationName(((PsiDirectory)contextElement).getName()));
+            configuration.setKind(GoTestRunConfiguration.Kind.DIRECTORY);
+            String directoryPath = ((PsiDirectory)contextElement).getVirtualFile().getPath();
+            configuration.setDirectoryPath(directoryPath);
+            configuration.setWorkingDirectory(directoryPath);
+            return false;
+          }
           return true;
         }
-      }
-      return false;
+      });
     }
     else {
       PsiFile file = contextElement.getContainingFile();

src/com/goide/runconfig/testing/frameworks/gocheck/GocheckFramework.java

@@ -78,7 +78,7 @@ private static boolean isGoCheckImportPath(String importPath) {
   }
 
   @Override
-  public boolean isAvailableOnFile(@NotNull PsiFile file) {
+  public boolean isAvailableOnFile(@Nullable PsiFile file) {
     if (!GoTestFinder.isTestFile(file)) {
       return false;
     }

src/com/goide/runconfig/testing/frameworks/gotest/GotestFramework.java

@@ -47,7 +47,7 @@ public boolean isAvailable(@Nullable Module module) {
   }
 
   @Override
-  public boolean isAvailableOnFile(@NotNull PsiFile file) {
+  public boolean isAvailableOnFile(@Nullable PsiFile file) {
     return GoTestFinder.isTestFile(file);
   }