CSHARP-6069: Add await to async tests missing await (#2028)

mongodb-sage-bot[bot] · web-flow · commit 0e2f6bf052c8 · 2026-06-05T17:53:40.000-07:00
diff --git a/tests/MongoDB.Driver.Tests/IMongoClientExtensionsTests.cs b/tests/MongoDB.Driver.Tests/IMongoClientExtensionsTests.cs
@@ -14,6 +14,7 @@
 */
 
 using System.Threading;
+using System.Threading.Tasks;
 using MongoDB.Bson;
 using MongoDB.TestHelpers.XunitExtensions;
 using Moq;
@@ -25,7 +26,7 @@ public class IMongoClientExtensionsTests
     {
         [Theory]
         [ParameterAttributeData]
-        public void Watch_should_call_client_with_expected_arguments(
+        public async Task Watch_should_call_client_with_expected_arguments(
             [Values(false, true)] bool usingSession,
             [Values(false, true)] bool async)
         {
@@ -40,7 +41,7 @@ public void Watch_should_call_client_with_expected_arguments(
             {
                 if (async)
                 {
-                    client.WatchAsync(session, options, cancellationToken);
+                    await client.WatchAsync(session, options, cancellationToken);
                     mockClient.Verify(m => m.WatchAsync(session, It.IsAny<EmptyPipelineDefinition<ChangeStreamDocument<BsonDocument>>>(), options, cancellationToken), Times.Once);
                 }
                 else
@@ -53,7 +54,7 @@ public void Watch_should_call_client_with_expected_arguments(
             {
                 if (async)
                 {
-                    client.WatchAsync(options, cancellationToken);
+                    await client.WatchAsync(options, cancellationToken);
                     mockClient.Verify(m => m.WatchAsync(It.IsAny<EmptyPipelineDefinition<ChangeStreamDocument<BsonDocument>>>(), options, cancellationToken), Times.Once);
                 }
                 else
diff --git a/tests/MongoDB.Driver.Tests/IMongoDatabaseExtensionsTests.cs b/tests/MongoDB.Driver.Tests/IMongoDatabaseExtensionsTests.cs
@@ -14,6 +14,7 @@
 */
 
 using System.Threading;
+using System.Threading.Tasks;
 using MongoDB.Bson;
 using MongoDB.TestHelpers.XunitExtensions;
 using Moq;
@@ -25,7 +26,7 @@ public class IMongoDatabaseExtensionsTests
     {
         [Theory]
         [ParameterAttributeData]
-        public void Watch_should_call_client_with_expected_arguments(
+        public async Task Watch_should_call_client_with_expected_arguments(
             [Values(false, true)] bool usingSession,
             [Values(false, true)] bool async)
         {
@@ -40,7 +41,7 @@ public void Watch_should_call_client_with_expected_arguments(
             {
                 if (async)
                 {
-                    database.WatchAsync(session, options, cancellationToken);
+                    await database.WatchAsync(session, options, cancellationToken);
                     mockDatabase.Verify(m => m.WatchAsync(session, It.IsAny<EmptyPipelineDefinition<ChangeStreamDocument<BsonDocument>>>(), options, cancellationToken), Times.Once);
                 }
                 else
@@ -53,7 +54,7 @@ public void Watch_should_call_client_with_expected_arguments(
             {
                 if (async)
                 {
-                    database.WatchAsync(options, cancellationToken);
+                    await database.WatchAsync(options, cancellationToken);
                     mockDatabase.Verify(m => m.WatchAsync(It.IsAny<EmptyPipelineDefinition<ChangeStreamDocument<BsonDocument>>>(), options, cancellationToken), Times.Once);
                 }
                 else
diff --git a/tests/MongoDB.Driver.Tests/OfTypeMongoCollectionTests.cs b/tests/MongoDB.Driver.Tests/OfTypeMongoCollectionTests.cs

Original file line number	Diff line number	Diff line change
`@@ -14,6 +14,7 @@`
`14`	`14`	`*/`
`15`	`15`
`16`	`16`	`using System.Threading;`
	`17`	`+using System.Threading.Tasks;`
`17`	`18`	`using MongoDB.Bson;`
`18`	`19`	`using MongoDB.TestHelpers.XunitExtensions;`
`19`	`20`	`using Moq;`
`@@ -25,7 +26,7 @@ public class IMongoClientExtensionsTests`
`25`	`26`	`{`
`26`	`27`	`[Theory]`
`27`	`28`	`[ParameterAttributeData]`
`28`		`- public void Watch_should_call_client_with_expected_arguments(`
	`29`	`+ public async Task Watch_should_call_client_with_expected_arguments(`
`29`	`30`	`[Values(false, true)] bool usingSession,`
`30`	`31`	`[Values(false, true)] bool async)`
`31`	`32`	`{`
`@@ -40,7 +41,7 @@ public void Watch_should_call_client_with_expected_arguments(`
`40`	`41`	`{`
`41`	`42`	`if (async)`
`42`	`43`	`{`
`43`		`- client.WatchAsync(session, options, cancellationToken);`
	`44`	`+ await client.WatchAsync(session, options, cancellationToken);`
`44`	`45`	`mockClient.Verify(m => m.WatchAsync(session, It.IsAny<EmptyPipelineDefinition<ChangeStreamDocument<BsonDocument>>>(), options, cancellationToken), Times.Once);`
`45`	`46`	`}`
`46`	`47`	`else`
`@@ -53,7 +54,7 @@ public void Watch_should_call_client_with_expected_arguments(`
`53`	`54`	`{`
`54`	`55`	`if (async)`
`55`	`56`	`{`
`56`		`- client.WatchAsync(options, cancellationToken);`
	`57`	`+ await client.WatchAsync(options, cancellationToken);`
`57`	`58`	`mockClient.Verify(m => m.WatchAsync(It.IsAny<EmptyPipelineDefinition<ChangeStreamDocument<BsonDocument>>>(), options, cancellationToken), Times.Once);`
`58`	`59`	`}`
`59`	`60`	`else`