如何用springboot测试AWT.*？

Question

我有一个生成屏幕截图的类：

@Component
public class ImagesHandlerImpl implements ImagesHandler {
    ....
    public boolean doScreen() throws IOException, AWTException {
        final Rectangle rectangle = new Rectangle(Toolkit.getDefaultToolkit().getScreenSize());
        ...
    }
}

我的应用程序是用spring boot创建的，我需要对它进行测试。但我得到java.awt.HeadlessException

我的测试：

@RunWith(SpringRunner.class) 
@SpringBootTest(classes = ImagesHandlerImpl.class) 
@ContextConfiguration(classes = App.class) 
public class ImagesHandlerImplTest { 
    ...
    @Test
    public void whenDoScreenThenFilenameLikeTemplate() throws IOException, AWTException {
        imagesHandler.doScreen();
        final String name = dir.listFiles()[0].getName();
        assertThat("SCREEN_0", is(name));
    }
}

我试图防止无头异常：

@SpringBootApplication
public class App {
    public static void main(String[] args) {
        SpringApplicationBuilder builder = new SpringApplicationBuilder(App.class);
        builder.headless(false).run(args);
    }
}

我使用的是spring启动版本1.5.6。

但这无济于事。我得到日志：

java.awt.HeadlessException在sun.awt.HeadlesStoolKit.GetScreenSize（headlesStoolKit.java:284)在org.robinhood.image.imagesHandlerImpl.doScreen（imagesHandlerImpl.java:42)在org.robinhood.image.imagesHandlerImpl.whendoScreenEncreatePrintScreen（imagesHandlerImpl.java:51)在sun.reflect.nativeMethodAccessorImpl.Invoke0（ate(RunBeForestMethodCallbacks.Java:75)在org.springframework.test.context.junit4.statements.runAfterTestMethodCallbacks.evalue(RunAfterTestMethodCallbacks.Java:86)在org.springframework.test.context.junit4.statements.springrepeat.evalue(SrunAfterTestMethodCallbacks.84)在xt.junit4.statements.runbeForestClassCallbacks.evaluate(RunbeForestClassCallbacks.Java:61)在org.junit.internal.runners.statements.runafters.evaluate(Runafters.Java:27)在org.springframework.test.context.junit4.statements.runaftertestClassCallbacks.evaluate(Runafters.Java:27)在

帮帮我.如何解决此问题？谢谢！

Answer 1

添加

// overcome @SpringBootTest setup
System.setProperty("java.awt.headless", "false")

在doscreen()方法调用之前。

Answer 2

spring测试运行程序不调用app class的main method。并且它会在默认情况下开启无头模式。

您可以指定用于执行测试用例的JVM参数-djava.awt.headless=false。另一个解决方案是在bean初始化期间设置属性。

@Component
public class ImagesHandlerImpl implements ImagesHandler, InitializingBean {
  @Override
  public void afterPropertiesSet() throws Exception {
    System.setProperty("java.awt.headless", "false");
  }

  public boolean doScreen() throws Exception {
    //...
  }
}