| /* |
| * Copyright 2000-2011 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.intellij.execution.junit; |
| |
| import com.intellij.execution.CantRunException; |
| import com.intellij.execution.configurations.RuntimeConfigurationError; |
| import com.intellij.execution.configurations.RuntimeConfigurationException; |
| import com.intellij.execution.configurations.RuntimeConfigurationWarning; |
| import com.intellij.execution.runners.ExecutionEnvironment; |
| import com.intellij.execution.testframework.SourceScope; |
| import com.intellij.execution.util.JavaParametersUtil; |
| import com.intellij.execution.util.ProgramParametersUtil; |
| import com.intellij.openapi.module.Module; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.util.Comparing; |
| import com.intellij.openapi.util.io.FileUtil; |
| import com.intellij.openapi.vfs.LocalFileSystem; |
| import com.intellij.openapi.vfs.VirtualFile; |
| import com.intellij.psi.*; |
| import com.intellij.psi.search.GlobalSearchScope; |
| import com.intellij.psi.search.GlobalSearchScopesCore; |
| |
| import java.util.Collection; |
| |
| /** |
| * User: anna |
| * Date: 4/21/11 |
| */ |
| class TestDirectory extends TestPackage { |
| public TestDirectory(JUnitConfiguration configuration, ExecutionEnvironment environment) { |
| super(configuration, environment); |
| } |
| |
| @Override |
| public SourceScope getSourceScope() { |
| final String dirName = myConfiguration.getPersistentData().getDirName(); |
| final VirtualFile file = LocalFileSystem.getInstance().findFileByPath(FileUtil.toSystemIndependentName(dirName)); |
| final GlobalSearchScope globalSearchScope = file == null ? GlobalSearchScope.EMPTY_SCOPE : GlobalSearchScopesCore.directoryScope(myEnvironment.getProject(), file, true); |
| return new SourceScope() { |
| @Override |
| public GlobalSearchScope getGlobalSearchScope() { |
| return globalSearchScope; |
| } |
| |
| @Override |
| public Project getProject() { |
| return myEnvironment.getProject(); |
| } |
| |
| @Override |
| public GlobalSearchScope getLibrariesScope() { |
| final Module module = myConfiguration.getConfigurationModule().getModule(); |
| return module != null ? GlobalSearchScope.moduleWithLibrariesScope(module) : GlobalSearchScope.allScope(myEnvironment.getProject()); |
| } |
| |
| @Override |
| public Module[] getModulesToCompile() { |
| final Collection<Module> validModules = myConfiguration.getValidModules(); |
| return validModules.toArray(new Module[validModules.size()]); |
| } |
| }; |
| } |
| |
| @Override |
| public void checkConfiguration() throws RuntimeConfigurationException { |
| JavaParametersUtil.checkAlternativeJRE(myConfiguration); |
| ProgramParametersUtil.checkWorkingDirectoryExist(myConfiguration, myConfiguration.getProject(), myConfiguration.getConfigurationModule().getModule()); |
| final String dirName = myConfiguration.getPersistentData().getDirName(); |
| if (dirName == null || dirName.isEmpty()) { |
| throw new RuntimeConfigurationError("Directory is not specified"); |
| } |
| final VirtualFile file = LocalFileSystem.getInstance().findFileByPath(FileUtil.toSystemIndependentName(dirName)); |
| if (file == null) { |
| throw new RuntimeConfigurationWarning("Directory \'" + dirName + "\' is not found"); |
| } |
| final Module module = myConfiguration.getConfigurationModule().getModule(); |
| if (module == null) { |
| throw new RuntimeConfigurationError("Module to choose classpath from is not specified"); |
| } |
| } |
| |
| @Override |
| protected PsiPackage getPackage(JUnitConfiguration.Data data) throws CantRunException { |
| final String dirName = data.getDirName(); |
| final VirtualFile file = LocalFileSystem.getInstance().findFileByPath(FileUtil.toSystemIndependentName(dirName)); |
| if (file == null) { |
| throw new CantRunException("Directory \'" + dirName + "\' is not found"); |
| } |
| final PsiDirectory directory = PsiManager.getInstance(myEnvironment.getProject()).findDirectory(file); |
| if (directory == null) { |
| throw new CantRunException("Directory \'" + dirName + "\' is not found"); |
| } |
| final PsiPackage aPackage = JavaDirectoryService.getInstance().getPackage(directory); |
| if (aPackage == null) { |
| throw new CantRunException("Package not found in directory"); |
| } |
| return aPackage; |
| } |
| |
| @Override |
| public boolean isConfiguredByElement(JUnitConfiguration configuration, |
| PsiClass testClass, |
| PsiMethod testMethod, |
| PsiPackage testPackage, |
| PsiDirectory testDir) { |
| if (JUnitConfiguration.TEST_DIRECTORY.equals(configuration.getPersistentData().TEST_OBJECT) && testDir != null) { |
| if (Comparing.strEqual(FileUtil.toSystemIndependentName(configuration.getPersistentData().getDirName()), |
| testDir.getVirtualFile().getPath())) { |
| return true; |
| } |
| } |
| return false; |
| } |
| } |