开发Eclipse插件：如何程序化控制/获取JUnit视图并添加自定义右键选项

我之前刚好做过类似的自定义JUnit右键菜单需求，给你捋捋具体怎么实现——核心是借助IntelliJ IDEA的插件开发能力，因为JUnit视图是IDE提供的内置组件，咱们得通过插件来扩展它的右键菜单：

1. 搭建插件开发环境 #

首先得在IntelliJ里新建一个IntelliJ Platform Plugin项目，选择和你日常使用的IDE版本一致的IDEA SDK，避免兼容性问题。

2. 编写自定义Action逻辑 #

创建一个继承自AnAction的类，这就是右键菜单点击后要执行的核心逻辑，同时要控制这个选项只在选中单个JUnit测试方法时显示：

import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.LangDataKeys;
import com.intellij.openapi.ui.Messages;
import com.intellij.psi.PsiAnnotation;
import com.intellij.psi.PsiMethod;

public class CustomTestMethodAction extends AnAction {

    @Override
    public void actionPerformed(AnActionEvent e) {
        // 获取当前选中的测试方法
        PsiMethod selectedTestMethod = e.getData(LangDataKeys.PSI_METHOD);
        if (selectedTestMethod != null) {
            // 这里替换成你的自定义处理逻辑，比如生成测试报告、调用外部脚本等
            Messages.showInfoMessage(
                "执行自定义操作：测试方法 " + selectedTestMethod.getName(),
                "自定义测试工具"
            );
        }
    }

    // 控制菜单选项的可见性：只在选中JUnit测试方法时显示
    @Override
    public void update(AnActionEvent e) {
        PsiMethod method = e.getData(LangDataKeys.PSI_METHOD);
        e.getPresentation().setVisible(method != null && isJUnitTestMethod(method));
    }

    // 判断是否为JUnit测试方法（兼容JUnit 4和JUnit 5）
    private boolean isJUnitTestMethod(PsiMethod method) {
        for (PsiAnnotation annotation : method.getAnnotations()) {
            String annoQualifiedName = annotation.getQualifiedName();
            if ("org.junit.Test".equals(annoQualifiedName) || 
                "org.junit.jupiter.api.Test".equals(annoQualifiedName)) {
                return true;
            }
        }
        return false;
    }
}

3. 配置菜单位置 #

打开项目里的plugin.xml，在<actions>标签下注册你的自定义Action，并指定它要添加到JUnit右键菜单中：

<actions>
    <action 
        id="CustomTestMethodAction" 
        class="com.your.package.path.CustomTestMethodAction" 
        text="我的自定义测试操作" 
        description="对选中的JUnit测试方法执行自定义逻辑">
        
        <!-- 将选项添加到JUnit右键菜单的Run选项之后 -->
        <add-to-group group-id="RunPopupGroup" anchor="after" relative-to-action="Run"/>
        
        <!-- 可选：添加快捷键 -->
        <keyboard-shortcut keymap="$default" first-keystroke="ctrl alt T"/>
    </action>
</actions>

注：RunPopupGroup是JUnit视图右键菜单的内置组ID，不同IDE版本基本通用，你也可以通过Plugin DevKit的Action Search工具确认这个ID的准确性。

4. 测试插件 #

直接运行插件项目，会启动一个全新的IntelliJ实例。在这个实例里打开一个带有JUnit测试的项目，右键单个测试方法，就能看到你添加的自定义选项了，点击后会执行你写的逻辑。

额外提示 #

• 如果需要依赖JUnit相关的API，记得在plugin.xml里添加依赖：

  <depends>com.intellij.java</depends>
  <depends>junit</depends>
  

• 要是想更精准地限定只在JUnit视图中显示，可以通过e.getData(PlatformDataKeys.CONTEXT_COMPONENT)获取当前组件，判断是否为JUnit视图的相关组件。

内容的提问来源于stack exchange，提问作者luft