plugins/junit/src/com/intellij/execution/junit/TestDirectory.java

   1 /*
   2  * Copyright 2000-2011 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.execution.junit;
  17
  18 import com.intellij.execution.CantRunException;
  19 import com.intellij.execution.configurations.*;
  20 import com.intellij.execution.testframework.SourceScope;
  21 import com.intellij.execution.util.JavaParametersUtil;
  22 import com.intellij.execution.util.ProgramParametersUtil;
  23 import com.intellij.openapi.module.Module;
  24 import com.intellij.openapi.project.Project;
  25 import com.intellij.openapi.util.io.FileUtil;
  26 import com.intellij.openapi.vfs.LocalFileSystem;
  27 import com.intellij.openapi.vfs.VirtualFile;
  28 import com.intellij.psi.JavaDirectoryService;
  29 import com.intellij.psi.PsiDirectory;
  30 import com.intellij.psi.PsiManager;
  31 import com.intellij.psi.PsiPackage;
  32 import com.intellij.psi.search.GlobalSearchScope;
  33 import com.intellij.psi.search.GlobalSearchScopes;
  34
  35 import java.util.Collection;
  36
  37 /**
  38 * User: anna
  39 * Date: 4/21/11
  40 */
  41 class TestDirectory extends TestPackage {
  42   private final Project myProject;
  43
  44   public TestDirectory(Project project,
  45                        JUnitConfiguration configuration,
  46                        RunnerSettings runnerSettings,
  47                        ConfigurationPerRunnerSettings configurationSettings) {
  48     super(project, configuration, runnerSettings, configurationSettings);
  49     myProject = project;
  50   }
  51
  52   @Override
  53   public SourceScope getSourceScope() {
  54     final String dirName = myConfiguration.getPersistentData().getDirName();
  55     final VirtualFile file = LocalFileSystem.getInstance().findFileByPath(FileUtil.toSystemIndependentName(dirName));
  56     final GlobalSearchScope globalSearchScope = file == null ? GlobalSearchScope.EMPTY_SCOPE : GlobalSearchScopes
  57         .directoryScope(myProject, file, true);
  58     return new SourceScope() {
  59       @Override
  60       public GlobalSearchScope getGlobalSearchScope() {
  61         return globalSearchScope;
  62       }
  63
  64       @Override
  65       public Project getProject() {
  66         return myProject;
  67       }
  68
  69       @Override
  70       public GlobalSearchScope getLibrariesScope() {
  71         final Module module = myConfiguration.getConfigurationModule().getModule();
  72         LOG.assertTrue(module != null);
  73         return GlobalSearchScope.moduleWithLibrariesScope(module);
  74       }
  75
  76       @Override
  77       public Module[] getModulesToCompile() {
  78         final Collection<Module> validModules = myConfiguration.getValidModules();
  79         return validModules.toArray(new Module[validModules.size()]);
  80       }
  81     };
  82   }
  83
  84   @Override
  85   public void checkConfiguration() throws RuntimeConfigurationException {
  86     JavaParametersUtil.checkAlternativeJRE(myConfiguration);
  87     ProgramParametersUtil.checkWorkingDirectoryExist(myConfiguration, myConfiguration.getProject(), myConfiguration.getConfigurationModule().getModule());
  88     final String dirName = myConfiguration.getPersistentData().getDirName();
  89     if (dirName == null || dirName.isEmpty()) {
  90       throw new RuntimeConfigurationError("Directory is not specified");
  91     }
  92     final VirtualFile file = LocalFileSystem.getInstance().findFileByPath(FileUtil.toSystemIndependentName(dirName));
  93     if (file == null) {
  94       throw new RuntimeConfigurationWarning("Directory \'" + dirName + "\' is not found");
  95     }
  96     final Module module = myConfiguration.getConfigurationModule().getModule();
  97     if (module == null) {
  98       throw new RuntimeConfigurationError("Module to choose classpath from is not specified");
  99     }
 100   }
 101
 102   @Override
 103   protected PsiPackage getPackage(JUnitConfiguration.Data data) throws CantRunException {
 104     final String dirName = data.getDirName();
 105     final VirtualFile file = LocalFileSystem.getInstance().findFileByPath(FileUtil.toSystemIndependentName(dirName));
 106     if (file == null) {
 107       throw new CantRunException("Directory \'" + dirName + "\' is not found");
 108     }
 109     final PsiDirectory directory = PsiManager.getInstance(myProject).findDirectory(file);
 110     if (directory == null) {
 111       throw new CantRunException("Directory \'" + dirName + "\' is not found");
 112     }
 113     final PsiPackage aPackage = JavaDirectoryService.getInstance().getPackage(directory);
 114     if (aPackage == null) {
 115       throw new CantRunException("Package not found in directory");
 116     }
 117     return aPackage;
 118   }
 119 }