WebSocketServerHandshaker13Test Class — netty Architecture

Source Code

codec-http/src/test/java/io/netty/handler/codec/http/websocketx/WebSocketServerHandshaker13Test.java lines 50–225

public class WebSocketServerHandshaker13Test extends WebSocketServerHandshakerTest {

    @Override
    protected WebSocketServerHandshaker newHandshaker(String webSocketURL, String subprotocols,
            WebSocketDecoderConfig decoderConfig) {
        return new WebSocketServerHandshaker13(webSocketURL, subprotocols, decoderConfig);
    }

    @Override
    protected WebSocketVersion webSocketVersion() {
        return WebSocketVersion.V13;
    }

    @Test
    public void testPerformOpeningHandshake() {
        testPerformOpeningHandshake0(true);
    }

    @Test
    public void testPerformOpeningHandshakeSubProtocolNotSupported() {
        testPerformOpeningHandshake0(false);
    }

    private static void testPerformOpeningHandshake0(boolean subProtocol) {
        EmbeddedChannel ch = new EmbeddedChannel(
                new HttpObjectAggregator(42), new HttpResponseEncoder(), new HttpRequestDecoder());

        if (subProtocol) {
            testUpgrade0(ch, new WebSocketServerHandshaker13(
                    "ws://example.com/chat", "chat", false, Integer.MAX_VALUE, false));
        } else {
            testUpgrade0(ch, new WebSocketServerHandshaker13(
                    "ws://example.com/chat", null, false, Integer.MAX_VALUE, false));
        }
        assertFalse(ch.finish());
    }

    @Test
    public void testCloseReasonWithEncoderAndDecoder() {
        testCloseReason0(new HttpResponseEncoder(), new HttpRequestDecoder());
    }

    @Test
    public void testCloseReasonWithCodec() {
        testCloseReason0(new HttpServerCodec());
    }

    @Test
    public void testHandshakeExceptionWhenConnectionHeaderIsAbsent() {
        final WebSocketServerHandshaker serverHandshaker = newHandshaker("ws://example.com/chat",
                                                                         "chat", WebSocketDecoderConfig.DEFAULT);
        final FullHttpRequest request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.GET,
                                                                   "ws://example.com/chat");
        request.headers()
               .set(HttpHeaderNames.HOST, "server.example.com")
               .set(HttpHeaderNames.UPGRADE, HttpHeaderValues.WEBSOCKET)
               .set(HttpHeaderNames.SEC_WEBSOCKET_KEY, "dGhlIHNhbXBsZSBub25jZQ==")
               .set(HttpHeaderNames.SEC_WEBSOCKET_ORIGIN, "http://example.com")
               .set(HttpHeaderNames.SEC_WEBSOCKET_PROTOCOL, "chat, superchat")
               .set(HttpHeaderNames.SEC_WEBSOCKET_VERSION, "13");
        Throwable exception = assertThrows(WebSocketServerHandshakeException.class, new Executable() {
            @Override
            public void execute() throws Throwable {
                serverHandshaker.handshake(null, request, null, null);
            }
        });

        assertEquals("not a WebSocket request: a |Connection| header must includes a token 'Upgrade'",
                     exception.getMessage());
        assertTrue(request.release());
    }

    @Test
    public void testHandshakeExceptionWhenInvalidConnectionHeader() {
        final WebSocketServerHandshaker serverHandshaker = newHandshaker("ws://example.com/chat",
                                                                         "chat", WebSocketDecoderConfig.DEFAULT);
        final FullHttpRequest request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.GET,
                                                                   "ws://example.com/chat");
        request.headers()
               .set(HttpHeaderNames.HOST, "server.example.com")
               .set(HttpHeaderNames.CONNECTION, "close")