Spring Boot WebSocket在SockJS客户端测试中抛出“拒绝连接”

Question

我正在开发Spring Boot Server项目，该项目提供了简单的REST资源，直到现在。为了将通知推送到客户端，我想添加一个websocket连接。为了测试这方面，我已经使用基于本教程SockJS客户端写的集成测试：

http://rafaelhz.github.io/testing-websockets/

问题是连接与以下错误拒绝：

org.springframework.web.client.ResourceAccessException: I/O error on GET request for "http://localhost:9090/websocket/info": Connection refused (Connection refused); nested exception is java.net.ConnectException: Connection refused (Connection refused) 

我的WebSocket配置如下：

import org.springframework.context.annotation.Configuration; 
import org.springframework.messaging.simp.config.MessageBrokerRegistry; 
import org.springframework.web.socket.config.annotation.AbstractWebSocketMessageBrokerConfigurer; 
import org.springframework.web.socket.config.annotation.EnableWebSocketMessageBroker; 
import org.springframework.web.socket.config.annotation.StompEndpointRegistry; 

@Configuration 
@EnableWebSocketMessageBroker 
public class WebSocketConfig extends AbstractWebSocketMessageBrokerConfigurer { 

    @Override 
    public void configureMessageBroker(MessageBrokerRegistry config) { 
     config.enableSimpleBroker("/topic"); 
     config.setApplicationDestinationPrefixes("/app"); 
    } 

    @Override 
    public void registerStompEndpoints(StompEndpointRegistry registry) { 
     registry 
       .addEndpoint("/websocket") 
       .setAllowedOrigins("*") 
       .withSockJS(); 
    } 

} 

我可以看到，在套接字端点映射到日志：

2017-07-14 15:22:59.561 INFO 13765 --- [   main] o.s.w.s.s.s.WebSocketHandlerMapping  : Mapped URL path [/websocket/**] onto handler of type [class org.springframework.web.socket.sockjs.support.SockJsHttpRequestHandler] 

服务器端口设置为9090在application.yml文件：

server: 
port: 9090 

下面的单元测试是无法连接到插座：

import org.junit.Assert; 
import org.junit.Before; 
import org.junit.Test; 
import org.junit.runner.RunWith; 
import org.springframework.boot.test.context.SpringBootTest; 
import org.springframework.test.annotation.DirtiesContext; 
import org.springframework.test.context.ActiveProfiles; 
import org.springframework.test.context.junit4.SpringRunner; 
import org.springframework.web.socket.client.standard.StandardWebSocketClient; 
import org.springframework.web.socket.messaging.WebSocketStompClient; 
import org.springframework.web.socket.sockjs.client.SockJsClient; 
import org.springframework.web.socket.sockjs.client.WebSocketTransport; 
import org.springframework.messaging.simp.stomp.StompFrameHandler; 
import org.springframework.messaging.simp.stomp.StompHeaders; 
import org.springframework.messaging.simp.stomp.StompSession; 
import org.springframework.messaging.simp.stomp.StompSessionHandlerAdapter; 

import java.lang.reflect.Type; 
import java.util.concurrent.BlockingQueue; 
import java.util.concurrent.LinkedBlockingDeque; 

import static java.util.Arrays.asList; 
import static java.util.concurrent.TimeUnit.SECONDS; 

@RunWith(SpringRunner.class) 
@SpringBootTest 
@ActiveProfiles("test") 
//@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_EACH_TEST_METHOD) 
public class WebSocketConnectionTest { 

    static final String WEBSOCKET_URI = "ws://localhost:9090/websocket"; 
    static final String WEBSOCKET_TOPIC = "/topic"; 

    BlockingQueue<String> blockingQueue; 
    WebSocketStompClient stompClient; 

    @Before 
    public void setup() { 
     blockingQueue = new LinkedBlockingDeque<>(); 
     stompClient = new WebSocketStompClient(new SockJsClient(
       asList(new WebSocketTransport(new StandardWebSocketClient())))); 

     System.out.println(WEBSOCKET_URI); 
    } 

    @Test 
    public void shouldReceiveAMessageFromTheServer() throws Exception { 
     StompSession session = stompClient 
       .connect(WEBSOCKET_URI, new StompSessionHandlerAdapter() {}) 
       .get(1, SECONDS); 
     session.subscribe(WEBSOCKET_TOPIC, new DefaultStompFrameHandler()); 

     String message = "MESSAGE TEST"; 
     session.send(WEBSOCKET_TOPIC, message.getBytes()); 

     Assert.assertEquals(message, blockingQueue.poll(1, SECONDS)); 
    } 

    class DefaultStompFrameHandler implements StompFrameHandler { 
     @Override 
     public Type getPayloadType(StompHeaders stompHeaders) { 
      return byte[].class; 
     } 

     @Override 
     public void handleFrame(StompHeaders stompHeaders, Object o) { 
      blockingQueue.offer(new String((byte[]) o)); 
     } 
    } 
} 

连接被拒绝。我相当肯定，这是因为URI端点不存在，但我不知道为什么。有人知道URI中是否有错误，或者是否有其他内容导致拒绝连接？

Answer 1

我发现问题的原因。 PORT 9090上不存在端点。这是因为@SpringBootTest注释默认情况下将Web环境设置为WebEnvironment.MOCK。在这个配置中，没有嵌入式Servlet被启动并且因此没有端口存在，只有基于MockMvc的测试是可能的。为了启动嵌入式Servlet ，环境必须设置为WebEnvironment.RANDOM_PORT或WebEnvironment.DEFINED_PORT。我将它设置为DEFINED_PORT，以便使用我的application.yml中的端口9090。通过设置环境，测试运行正确。

@RunWith(SpringRunner.class) 
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.DEFINED_PORT)//!!!!! 
@ActiveProfiles("test") 
@DirtiesContext(classMode = DirtiesContext.ClassMode.AFTER_EACH_TEST_METHOD) 
public class WebSocketConnectionTest { 

String WEBSOCKET_URI = "ws://localhost:9090/websocket"; 
String WEBSOCKET_TOPIC = "/topic"; 
. 
. 
.