CSHARP-2309: Transaction JSON tests should be run using the async API also.

Author: rstam

tests/MongoDB.Driver.Tests/JsonDrivenTests/JsonDrivenAbortTransactionTest.cs

@@ -15,6 +15,7 @@
 
 using System.Collections.Generic;
 using System.Threading;
+using System.Threading.Tasks;
 using MongoDB.Bson;
 using MongoDB.Bson.TestHelpers.JsonDrivenTests;
 
@@ -44,5 +45,10 @@ protected override void CallMethod(CancellationToken cancellationToken)
         {
             _session.AbortTransaction(cancellationToken);
         }
+
+        protected override Task CallMethodAsync(CancellationToken cancellationToken)
+        {
+            return _session.AbortTransactionAsync(cancellationToken);
+        }
     }
 }

tests/MongoDB.Driver.Tests/JsonDrivenTests/JsonDrivenCommitTransactionTest.cs

@@ -15,6 +15,7 @@
 
 using System.Collections.Generic;
 using System.Threading;
+using System.Threading.Tasks;
 using MongoDB.Bson;
 using MongoDB.Bson.TestHelpers.JsonDrivenTests;
 
@@ -44,5 +45,10 @@ protected override void CallMethod(CancellationToken cancellationToken)
         {
             _session.CommitTransaction(cancellationToken);
         }
+
+        protected override Task CallMethodAsync(CancellationToken cancellationToken)
+        {
+            return _session.CommitTransactionAsync(cancellationToken);
+        }
     }
 }

tests/MongoDB.Driver.Tests/JsonDrivenTests/JsonDrivenStartTransactionTest.cs

@@ -15,6 +15,7 @@
 
 using System.Collections.Generic;
 using System.Threading;
+using System.Threading.Tasks;
 using MongoDB.Bson;
 using MongoDB.Bson.TestHelpers.JsonDrivenTests;
 
@@ -48,6 +49,12 @@ protected override void CallMethod(CancellationToken cancellationToken)
             _session.StartTransaction(_options);
         }
 
+        protected override Task CallMethodAsync(CancellationToken cancellationToken)
+        {
+            _session.StartTransaction(_options); // there is no async version, just call the sync version
+            return Task.FromResult(true);
+        }
+
         protected override void SetArgument(string name, BsonValue value)
         {
             switch (name)

tests/MongoDB.Driver.Tests/Specifications/transactions/TransactionTestRunner.cs

@@ -77,7 +77,7 @@ private void Run(BsonDocument shared, BsonDocument test)
             //}
 
             JsonDrivenHelper.EnsureAllFieldsAreValid(shared, "_path", "database_name", "collection_name", "data", "tests");
-            JsonDrivenHelper.EnsureAllFieldsAreValid(test, "description", "clientOptions", "failPoint", "sessionOptions", "operations", "expectations", "outcome");
+            JsonDrivenHelper.EnsureAllFieldsAreValid(test, "description", "clientOptions", "failPoint", "sessionOptions", "operations", "expectations", "outcome", "async");
 
             _databaseName = shared["database_name"].AsString;
             _collectionName = shared["collection_name"].AsString;
@@ -281,7 +281,14 @@ private void ExecuteOperations(IMongoClient client, Dictionary<string, object> o
                 var jsonDrivenTest = factory.CreateTest(receiver, name);
 
                 jsonDrivenTest.Arrange(operation);
-                jsonDrivenTest.Act(CancellationToken.None);
+                if (test["async"].AsBoolean)
+                {
+                    jsonDrivenTest.ActAsync(CancellationToken.None).GetAwaiter().GetResult();
+                }
+                else
+                {
+                    jsonDrivenTest.Act(CancellationToken.None);
+                }
                 jsonDrivenTest.Assert();
             }
         }
@@ -415,6 +422,20 @@ protected override string PathPrefix
                 }
             }
 
+            // protected methods
+            protected override IEnumerable<JsonDrivenTestCase> CreateTestCases(BsonDocument document)
+            {
+                foreach (var testCase in base.CreateTestCases(document))
+                {
+                    foreach (var async in new[] { false, true })
+                    {
+                        var name = $"{testCase.Name}:async={async}";
+                        var test = testCase.Test.DeepClone().AsBsonDocument.Add("async", async);
+                        yield return new JsonDrivenTestCase(name, testCase.Shared, test);
+                    }
+                }
+            }
+
             protected override bool ShouldReadJsonDocument(string path)
             {
                 return base.ShouldReadJsonDocument(path); // && path.EndsWith("commit.json");