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

ajcvickers · ajcvickers · commit 9b311b0da62e · 2026-03-19T12:35:05.000Z
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.
diff --git a/src/MongoDB.Driver/Encryption/AutoEncryptionProviderRegistry.cs b/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;
         }
 
diff --git a/tests/MongoDB.Driver.Examples/CausalConsistencyExamples.cs b/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)
         {
diff --git a/tests/MongoDB.Driver.Examples/ChangeStreamExamples.cs b/tests/MongoDB.Driver.Examples/ChangeStreamExamples.cs
@@ -16,7 +16,6 @@
 using System;
 using System.Linq;
 using System.Threading;
-using System.Threading.Tasks;
 using FluentAssertions;
 using MongoDB.Bson;
 using MongoDB.Driver.Core.TestHelpers.XunitExtensions;
@@ -37,12 +36,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 +52,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 +67,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 +87,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 +108,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 +146,24 @@ public void ChangestreamExample4()
             var database = client.GetDatabase("ChangeStreamExamples");
             database.DropCollection("inventory");
 
-            using var cancelationTokenSource = new CancellationTokenSource();
-            try
+            var stopEvent = new ManualResetEventSlim(false);
+            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 (!stopEvent.IsSet)
                 {
-                    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 +183,12 @@ public void ChangestreamExample4()
             }
             finally
             {
-                cancelationTokenSource.Cancel();
+                stopEvent.Set();
+
+                // Make sure the 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();
+                stopEvent.Dispose();
             }
         }
     }
diff --git a/tests/MongoDB.Driver.Examples/ClientEncryptionExamples.cs b/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");
diff --git a/tests/MongoDB.Driver.Examples/ClientSideEncryption2Examples.cs b/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);
diff --git a/tests/MongoDB.Driver.Examples/DocumentationExamples.cs b/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]
diff --git a/tests/MongoDB.Driver.Examples/ExplicitEncryptionExamples.cs b/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);
 
diff --git a/tests/MongoDB.Driver.Examples/InfrastructureUtilities.cs b/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";
+}
diff --git a/tests/MongoDB.Driver.Examples/StableApiExamples.cs b/tests/MongoDB.Driver.Examples/StableApiExamples.cs
@@ -26,7 +26,7 @@ public class StableApiExamples
         public void ConfigureServerApi()
         {
             // Start Stable API Example 1
-            var connectionString = "mongodb://localhost";
+            var connectionString = InfrastructureUtilities.MongoUri;
             var serverApi = new ServerApi(ServerApiVersion.V1);
             var mongoClientSettings = MongoClientSettings.FromConnectionString(connectionString);
             mongoClientSettings.ServerApi = serverApi;
@@ -38,7 +38,7 @@ public void ConfigureServerApi()
         public void ConfigureServerApiStrict()
         {
             // Start Stable API Example 2
-            var connectionString = "mongodb://localhost";
+            var connectionString = InfrastructureUtilities.MongoUri;
             var serverApi = new ServerApi(ServerApiVersion.V1, strict: true);
             var mongoClientSettings = MongoClientSettings.FromConnectionString(connectionString);
             mongoClientSettings.ServerApi = serverApi;
@@ -50,7 +50,7 @@ public void ConfigureServerApiStrict()
         public void ConfigureServerApiNonStrict()
         {
             // Start Stable API Example 3
-            var connectionString = "mongodb://localhost";
+            var connectionString = InfrastructureUtilities.MongoUri;
             var serverApi = new ServerApi(ServerApiVersion.V1, strict: false);
             var mongoClientSettings = MongoClientSettings.FromConnectionString(connectionString);
             mongoClientSettings.ServerApi = serverApi;
@@ -62,7 +62,7 @@ public void ConfigureServerApiNonStrict()
         public void ConfigureServerApiDeprecationErrors()
         {
             // Start Stable API Example 4
-            var connectionString = "mongodb://localhost";
+            var connectionString = InfrastructureUtilities.MongoUri;
             var serverApi = new ServerApi(ServerApiVersion.V1, deprecationErrors: true);
             var mongoClientSettings = MongoClientSettings.FromConnectionString(connectionString);
             mongoClientSettings.ServerApi = serverApi;
@@ -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;
diff --git a/tests/MongoDB.Driver.Examples/TransactionExamplesForDocs/WithTransactionExample1.cs b/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(
diff --git a/tests/MongoDB.Driver.TestHelpers/Core/XunitExtensions/RequireServer.cs b/tests/MongoDB.Driver.TestHelpers/Core/XunitExtensions/RequireServer.cs
@@ -172,7 +172,7 @@ public RequireServer Supports(params Feature[] features)
 
         public RequireServer SupportsCausalConsistency()
         {
-            return ClusterTypes(Clusters.ClusterType.Sharded, Clusters.ClusterType.ReplicaSet).SupportsSessions();
+            return ClusterTypes(Clusters.ClusterType.Sharded, Clusters.ClusterType.ReplicaSet, Clusters.ClusterType.Standalone).SupportsSessions();
         }
 
         public RequireServer SupportsSessions()
diff --git a/tests/MongoDB.TestHelpers/XunitExtensions/TimeoutEnforcing/TimeoutEnforcingXunitTestAssemblyRunner.cs b/tests/MongoDB.TestHelpers/XunitExtensions/TimeoutEnforcing/TimeoutEnforcingXunitTestAssemblyRunner.cs
@@ -41,7 +41,7 @@ public TimeoutEnforcingXunitTestAssemblyRunner(
                 executionMessageSink,
                 executionOptions)
         {
-            _unobservedExceptionTrackingTestCase = testCases.SingleOrDefault(IsUnobservedExceptionTrackingTestCase);
+            _unobservedExceptionTrackingTestCase = testCases.FirstOrDefault(IsUnobservedExceptionTrackingTestCase);
         }
 
         protected override Task<RunSummary> RunTestCollectionAsync(
diff --git a/tests/SmokeTests/MongoDB.Driver.SmokeTests.Sdk/LibmongocryptTests.cs b/tests/SmokeTests/MongoDB.Driver.SmokeTests.Sdk/LibmongocryptTests.cs
@@ -59,7 +59,7 @@ public void Explicit_encryption_with_libmongocrypt_package_works()
                 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,

Original file line number	Diff line number	Diff line change
`@@ -28,10 +28,16 @@ internal sealed class AutoEncryptionProviderRegistry : IAutoEncryptionProviderRe`
`28`	`28`
`29`	`29`	`public void Register(Func<IMongoClient, AutoEncryptionOptions, IAutoEncryptionLibMongoCryptController> factory)`
`30`	`30`	`{`
	`31`	`+ if(ReferenceEquals(factory, _autoCryptClientControllerFactory))`
	`32`	`+ {`
	`33`	`+ return;`
	`34`	`+ }`
	`35`	`+`
`31`	`36`	`if (_autoCryptClientControllerFactory != null)`
`32`	`37`	`{`
`33`	`38`	`throw new MongoConfigurationException("AutoEncryption Provider already registered.");`
`34`	`39`	`}`
	`40`	`+`
`35`	`41`	`_autoCryptClientControllerFactory = factory;`
`36`	`42`	`}`
`37`	`43`
Original file line number	Diff line number	Diff line change
`@@ -987,7 +987,7 @@ public void Example_50()`
`987`	`987`	`// End Example 50`
`988`	`988`
`989`	`989`	`Render(filter).Should().Be("{ status: \"A\" }");`
`990`		`- Render(projection).Should().Be("{ item: 1, status: 1, instock: { $slice: -1 } }");`
	`990`	`+ Render(projection).Should().Be("{ item: 1, status: 1, instock: { $slice: [ '$instock', -1 ] } }");`
`991`	`991`	`}`
`992`	`992`
`993`	`993`	`[Fact]`
Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ public class WithTransactionExample1`
`29`	`29`	`[Fact]`
`30`	`30`	`public void Example1()`
`31`	`31`	`{`
`32`		`- RequireServer.Check().ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded).Supports(Feature.Transactions);`
	`32`	`+ RequireServer.Check().ClusterTypes(ClusterType.ReplicaSet, ClusterType.Sharded, ClusterType.Standalone).Supports(Feature.Transactions);`
`33`	`33`
`34`	`34`	`var connectionString = CoreTestConfiguration.ConnectionString.ToString();`
`35`	`35`	`DropCollections(`