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

   1 /*
   2  * Copyright 2000-2012 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.openapi.roots.DependencyScope;
  19 import com.intellij.openapi.roots.LibraryDependencyScopeSuggester;
  20 import com.intellij.openapi.roots.OrderRootType;
  21 import com.intellij.openapi.roots.libraries.Library;
  22 import com.intellij.openapi.roots.libraries.LibraryUtil;
  23 import com.intellij.openapi.vfs.VirtualFile;
  24 import org.jetbrains.annotations.NotNull;
  25 import org.jetbrains.annotations.Nullable;
  26
  27 /**
  28  * @author nik
  29  */
  30 public class JUnitDependencyScopeSuggester extends LibraryDependencyScopeSuggester {
  31   @Nullable
  32   @Override
  33   public DependencyScope getDefaultDependencyScope(@NotNull Library library) {
  34     VirtualFile[] files = library.getFiles(OrderRootType.CLASSES);
  35     if (files.length == 1 && (LibraryUtil.isClassAvailableInLibrary(files, "junit.framework.TestCase")
  36                               || LibraryUtil.isClassAvailableInLibrary(library, "org.junit.Test"))) {
  37       return DependencyScope.TEST;
  38     }
  39     return null;
  40   }
  41 }