testMultipleHandshakes() — netty Function Reference

Source Code

handler/src/test/java/io/netty/handler/ssl/JdkDelegatingPrivateKeyMethodTest.java lines 292–347

    @Test
    void testMultipleHandshakes() throws Exception {
        // Create certificate and wrap the private key
        PrivateKey wrappedPrivateKey = wrapPrivateKey(RSA_BUNDLE.getKeyPair().getPrivate());
        X509Certificate certificate = RSA_BUNDLE.getCertificate();

        // Set up contexts
        SslContext serverSslContext = SslContextBuilder.forServer(wrappedPrivateKey, certificate)
                .sslProvider(SslProvider.OPENSSL)
                .option(OpenSslContextOption.USE_JDK_PROVIDER_SIGNATURES, true)
                .build();

        SslContext clientSslContext = SslContextBuilder.forClient()
                .sslProvider(SslProvider.OPENSSL)
                .trustManager(certificate)
                .build();

        // Test multiple connections to ensure provider caching works correctly
        int numberOfConnections = 3;
        List<Future<String>> results = new ArrayList<>();

        // Set up server
        ServerBootstrap serverBootstrap = new ServerBootstrap()
                .group(GROUP)
                .channel(NioServerSocketChannel.class)
                .childHandler(new ChannelInitializer<SocketChannel>() {
                    @Override
                    protected void initChannel(SocketChannel ch) {
                        ChannelPipeline pipeline = ch.pipeline();
                        SslHandler sslHandler = serverSslContext.newHandler(ch.alloc());
                        pipeline.addLast(sslHandler);
                        pipeline.addLast(ServerHandler.INSTANCE);
                    }
                });

        ChannelFuture serverChannelFuture = serverBootstrap.bind(0).sync();
        Channel serverChannel = serverChannelFuture.channel();
        int serverPort = ((java.net.InetSocketAddress) serverChannel.localAddress()).getPort();

        try {
            // Create multiple client connections
            for (int i = 0; i < numberOfConnections; i++) {
                results.add(runClient(serverPort, clientSslContext));
            }

            // Wait for all connections to complete
            for (Future<String> result : results) {
                assertEquals("R", result.get(20, TimeUnit.SECONDS),
                          "All connections should complete within timeout");
            }
        } finally {
            serverChannel.close().sync();
            ReferenceCountUtil.release(serverSslContext);
            ReferenceCountUtil.release(clientSslContext);
        }
    }