285a1b33ebe2db312a87f8963552c8164f68152a
[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       actions.add(new AdditionalAction() {
57         @Override
58         public String getText() {
59           return "Create New Test...";
60         }
61
62         @Override
63         public Icon getIcon() {
64           return IconLoader.getIcon("/actions/intentionBulb.png");
65         }
66
67         @Override
68         public void execute() {
69           final TestCreator creator = LanguageTestCreators.INSTANCE.forLanguage(file.getLanguage());
70           if (creator != null) creator.createTest(file.getProject(), editor, file);
71         }
72       });
73     }
74
75     return new GotoData(sourceElement, candidates.toArray(new PsiElement[candidates.size()]), actions);
76   }
77
78   @NotNull
79   public static PsiElement getSelectedElement(Editor editor, PsiFile file) {
80     return PsiUtilBase.getElementAtOffset(file, editor.getCaretModel().getOffset());
81   }
82
83   @Override
84   protected boolean shouldSortTargets() {
85     return false;
86   }
87
88   protected String getChooserTitle(PsiElement sourceElement, String name, int length) {
89     if (TestFinderHelper.isTest(sourceElement)) {
90       return CodeInsightBundle.message("goto.test.chooserTitle.subject", name, length);
91     }
92     else {
93       return CodeInsightBundle.message("goto.test.chooserTitle.test", name, length);
94     }
95   }
96
97   @Override
98   protected String getNotFoundMessage(Project project, Editor editor, PsiFile file) {
99     return CodeInsightBundle.message("goto.test.notFound");
100   }
101
102   @Override
103   protected void navigateToElement(Navigatable element) {
104     if (element instanceof PsiElement) {
105       NavigationUtil.activateFileWithPsiElement((PsiElement)element);
106     }
107     else {
108       element.navigate(true);
109     }
110   }
111 }