| /* |
| * Copyright 2000-2010 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. |
| */ |
| |
| /* |
| * User: anna |
| * Date: 15-Jun-2010 |
| */ |
| package com.intellij.execution.junit; |
| |
| import com.intellij.execution.RunManager; |
| import com.intellij.execution.configurations.RunConfiguration; |
| import com.intellij.openapi.actionSystem.*; |
| import com.intellij.openapi.project.Project; |
| import com.intellij.openapi.ui.popup.JBPopupFactory; |
| import com.intellij.openapi.ui.popup.PopupStep; |
| import com.intellij.openapi.ui.popup.util.BaseListPopupStep; |
| import com.intellij.psi.PsiElement; |
| import org.jetbrains.annotations.NotNull; |
| |
| import javax.swing.*; |
| import java.util.ArrayList; |
| import java.util.List; |
| import java.util.Set; |
| |
| public class AddToTestsPatternAction extends AnAction { |
| @Override |
| public void actionPerformed(AnActionEvent e) { |
| final DataContext dataContext = e.getDataContext(); |
| final PsiElement[] psiElements = LangDataKeys.PSI_ELEMENT_ARRAY.getData(dataContext); |
| final Set<PsiElement> classes = PatternConfigurationProducer.collectTestMembers(psiElements, true); |
| |
| final Project project = CommonDataKeys.PROJECT.getData(dataContext); |
| final List<JUnitConfiguration> patternConfigurations = collectPatternConfigurations(classes, project); |
| if (patternConfigurations.size() == 1) { |
| final JUnitConfiguration configuration = patternConfigurations.get(0); |
| for (PsiElement aClass : classes) { |
| configuration.getPersistentData().getPatterns().add(PatternConfigurationProducer.getQName(aClass)); |
| } |
| } else { |
| JBPopupFactory.getInstance().createListPopup(new BaseListPopupStep<JUnitConfiguration>("Choose suite to add", patternConfigurations) { |
| @Override |
| public PopupStep onChosen(JUnitConfiguration configuration, boolean finalChoice) { |
| for (PsiElement aClass : classes) { |
| configuration.getPersistentData().getPatterns().add(PatternConfigurationProducer.getQName(aClass)); |
| } |
| return FINAL_CHOICE; |
| } |
| |
| @Override |
| public Icon getIconFor(JUnitConfiguration configuration) { |
| return configuration.getIcon(); |
| } |
| |
| @NotNull |
| @Override |
| public String getTextFor(JUnitConfiguration value) { |
| return value.getName(); |
| } |
| }).showInBestPositionFor(dataContext); |
| } |
| } |
| |
| @Override |
| public void update(AnActionEvent e) { |
| final Presentation presentation = e.getPresentation(); |
| presentation.setVisible(false); |
| final DataContext dataContext = e.getDataContext(); |
| final PsiElement[] psiElements = LangDataKeys.PSI_ELEMENT_ARRAY.getData(dataContext); |
| if (psiElements != null) { |
| final Set<PsiElement> foundMembers = PatternConfigurationProducer.collectTestMembers(psiElements, true); |
| if (foundMembers.isEmpty()) return; |
| final Project project = CommonDataKeys.PROJECT.getData(dataContext); |
| if (project != null) { |
| final List<JUnitConfiguration> foundConfigurations = collectPatternConfigurations(foundMembers, project); |
| if (!foundConfigurations.isEmpty()) { |
| presentation.setVisible(true); |
| if (foundConfigurations.size() == 1) { |
| presentation.setText("Add to temp suite: " + foundConfigurations.get(0).getName()); |
| } |
| } |
| } |
| } |
| } |
| |
| private static List<JUnitConfiguration> collectPatternConfigurations(Set<PsiElement> foundClasses, Project project) { |
| final List<RunConfiguration> configurations = RunManager.getInstance(project).getConfigurationsList( |
| JUnitConfigurationType.getInstance()); |
| final List<JUnitConfiguration> foundConfigurations = new ArrayList<JUnitConfiguration>(); |
| for (RunConfiguration configuration : configurations) { |
| final JUnitConfiguration.Data data = ((JUnitConfiguration)configuration).getPersistentData(); |
| if (data.TEST_OBJECT == JUnitConfiguration.TEST_PATTERN) { |
| if (foundClasses.size() > 1 || !data.getPatterns().contains(PatternConfigurationProducer.getQName(foundClasses.iterator().next())) ) { |
| foundConfigurations.add((JUnitConfiguration)configuration); |
| } |
| } |
| } |
| return foundConfigurations; |
| } |
| |
| } |