Fix AbstractWriteSearchIndexOperation executeAsync releasing connections

driver-core/src/main/com/mongodb/internal/operation/AbstractWriteSearchIndexOperation.java

@@ -66,9 +66,9 @@ public void executeAsync(final AsyncWriteBinding binding, final SingleResultCall
                                 writeConcernErrorTransformerAsync(binding.getOperationContext().getTimeoutContext()), (result, commandExecutionError) -> {
                                     try {
                                         swallowOrThrow(commandExecutionError);
-                                        callback.onResult(result, null);
+                                        cb.onResult(result, null);
                                     } catch (Throwable mongoCommandException) {
-                                        callback.onResult(null, mongoCommandException);
+                                        cb.onResult(null, mongoCommandException);
                                     }
                                 }
                         )

driver-core/src/main/com/mongodb/internal/operation/AsyncOperationHelper.java

@@ -138,6 +138,9 @@ static <R, T extends ReferenceCounted> void withAsyncSuppliedResource(final Asyn
                     AsyncCallbackSupplier<R> curriedFunction = c -> function.apply(resource, c);
                     curriedFunction.whenComplete(resource::release).get(errorHandlingCallback);
                 } catch (Exception e) {
+                    if (resource.getCount() > 0) {
+                        resource.release();
+                    }
                     errorHandlingCallback.onResult(null, e);
                 }
             }

driver-reactive-streams/src/test/functional/com/mongodb/internal/connection/OidcAuthenticationAsyncProseTests.java

@@ -33,7 +33,7 @@ public class OidcAuthenticationAsyncProseTests extends OidcAuthenticationProseTe
 
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Test

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/AbstractClientMetadataProseTest.java

@@ -29,6 +29,6 @@
 class ClientMetadataProseTest extends AbstractClientMetadataProseTest {
 
     protected MongoClient createMongoClient(@Nullable final MongoDriverInformation mongoDriverInformation, final MongoClientSettings mongoClientSettings) {
-        return new SyncMongoClient(MongoClients.create(mongoClientSettings, mongoDriverInformation));
+        return new SyncMongoClient(mongoClientSettings, mongoDriverInformation);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/AsyncTransportSettingsTest.java

@@ -48,7 +48,7 @@ void testAsyncTransportSettings() {
                 .transportSettings(asyncTransportSettings)
                 .build();
 
-        try (MongoClient client = new SyncMongoClient(MongoClients.create(mongoClientSettings))) {
+        try (MongoClient client = new SyncMongoClient(mongoClientSettings)) {
             client.listDatabases().first();
         }
         verify(executorService, atLeastOnce()).execute(any());
@@ -67,7 +67,7 @@ void testExternalExecutorWasShutDown(final boolean tlsEnabled) throws Interrupte
                 .transportSettings(asyncTransportSettings)
                 .build();
 
-        try (MongoClient ignored = new SyncMongoClient(MongoClients.create(mongoClientSettings))) {
+        try (MongoClient ignored = new SyncMongoClient(mongoClientSettings)) {
             // ignored
         }
 

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/AtlasSearchIndexManagementProseTest.java

@@ -27,6 +27,6 @@
  */
 public class AtlasSearchIndexManagementProseTest extends AbstractAtlasSearchIndexManagementProseTest {
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientEncryptionDataKeyAndDoubleEncryptionTest.java

@@ -73,7 +73,7 @@ public void setUp() {
 
         // Step 1: create unencrypted client
         commandListener = new TestCommandListener();
-        client = new SyncMongoClient(MongoClients.create(getMongoClientSettingsBuilder().addCommandListener(commandListener).build()));
+        client = new SyncMongoClient(getMongoClientSettingsBuilder().addCommandListener(commandListener));
         client.getDatabase("keyvault").getCollection("datakeys").drop();
         client.getDatabase("db").getCollection("coll").drop();
 
@@ -114,12 +114,11 @@ public void setUp() {
         }};
 
         String keyVaultNamespace = "keyvault.datakeys";
-        clientEncrypted = new SyncMongoClient(MongoClients.create(getMongoClientSettingsBuilder()
+        clientEncrypted = new SyncMongoClient(getMongoClientSettingsBuilder()
                 .autoEncryptionSettings(AutoEncryptionSettings.builder()
                         .keyVaultNamespace(keyVaultNamespace)
                         .kmsProviders(kmsProviders)
                         .schemaMap(schemaMap)
-                .build())
                 .build()));
 
         clientEncryption = ClientEncryptions.create(

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientEncryptionRewrapManyDataKeyProseTest.java

@@ -28,7 +28,7 @@ public class ClientEncryptionRewrapManyDataKeyProseTest extends AbstractClientEn
 
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientEncryptionTextExplicitEncryptionTest.java

@@ -28,7 +28,7 @@
 public class ClientEncryptionTextExplicitEncryptionTest extends AbstractClientEncryptionTextExplicitEncryptionTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryption25LookupProseTests.java

@@ -28,7 +28,7 @@ public class ClientSideEncryption25LookupProseTests extends com.mongodb.client.C
 
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionAutoDataKeysTest.java

@@ -28,7 +28,7 @@
 final class ClientSideEncryptionAutoDataKeysTest extends AbstractClientSideEncryptionAutoDataKeysTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionAwsCredentialFromEnvironmentTest.java

@@ -33,6 +33,6 @@ protected ClientEncryption createClientEncryption(final ClientEncryptionSettings
 
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionDeadlockTest.java

@@ -26,6 +26,6 @@
 public class ClientSideEncryptionDeadlockTest extends AbstractClientSideEncryptionDeadlockTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionDecryptionEventsTest.java

@@ -28,7 +28,7 @@
 public class ClientSideEncryptionDecryptionEventsTest extends AbstractClientSideEncryptionDecryptionEventsTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionNotCreateMongocryptdClientTest.java

@@ -24,6 +24,6 @@
 final class ClientSideEncryptionNotCreateMongocryptdClientTest extends AbstractClientSideEncryptionNotCreateMongocryptdClientTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionNotSpawnMongocryptdTest.java

@@ -24,6 +24,6 @@
 final class ClientSideEncryptionNotSpawnMongocryptdTest extends AbstractClientSideEncryptionNotSpawnMongocryptdTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionRangeExplicitEncryptionTest.java

@@ -29,7 +29,7 @@
 public class ClientSideEncryptionRangeExplicitEncryptionTest extends AbstractClientSideEncryptionRangeExplicitEncryptionTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideEncryptionUniqueIndexKeyAltNamesTest.java

@@ -28,7 +28,7 @@
 public class ClientSideEncryptionUniqueIndexKeyAltNamesTest extends AbstractClientSideEncryptionUniqueIndexKeyAltNamesTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideExplicitEncryptionTest.java

@@ -29,7 +29,7 @@ public class ClientSideExplicitEncryptionTest extends AbstractClientSideEncrypti
 
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 
     @Override

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ClientSideOperationTimeoutProseTest.java

@@ -75,18 +75,15 @@ public final class ClientSideOperationTimeoutProseTest extends AbstractClientSid
 
     @Override
     protected com.mongodb.client.MongoClient createMongoClient(final MongoClientSettings mongoClientSettings) {
-        wrapped = createReactiveClient(mongoClientSettings);
-        return new SyncMongoClient(wrapped);
+        SyncMongoClient client = new SyncMongoClient(mongoClientSettings);
+        wrapped = client.getWrapped();
+        return client;
     }
 
     private static MongoClient createReactiveClient(final MongoClientSettings.Builder builder) {
         return MongoClients.create(builder.build());
     }
 
-    private static MongoClient createReactiveClient(final MongoClientSettings mongoClientSettings) {
-        return MongoClients.create(mongoClientSettings);
-    }
-
     @Override
     protected com.mongodb.client.gridfs.GridFSBucket createGridFsBucket(final com.mongodb.client.MongoDatabase mongoDatabase,
                                                                         final String bucketName) {

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/CrudProseTest.java

@@ -26,6 +26,6 @@
 final class CrudProseTest extends com.mongodb.client.CrudProseTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings.Builder mongoClientSettingsBuilder) {
-        return new SyncMongoClient(MongoClients.create(mongoClientSettingsBuilder.build()));
+        return new SyncMongoClient(mongoClientSettingsBuilder);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/DnsConfigurationTest.java

@@ -24,6 +24,6 @@
 public class DnsConfigurationTest extends AbstractDnsConfigurationTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ExplainTest.java

@@ -28,9 +28,7 @@
 public class ExplainTest extends AbstractExplainTest {
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(
-                MongoClientSettings.builder(settings).contextProvider(CONTEXT_PROVIDER).build()
-        ));
+        return new SyncMongoClient(MongoClientSettings.builder(settings).contextProvider(CONTEXT_PROVIDER));
     }
 
     @Test

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/MongoCollectionTest.java

@@ -34,7 +34,7 @@ protected MongoDatabase getDatabase(final String databaseName) {
 
     private com.mongodb.client.MongoClient createMongoClient() {
         if (mongoClient == null) {
-            mongoClient = new SyncMongoClient(MongoClients.create(getMongoClientSettingsBuilder().build()));
+            mongoClient = new SyncMongoClient(getMongoClientSettingsBuilder());
         }
         return mongoClient;
     }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/MongoWriteConcernWithResponseExceptionTest.java

@@ -25,7 +25,6 @@
 public class MongoWriteConcernWithResponseExceptionTest {
     @Test
     public void doesNotLeak() throws InterruptedException {
-        com.mongodb.client.MongoWriteConcernWithResponseExceptionTest.doesNotLeak(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)));
+        com.mongodb.client.MongoWriteConcernWithResponseExceptionTest.doesNotLeak(SyncMongoClient::new);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ReactiveContextProviderTest.java

@@ -109,10 +109,8 @@ List<DynamicTest> testMongoIterableFirstPassesTheContext() {
 
     private MongoClient getMongoClient() {
         if (mongoClient == null) {
-            mongoClient = new SyncMongoClient(
-                    MongoClients.create(getMongoClientSettingsBuilder()
-                            .contextProvider(CONTEXT_PROVIDER)
-                            .build()));
+            mongoClient = new SyncMongoClient(getMongoClientSettingsBuilder()
+                            .contextProvider(CONTEXT_PROVIDER));
         }
         return mongoClient;
     }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ReactiveInitialDnsSeedlistDiscoveryTest.java

@@ -39,6 +39,6 @@ public ReactiveInitialDnsSeedlistDiscoveryTest(final String filename, final Stri
 
     @Override
     public MongoClient createMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/RetryableReadsProseTest.java

@@ -40,7 +40,7 @@ final class RetryableReadsProseTest {
     @Test
     void poolClearedExceptionMustBeRetryable() throws InterruptedException, ExecutionException, TimeoutException {
         RetryableWritesProseTest.poolClearedExceptionMustBeRetryable(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)),
+                SyncMongoClient::new,
                 mongoCollection -> mongoCollection.find(eq(0)).iterator().hasNext(), "find", false);
     }
 
@@ -52,7 +52,7 @@ void poolClearedExceptionMustBeRetryable() throws InterruptedException, Executio
     @Test
     void retriesOnDifferentMongosWhenAvailable() {
         RetryableWritesProseTest.retriesOnDifferentMongosWhenAvailable(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)),
+                SyncMongoClient::new,
                 mongoCollection -> {
                     try (MongoCursor<Document> cursor = mongoCollection.find().iterator()) {
                         return cursor.hasNext();
@@ -68,7 +68,7 @@ void retriesOnDifferentMongosWhenAvailable() {
     @Test
     void retriesOnSameMongosWhenAnotherNotAvailable() {
         RetryableWritesProseTest.retriesOnSameMongosWhenAnotherNotAvailable(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)),
+                SyncMongoClient::new,
                 mongoCollection -> {
                     try (MongoCursor<Document> cursor = mongoCollection.find().iterator()) {
                         return cursor.hasNext();

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/RetryableWritesProseTest.java

@@ -48,7 +48,7 @@ public void setUp() {
     @Test
     public void poolClearedExceptionMustBeRetryable() throws InterruptedException, ExecutionException, TimeoutException {
         com.mongodb.client.RetryableWritesProseTest.poolClearedExceptionMustBeRetryable(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)),
+                SyncMongoClient::new,
                 mongoCollection -> mongoCollection.insertOne(new Document()), "insert", true);
     }
 
@@ -58,7 +58,7 @@ public void poolClearedExceptionMustBeRetryable() throws InterruptedException, E
     @Test
     public void originalErrorMustBePropagatedIfNoWritesPerformed() throws InterruptedException {
         com.mongodb.client.RetryableWritesProseTest.originalErrorMustBePropagatedIfNoWritesPerformed(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)));
+                SyncMongoClient::new);
     }
 
     /**
@@ -67,7 +67,7 @@ public void originalErrorMustBePropagatedIfNoWritesPerformed() throws Interrupte
     @Test
     public void retriesOnDifferentMongosWhenAvailable() {
         com.mongodb.client.RetryableWritesProseTest.retriesOnDifferentMongosWhenAvailable(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)),
+                SyncMongoClient::new,
                 mongoCollection -> mongoCollection.insertOne(new Document()), "insert", true);
     }
 
@@ -77,7 +77,7 @@ public void retriesOnDifferentMongosWhenAvailable() {
     @Test
     public void retriesOnSameMongosWhenAnotherNotAvailable() {
         com.mongodb.client.RetryableWritesProseTest.retriesOnSameMongosWhenAnotherNotAvailable(
-                mongoClientSettings -> new SyncMongoClient(MongoClients.create(mongoClientSettings)),
+                SyncMongoClient::new,
                 mongoCollection -> mongoCollection.insertOne(new Document()), "insert", true);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/ServerSelectionProseTest.java

@@ -22,6 +22,6 @@
 
 final class ServerSelectionProseTest extends AbstractServerSelectionProseTest {
     protected MongoClient createClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/SessionsProseTest.java

@@ -24,6 +24,6 @@
 public class SessionsProseTest extends AbstractSessionsProseTest {
     @Override
     protected MongoClient getMongoClient(final MongoClientSettings settings) {
-        return new SyncMongoClient(MongoClients.create(settings));
+        return new SyncMongoClient(settings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/X509AuthenticationTest.java

@@ -23,6 +23,6 @@
 public class X509AuthenticationTest extends AbstractX509AuthenticationTest {
     @Override
     protected com.mongodb.client.MongoClient createMongoClient(final MongoClientSettings mongoClientSettings) {
-        return new SyncMongoClient(MongoClients.create(mongoClientSettings));
+        return new SyncMongoClient(mongoClientSettings);
     }
 }

driver-reactive-streams/src/test/functional/com/mongodb/reactivestreams/client/csot/ClientSideOperationsEncryptionTimeoutProseTest.java

@@ -33,6 +33,6 @@ public ClientEncryption createClientEncryption(final ClientEncryptionSettings.Bu
 
     @Override
     protected MongoClient createMongoClient(final MongoClientSettings.Builder builder) {
-        return new SyncMongoClient(com.mongodb.reactivestreams.client.MongoClients.create(builder.build()));
+        return new SyncMongoClient(builder);
     }
 }