CSHARP-5930: Enable SmoteTests and Driver.Examples running locally

I have one env var set: MONGODB_URI

I'm using Atlas Local 8.2, which is a standalone cluster by default, but tests that pass were being excluded from running. If people are using non-Atlas and this breaks, then we'll have to figure something out.

Author: ajcvickers

src/MongoDB.Driver/Encryption/AutoEncryptionProviderRegistry.cs

@@ -28,10 +28,16 @@ internal sealed class AutoEncryptionProviderRegistry : IAutoEncryptionProviderRe
 
         public void Register(Func<IMongoClient, AutoEncryptionOptions, IAutoEncryptionLibMongoCryptController> factory)
         {
+            if(ReferenceEquals(factory, _autoCryptClientControllerFactory))
+            {
+                return;
+            }
+
             if (_autoCryptClientControllerFactory != null)
             {
                 throw new MongoConfigurationException("AutoEncryption Provider already registered.");
             }
+
             _autoCryptClientControllerFactory = factory;
         }
 

tests/MongoDB.Driver.Examples/CausalConsistencyExamples.cs

@@ -120,9 +120,10 @@ public void Causal_Consistency_Example_3()
             DropCollection(CreateClient(), "myDatabase", "myCollection");
 
             // Start Tunable Consistency Controls Example
-            var connectionString = "mongodb://localhost/?readPreference=secondaryPreferred";
+            var clientSettings = MongoClientSettings.FromConnectionString(CoreTestConfiguration.ConnectionString.ToString());
+            clientSettings.ReadPreference = ReadPreference.SecondaryPreferred;
 
-            var client = new MongoClient(connectionString);
+            var client = new MongoClient(clientSettings);
             var database = client.GetDatabase("myDatabase");
             var collection = database.GetCollection<BsonDocument>("myCollection");
 
@@ -147,11 +148,7 @@ public void Causal_Consistency_Example_3()
             result.Should().Be("{ name: 'John' }");
         }
 
-        private IMongoClient CreateClient()
-        {
-            var connectionString = CoreTestConfiguration.ConnectionString.ToString();
-            return new MongoClient(connectionString);
-        }
+        private IMongoClient CreateClient() => new MongoClient(CoreTestConfiguration.ConnectionString.ToString());
 
         private void DropCollection(IMongoClient client, string databaseName, string collectionName)
         {

tests/MongoDB.Driver.Examples/ChangeStreamExamples.cs

@@ -37,12 +37,13 @@ public void ChangeStreamExample1()
             var inventory = database.GetCollection<BsonDocument>("inventory");
 
             var document = new BsonDocument("x", 1);
-            new Thread(() =>
+            var thread = new Thread(() =>
             {
                 Thread.Sleep(TimeSpan.FromMilliseconds(100));
                 inventory.InsertOne(document);
-            })
-            .Start();
+            });
+
+            thread.Start();
 
             // Start Changestream Example 1
             var cursor = inventory.Watch();
@@ -52,6 +53,9 @@ public void ChangeStreamExample1()
             // End Changestream Example 1
 
             next.FullDocument.Should().Be(document);
+
+            // Make sure worker thread is finished before we let the next test run.
+            thread.Join();
         }
 
         [Fact]
@@ -64,14 +68,15 @@ public void ChangeStreamExample2()
 
             var document = new BsonDocument("x", 1);
             inventory.InsertOne(document);
-            new Thread(() =>
+            var thread = new Thread(() =>
             {
                 Thread.Sleep(TimeSpan.FromMilliseconds(100));
                 var filter = new BsonDocument("_id", document["_id"]);
                 var update = "{ $set : { x : 2 } }";
                 inventory.UpdateOne(filter, update);
-            })
-            .Start();
+            });
+
+            thread.Start();
 
             // Start Changestream Example 2
             var options = new ChangeStreamOptions { FullDocument = ChangeStreamFullDocumentOption.UpdateLookup };
@@ -83,6 +88,9 @@ public void ChangeStreamExample2()
 
             var expectedFullDocument = document.Set("x", 2);
             next.FullDocument.Should().Be(expectedFullDocument);
+
+            // Make sure worker thread is finished before we let the next test run.
+            thread.Join();
         }
 
         [Fact]
@@ -101,15 +109,19 @@ public void ChangeStreamExample3()
 
             IChangeStreamCursor<ChangeStreamDocument<BsonDocument>> previousCursor;
             {
-                new Thread(() =>
+                var thread = new Thread(() =>
                 {
                     Thread.Sleep(TimeSpan.FromMilliseconds(100));
                     inventory.InsertMany(documents);
-                })
-                .Start();
+                });
+
+                thread.Start();
 
                 previousCursor = inventory.Watch(new ChangeStreamOptions { BatchSize = 1 });
                 while (previousCursor.MoveNext() && previousCursor.Current.Count() == 0) { } // keep calling MoveNext until we've read the first batch
+
+                // Make sure worker thread is finished before we let the next test run.
+                thread.Join();
             }
 
             {
@@ -135,22 +147,24 @@ public void ChangestreamExample4()
             var database = client.GetDatabase("ChangeStreamExamples");
             database.DropCollection("inventory");
 
-            using var cancelationTokenSource = new CancellationTokenSource();
-            try
+            var workerEnabled = true;
+            var document = new BsonDocument("username", "alice");
+
+            var thread = new Thread(() =>
             {
-                var document = new BsonDocument("username", "alice");
+                var inventoryCollection = database.GetCollection<BsonDocument>("inventory");
 
-                Task.Run(() =>
+                while (workerEnabled)
                 {
-                    var inventoryCollection = database.GetCollection<BsonDocument>("inventory");
+                    Thread.Sleep(TimeSpan.FromMilliseconds(100));
+                    document["_id"] = ObjectId.GenerateNewId();
+                    inventoryCollection.InsertOne(document);
+                }
+            });
 
-                    while (!cancelationTokenSource.IsCancellationRequested)
-                    {
-                        Thread.Sleep(TimeSpan.FromMilliseconds(100));
-                        document["_id"] = ObjectId.GenerateNewId();
-                        inventoryCollection.InsertOne(document);
-                    }
-                }, cancelationTokenSource.Token);
+            try
+            {
+                thread.Start();
 
                 // Start Changestream Example 4
                 var pipeline = new EmptyPipelineDefinition<ChangeStreamDocument<BsonDocument>>()
@@ -170,7 +184,11 @@ public void ChangestreamExample4()
             }
             finally
             {
-                cancelationTokenSource.Cancel();
+                workerEnabled = false;
+
+                // Make sure worker thread is finished before we let the next test run.
+                // Especially important for this test case since the thread continually inserts documents.
+                thread.Join();
             }
         }
     }

tests/MongoDB.Driver.Examples/ClientEncryptionExamples.cs

@@ -56,10 +56,8 @@ public void ClientSideEncryptionSimpleTour()
             var keyVaultNamespace = CollectionNamespace.FromFullName("encryption.__keyVault");
             var autoEncryptionOptions = new AutoEncryptionOptions(keyVaultNamespace, kmsProviders);
 
-            var mongoClientSettings = new MongoClientSettings
-            {
-                AutoEncryptionOptions = autoEncryptionOptions
-            };
+            var mongoClientSettings = MongoClientSettings.FromConnectionString(InfrastructureUtilities.MongoUri);
+            mongoClientSettings.AutoEncryptionOptions = autoEncryptionOptions;
 
             var client = new MongoClient(mongoClientSettings);
             var database = client.GetDatabase("test");
@@ -87,7 +85,7 @@ public void ClientSideEncryptionAutoEncryptionSettingsTour()
             kmsProviders.Add("local", localKey);
 
             var keyVaultNamespace = CollectionNamespace.FromFullName("encryption.__keyVault");
-            var keyVaultMongoClient = new MongoClient();
+            var keyVaultMongoClient = new MongoClient(InfrastructureUtilities.MongoUri);
             var clientEncryptionSettings = new ClientEncryptionOptions(
                 keyVaultMongoClient,
                 keyVaultNamespace,
@@ -124,10 +122,9 @@ public void ClientSideEncryptionAutoEncryptionSettingsTour()
                 {
                     { collectionNamespace.ToString(), BsonDocument.Parse(schemaMap) }
                 });
-            var clientSettings = new MongoClientSettings
-            {
-                AutoEncryptionOptions = autoEncryptionSettings
-            };
+
+            var clientSettings = MongoClientSettings.FromConnectionString(InfrastructureUtilities.MongoUri);
+            clientSettings.AutoEncryptionOptions = autoEncryptionSettings;
 
             var client = new MongoClient(clientSettings);
             var database = client.GetDatabase("test");

tests/MongoDB.Driver.Examples/ClientSideEncryption2Examples.cs

@@ -37,7 +37,7 @@ public class ClientSideEncryption2Examples
         [Fact]
         public void FLE2AutomaticEncryption()
         {
-            RequireServer.Check().Supports(Feature.Csfle2).ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded, ClusterType.LoadBalanced);
+            RequireServer.Check().Supports(Feature.Csfle2).ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded, ClusterType.LoadBalanced, ClusterType.Standalone);
 
             var unencryptedClient = DriverTestConfiguration.Client;
 
@@ -52,7 +52,7 @@ public void FLE2AutomaticEncryption()
             };
             kmsProviders.Add("local", localKey);
 
-            var keyVaultClient = new MongoClient();
+            var keyVaultClient = new MongoClient(InfrastructureUtilities.MongoUri);
 
             // Create two data keys.
             var clientEncryptionOptions = new ClientEncryptionOptions(keyVaultClient, KeyVaultNamespace, kmsProviders);
@@ -92,7 +92,9 @@ public void FLE2AutomaticEncryption()
             };
 
             var autoEncryptionOptions = new AutoEncryptionOptions(KeyVaultNamespace, kmsProviders, encryptedFieldsMap: encryptedFieldsMap);
-            var encryptedClient = new MongoClient(new MongoClientSettings { AutoEncryptionOptions = autoEncryptionOptions });
+            var mongoClientSettings = MongoClientSettings.FromConnectionString(InfrastructureUtilities.MongoUri);
+            mongoClientSettings.AutoEncryptionOptions = autoEncryptionOptions;
+            var encryptedClient = new MongoClient(mongoClientSettings);
 
             // Create an FLE 2 collection.
             var database = encryptedClient.GetDatabase(CollectionNamespace.DatabaseNamespace.DatabaseName);

tests/MongoDB.Driver.Examples/DocumentationExamples.cs

@@ -987,7 +987,7 @@ public void Example_50()
             // End Example 50
 
             Render(filter).Should().Be("{ status: \"A\" }");
-            Render(projection).Should().Be("{ item: 1, status: 1, instock: { $slice: -1 } }");
+            Render(projection).Should().Be("{ item: 1, status: 1, instock: { $slice: [ '$instock', -1 ] } }");
         }
 
         [Fact]

tests/MongoDB.Driver.Examples/ExplicitEncryptionExamples.cs

@@ -54,7 +54,8 @@ public void ClientSideExplicitEncryptionAndDecryptionTour()
             kmsProviders.Add("local", localKey);
 
             var keyVaultNamespace = CollectionNamespace.FromFullName("encryption.__keyVault");
-            var keyVaultClient = new MongoClient("mongodb://localhost");
+            var clientSettings = MongoClientSettings.FromConnectionString(InfrastructureUtilities.MongoUri);
+            var keyVaultClient = new MongoClient(clientSettings);
             var keyVaultDatabase = keyVaultClient.GetDatabase(keyVaultNamespace.DatabaseNamespace.DatabaseName);
             keyVaultDatabase.DropCollection(keyVaultNamespace.CollectionName);
 
@@ -109,14 +110,15 @@ public void ClientSideExplicitEncryptionAndAutoDecryptionTour()
                 keyVaultNamespace,
                 kmsProviders,
                 bypassAutoEncryption: true);
-            var clientSettings = MongoClientSettings.FromConnectionString("mongodb://localhost");
+            var clientSettings = MongoClientSettings.FromConnectionString(InfrastructureUtilities.MongoUri);
             clientSettings.AutoEncryptionOptions = autoEncryptionOptions;
             var mongoClient = new MongoClient(clientSettings);
             var database = mongoClient.GetDatabase(collectionNamespace.DatabaseNamespace.DatabaseName);
             database.DropCollection(collectionNamespace.CollectionName);
             var collection = database.GetCollection<BsonDocument>(collectionNamespace.CollectionName);
 
-            var keyVaultClient = new MongoClient("mongodb://localhost");
+            var keyVaultClientSettings = MongoClientSettings.FromConnectionString(InfrastructureUtilities.MongoUri);
+            var keyVaultClient = new MongoClient(keyVaultClientSettings);
             var keyVaultDatabase = keyVaultClient.GetDatabase(keyVaultNamespace.DatabaseNamespace.DatabaseName);
             keyVaultDatabase.DropCollection(keyVaultNamespace.CollectionName);
 

tests/MongoDB.Driver.Examples/InfrastructureUtilities.cs

@@ -0,0 +1,25 @@
+/* Copyright 2017-present MongoDB Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+using System;
+
+namespace MongoDB.Driver.Examples;
+
+internal static class InfrastructureUtilities
+{
+    public static readonly string MongoUri = Environment.GetEnvironmentVariable("MONGODB_URI") ??
+                                             Environment.GetEnvironmentVariable("MONGO_URI") ??
+                                             "mongodb://localhost";
+}

tests/MongoDB.Driver.Examples/StableApiExamples.cs

@@ -73,7 +73,7 @@ public void ConfigureServerApiDeprecationErrors()
         [Fact]
         public void StableAPI_Strict_Migration_Example()
         {
-            var connectionString = "mongodb://localhost";
+            var connectionString = InfrastructureUtilities.MongoUri;
             var serverApi = new ServerApi(ServerApiVersion.V1, strict: true);
             var mongoClientSettings = MongoClientSettings.FromConnectionString(connectionString);
             mongoClientSettings.ServerApi = serverApi;

tests/MongoDB.Driver.Examples/TransactionExamplesForDocs/WithTransactionExample1.cs

@@ -29,7 +29,7 @@ public class WithTransactionExample1
         [Fact]
         public void Example1()
         {
-            RequireServer.Check().ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded).Supports(Feature.Transactions);
+            RequireServer.Check().ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded, ClusterType.Standalone).Supports(Feature.Transactions);
 
             var connectionString = CoreTestConfiguration.ConnectionString.ToString();
             DropCollections(