GoTo Test: do not show 'Create Test' action if not applicable.
[idea/community.git] / platform / lang-impl / src / com / intellij / testIntegration / GotoTestOrCodeHandler.java
1 /*
2  * Copyright 2000-2009 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
17 package com.intellij.testIntegration;
18
19 import com.intellij.codeInsight.CodeInsightBundle;
20 import com.intellij.codeInsight.navigation.GotoTargetHandler;
21 import com.intellij.codeInsight.navigation.NavigationUtil;
22 import com.intellij.openapi.editor.Editor;
23 import com.intellij.openapi.project.Project;
24 import com.intellij.openapi.util.IconLoader;
25 import com.intellij.pom.Navigatable;
26 import com.intellij.psi.PsiElement;
27 import com.intellij.psi.PsiFile;
28 import com.intellij.psi.util.PsiUtilBase;
29 import com.intellij.util.SmartList;
30 import org.jetbrains.annotations.NotNull;
31 import org.jetbrains.annotations.Nullable;
32
33 import javax.swing.*;
34 import java.util.Collection;
35 import java.util.List;
36
37 public class GotoTestOrCodeHandler extends GotoTargetHandler {
38   protected String getFeatureUsedKey() {
39     return "navigation.goto.testOrCode";
40   }
41
42   @Nullable
43   protected GotoData getSourceAndTargetElements(final Editor editor, final PsiFile file) {
44     PsiElement selectedElement = getSelectedElement(editor, file);
45     PsiElement sourceElement = TestFinderHelper.findSourceElement(selectedElement);
46     if (sourceElement == null) return null;
47
48     List<AdditionalAction> actions = new SmartList<AdditionalAction>();
49
50     Collection<PsiElement> candidates;
51     if (TestFinderHelper.isTest(selectedElement)) {
52       candidates = TestFinderHelper.findClassesForTest(selectedElement);
53     }
54     else {
55       candidates = TestFinderHelper.findTestsForClass(selectedElement);
56       final TestCreator creator = LanguageTestCreators.INSTANCE.forLanguage(file.getLanguage());
57       if (creator != null && creator.isAvailable(file.getProject(), editor, file)) {
58         actions.add(new AdditionalAction() {
59           @Override
60           public String getText() {
61             return "Create New Test...";
62           }
63
64           @Override
65           public Icon getIcon() {
66             return IconLoader.getIcon("/actions/intentionBulb.png");
67           }
68
69           @Override
70           public void execute() {
71             creator.createTest(file.getProject(), editor, file);
72           }
73         });
74       }
75     }
76
77     return new GotoData(sourceElement, candidates.toArray(new PsiElement[candidates.size()]), actions);
78   }
79
80   @NotNull
81   public static PsiElement getSelectedElement(Editor editor, PsiFile file) {
82     return PsiUtilBase.getElementAtOffset(file, editor.getCaretModel().getOffset());
83   }
84
85   @Override
86   protected boolean shouldSortTargets() {
87     return false;
88   }
89
90   protected String getChooserTitle(PsiElement sourceElement, String name, int length) {
91     if (TestFinderHelper.isTest(sourceElement)) {
92       return CodeInsightBundle.message("goto.test.chooserTitle.subject", name, length);
93     }
94     else {
95       return CodeInsightBundle.message("goto.test.chooserTitle.test", name, length);
96     }
97   }
98
99   @Override
100   protected String getNotFoundMessage(Project project, Editor editor, PsiFile file) {
101     return CodeInsightBundle.message("goto.test.notFound");
102   }
103
104   @Override
105   protected void navigateToElement(Navigatable element) {
106     if (element instanceof PsiElement) {
107       NavigationUtil.activateFileWithPsiElement((PsiElement)element);
108     }
109     else {
110       element.navigate(true);
111     }
112   }
113 }