spring-boot-test测试时出现不能创建bean的问题?报错

最近使用spring-boot写项目，集成了spring-boot-starter-websocket和spring-boot-starter-test;

websocket配置如下

@Configuration
public class WebSocketConfig {
    @Bean
    public ServerEndpointExporter serverEndpointExporter(){
        return new ServerEndpointExporter();
    }
}

正常运行spring-boot时websocket不报错

但是使用单元测试的时候报错

Error creating bean with name 'serverEndpointExporter' defined in class path resource [com/xgh/WebSocketConfig.class]: Invocation of init method failed; nested exception is java.lang.IllegalStateException: javax.websocket.server.ServerContainer not available

去掉websocket的配置，单元测试正常，有没有比较好的解决办法，

单元测试中能不能指定不加载websocket相关的bean呢？