Project import generated by Copybara.

PiperOrigin-RevId: 946867859

Author: miguelaranda0

common/src/test/java/org/conscrypt/javax/net/ssl/SSLEngineVersionCompatibilityTest.java

@@ -379,26 +379,29 @@ void beforeBeginHandshake(SSLEngine client, SSLEngine server) {
 
     @Test
     public void test_TestSSLEnginePair_create() throws Exception {
-        TestSSLEnginePair test = TestSSLEnginePair.create(TestSSLContext.newBuilder()
-                                                                  .clientProtocol(clientVersion)
-                                                                  .serverProtocol(serverVersion)
-                                                                  .build());
+        TestSSLContext c = TestSSLContext.newBuilder()
+                                   .clientProtocol(clientVersion)
+                                   .serverProtocol(serverVersion)
+                                   .build();
+        TestSSLEnginePair test = TestSSLEnginePair.create(c);
         assertNotNull(test.c);
         assertNotNull(test.server);
         assertNotNull(test.client);
         assertConnected(test);
         test.close();
+        c.close();
     }
 
     private final int NUM_STRESS_ITERATIONS = 1000;
 
     @Test
     public void test_SSLEngine_Multiple_Thread_Success() throws Exception {
-        final TestSSLEnginePair pair =
-                TestSSLEnginePair.create(TestSSLContext.newBuilder()
+        TestSSLContext c = TestSSLContext.newBuilder()
                                                  .clientProtocol(clientVersion)
                                                  .serverProtocol(serverVersion)
-                                                 .build());
+                                                 .build();
+        final TestSSLEnginePair pair =
+                TestSSLEnginePair.create(c);
         try {
             assertConnected(pair);
 
@@ -435,16 +438,18 @@ public Void call() throws Exception {
             server.get();
         } finally {
             pair.close();
+            c.close();
         }
     }
 
     @Test
     public void test_SSLEngine_CloseOutbound() throws Exception {
-        final TestSSLEnginePair pair =
-                TestSSLEnginePair.create(TestSSLContext.newBuilder()
+        TestSSLContext c = TestSSLContext.newBuilder()
                                                  .clientProtocol(clientVersion)
                                                  .serverProtocol(serverVersion)
-                                                 .build());
+                                                 .build();
+        final TestSSLEnginePair pair =
+                TestSSLEnginePair.create(c);
         try {
             assertConnected(pair);
 
@@ -487,16 +492,18 @@ public void test_SSLEngine_CloseOutbound() throws Exception {
             assertTrue(pair.server.isInboundDone() && pair.server.isOutboundDone());
         } finally {
             pair.close();
+            c.close();
         }
     }
 
     @Test
     public void test_SSLEngine_Closed() throws Exception {
-        final TestSSLEnginePair pair =
-                TestSSLEnginePair.create(TestSSLContext.newBuilder()
+        TestSSLContext c = TestSSLContext.newBuilder()
                                                  .clientProtocol(clientVersion)
                                                  .serverProtocol(serverVersion)
-                                                 .build());
+                                                 .build();
+        final TestSSLEnginePair pair =
+                TestSSLEnginePair.create();
         pair.close();
         ByteBuffer out = ByteBuffer.allocate(pair.client.getSession().getPacketBufferSize());
         SSLEngineResult res = pair.client.wrap(ByteBuffer.wrap(new byte[] {0x01}), out);
@@ -510,6 +517,7 @@ public void test_SSLEngine_Closed() throws Exception {
         assertEquals(Status.CLOSED, res.getStatus());
         assertEquals(0, res.bytesConsumed());
         assertEquals(0, res.bytesProduced());
+        c.close();
     }
 
     @Test
@@ -596,13 +604,14 @@ private static byte[] getFirstChunk(SSLEngine e) throws SSLException {
 
     @Test
     public void test_SSLEngine_TlsUnique() throws Exception {
+        TestSSLContext c = TestSSLContext.newBuilder()
+                                                 .clientProtocol(clientVersion)
+                                                 .serverProtocol(serverVersion)
+                                                 .build();
         // tls_unique isn't supported in TLS 1.3
         assumeTlsV1_2Connection();
         TestSSLEnginePair pair = TestSSLEnginePair.create(
-                TestSSLContext.newBuilder()
-                        .clientProtocol(clientVersion)
-                        .serverProtocol(serverVersion)
-                        .build(),
+                c,
                 new TestSSLEnginePair.Hooks() {
                     @Override
                     void beforeBeginHandshake(SSLEngine client, SSLEngine server) {
@@ -620,16 +629,18 @@ void beforeBeginHandshake(SSLEngine client, SSLEngine server) {
             assertArrayEquals(clientTlsUnique, serverTlsUnique);
         } finally {
             pair.close();
+            c.close();
         }
     }
 
     @Test
     public void test_SSLEngine_EKM() throws Exception {
+        TestSSLContext c = TestSSLContext.newBuilder()
+                                                 .clientProtocol(clientVersion)
+                                                 .serverProtocol(serverVersion)
+                                                 .build();
         TestSSLEnginePair pair = TestSSLEnginePair.create(
-                TestSSLContext.newBuilder()
-                        .clientProtocol(clientVersion)
-                        .serverProtocol(serverVersion)
-                        .build(),
+                c,
                 new TestSSLEnginePair.Hooks() {
                     @Override
                     void beforeBeginHandshake(SSLEngine client, SSLEngine server) {
@@ -671,6 +682,7 @@ void beforeBeginHandshake(SSLEngine client, SSLEngine server) {
             }
         } finally {
             pair.close();
+            c.close();
         }
     }
 
@@ -742,18 +754,20 @@ public X509Certificate[] getAcceptedIssuers() {
             assertTrue(expected.getCause() instanceof CertificateException);
         }
         assertTrue(trustManager.threw);
+        referenceContext.close();
     }
 
     @Test
     public void sniHandlerFailureResultsInHandshakeError() throws Exception {
         assumeJava8();
 
+        TestSSLContext c = TestSSLContext.newBuilder()
+                                                 .clientProtocol(clientVersion)
+                                                 .serverProtocol(serverVersion)
+                                                 .build();
         try {
             TestSSLEnginePair.create(
-                    TestSSLContext.newBuilder()
-                            .clientProtocol(clientVersion)
-                            .serverProtocol(serverVersion)
-                            .build(),
+                    c,
                     new TestSSLEnginePair.Hooks() {
                         @Override
                         void beforeBeginHandshake(SSLEngine client, SSLEngine server) {
@@ -768,6 +782,7 @@ void beforeBeginHandshake(SSLEngine client, SSLEngine server) {
         } catch (SSLHandshakeException e) {
             assertEquals("SNI match failed: any.host", e.getMessage());
         }
+        c.close();
     }
 
     @Test