如何在无头环境中测试Swing GUI组件？

更新CI的Docker镜像绝对不是唯一解决办法，下面是几个可以直接落地的稳健方案：

1. 模拟核心依赖，避开AWT渲染逻辑 #

你之前用Mockito spy没成功，大概率是没mock到关键触发点。不用真的显示组件，只需要验证弹出菜单的显示方法是否被调用即可：

• 提前设置Headless模式：

  @BeforeEach
  void setUp() {
      System.setProperty("java.awt.headless", "true");
  }
  

• 直接mock JPopupMenu实例并注入到自定义按钮中，模拟点击后验证show()方法是否触发：

  @Test
  void whenButtonClicked_thenPopupMenuShows() {
      // 模拟弹出菜单
      JPopupMenu mockPopup = Mockito.mock(JPopupMenu.class);
      // 初始化自定义按钮并注入模拟菜单
      JDropdownButton button = new JDropdownButton(mockPopup);
      
      // 直接调用按钮点击事件处理方法，跳过AWT事件触发
      button.doClick();
      
      // 验证菜单的show方法是否被调用
      Mockito.verify(mockPopup).show(button, Mockito.anyInt(), Mockito.anyInt());
  }
  

这种方式完全绕开组件渲染和位置计算，只测试核心逻辑，无头环境下毫无压力。

2. 用支持无头模式的GUI测试框架 #

比如AssertJ Swing，它原生支持Headless环境，无需自己编写大量mock代码：

• 引入AssertJ Swing依赖（Maven/Gradle）后，编写测试用例：

  import org.assertj.swing.edt.GuiActionRunner;
  import org.assertj.swing.fixture.JButtonFixture;
  import org.assertj.swing.junit.testcase.AssertJSwingJUnitTestCase;
  
  public class JDropdownButtonTest extends AssertJSwingJUnitTestCase {
      private JButtonFixture buttonFixture;
      private JDropdownButton button;
  
      @Override
      protected void onSetUp() {
          // 强制开启Headless模式
          System.setProperty("java.awt.headless", "true");
          // 在EDT线程中创建组件
          button = GuiActionRunner.execute(() -> new JDropdownButton());
          // 将组件绑定到测试用的容器（无需显示到屏幕）
          buttonFixture = new JButtonFixture(robot(), button);
      }
  
      @Test
      void clickButtonShowsPopup() {
          // 模拟按钮点击
          buttonFixture.click();
          // 验证弹出菜单是否可见
          assertThat(button.getPopupMenu().isVisible()).isTrue();
      }
  }
  

AssertJ Swing会自动处理无头环境下的UI事件模拟，无需手动处理组件位置问题。

3. 重构组件，分离UI与逻辑 #

如果自定义组件耦合度较高，可以把弹出菜单的触发逻辑抽成独立服务：

• 定义一个PopupTrigger接口：

  public interface PopupTrigger {
      void showPopup(JComponent source, int x, int y);
  }
  

• 让JDropdownButton依赖该接口，而非直接操作JPopupMenu：

  public class JDropdownButton extends JButton {
      private final PopupTrigger trigger;
      private JPopupMenu popup;
  
      public JDropdownButton(PopupTrigger trigger, JPopupMenu popup) {
          this.trigger = trigger;
          this.popup = popup;
          addActionListener(e -> trigger.showPopup(this, 0, getHeight()));
      }
  }
  

• 测试时直接mockPopupTrigger，验证点击时是否调用了showPopup：

  @Test
  void buttonClickTriggersPopup() {
      PopupTrigger mockTrigger = Mockito.mock(PopupTrigger.class);
      JDropdownButton button = new JDropdownButton(mockTrigger, new JPopupMenu());
      
      button.doClick();
      
      Mockito.verify(mockTrigger).showPopup(button, 0, button.getHeight());
  }
  

这种方式从根源上解耦了UI渲染和业务逻辑，测试完全脱离AWT环境，本地和CI环境都能稳定运行。

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