Fix ClientSessionHandleTests for tests.

Author: adelinowona

src/MongoDB.Driver/ClientSessionHandle.cs

@@ -160,7 +160,7 @@ public void StartTransaction(TransactionOptions transactionOptions = null)
             var effectiveTransactionOptions = GetEffectiveTransactionOptions(transactionOptions);
 
             // Check if tracing is enabled for this client
-            var tracingOptions = _client.Settings.TracingOptions;
+            var tracingOptions = _client?.Settings?.TracingOptions;
             var isTracingEnabled = tracingOptions == null || !tracingOptions.Disabled;
 
             ((ICoreSessionInternal)_coreSession).StartTransaction(effectiveTransactionOptions, isTracingEnabled);

tests/MongoDB.Driver.Tests/ClientSessionHandleTests.cs

@@ -275,7 +275,7 @@ public void StartTransaction_should_call_coreSession()
 
             subject.StartTransaction(transactionOptions);
 
-            Mock.Get(subject.WrappedCoreSession).Verify(m => m.StartTransaction(It.IsAny<TransactionOptions>()), Times.Once);
+            Mock.Get(subject.WrappedCoreSession).As<ICoreSessionInternal>().Verify(m => m.StartTransaction(It.IsAny<TransactionOptions>(), It.IsAny<bool>()), Times.Once);
         }
 
         [Theory]
@@ -405,7 +405,7 @@ public void WithTransaction_callback_should_be_processed_with_expected_result(
                 mockCoreSession.Verify(handle => handle.CommitTransaction(It.IsAny<CancellationToken>()), Times.Exactly(expectedCommitTransactionAttempts));
             }
 
-            mockCoreSession.Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>()), Times.Exactly(expectedStartTransactionAttempts));
+            mockCoreSession.As<ICoreSessionInternal>().Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>(), It.IsAny<bool>()), Times.Exactly(expectedStartTransactionAttempts));
         }
 
         [Theory]
@@ -428,7 +428,7 @@ public void WithTransaction_callback_with_a_custom_error_should_not_be_retried()
 
             Assert.Throws<MongoException>(() => subject.WithTransaction<bool>((handle, cancellationToken) => throw new MongoException("test")));
 
-            mockCoreSession.Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>()), Times.Once);
+            mockCoreSession.As<ICoreSessionInternal>().Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>(), It.IsAny<bool>()), Times.Once);
             mockCoreSession.Verify(handle => handle.CommitTransaction(It.IsAny<CancellationToken>()), Times.Never);
         }
 
@@ -595,7 +595,7 @@ public void WithTransaction_with_error_in_callback_should_call_AbortTransaction_
 
             Assert.Throws<Exception>(() => subject.WithTransaction<object>((handle, cancellationToken) => throw new Exception("test")));
 
-            mockCoreSession.Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>()), Times.Once);
+            mockCoreSession.As<ICoreSessionInternal>().Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>(), It.IsAny<bool>()), Times.Once);
             mockCoreSession.Verify(handle => handle.AbortTransaction(It.IsAny<CancellationToken>()), shouldAbortTransactionBeCalled ? Times.Once() : Times.Never());
             mockCoreSession.Verify(handle => handle.CommitTransaction(It.IsAny<CancellationToken>()), Times.Never);
         }
@@ -604,13 +604,13 @@ public void WithTransaction_with_error_in_callback_should_call_AbortTransaction_
         public void WithTransaction_with_error_in_StartTransaction_should_return_control_immediately()
         {
             var mockCoreSession = CreateCoreSessionMock();
-            mockCoreSession
-                .Setup(c => c.StartTransaction(It.IsAny<TransactionOptions>()))
+            mockCoreSession.As<ICoreSessionInternal>()
+                .Setup(c => c.StartTransaction(It.IsAny<TransactionOptions>(), It.IsAny<bool>()))
                 .Throws<Exception>();
             var subject = CreateSubject(coreSession: mockCoreSession.Object);
 
             Assert.Throws<Exception>(() => subject.WithTransaction<object>((handle, cancellationToken) => 1));
-            mockCoreSession.Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>()), Times.Once);
+            mockCoreSession.As<ICoreSessionInternal>().Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>(), It.IsAny<bool>()), Times.Once);
             mockCoreSession.Verify(handle => handle.AbortTransaction(It.IsAny<CancellationToken>()), Times.Never);
             mockCoreSession.Verify(handle => handle.CommitTransaction(It.IsAny<CancellationToken>()), Times.Never);
         }
@@ -625,7 +625,7 @@ public void WithTransaction_without_errors_should_call_transaction_infrastructur
 
             subject.WithTransaction<object>((handle, cancellationToken) => 1);
 
-            mockCoreSession.Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>()), Times.Once);
+            mockCoreSession.As<ICoreSessionInternal>().Verify(handle => handle.StartTransaction(It.IsAny<TransactionOptions>(), It.IsAny<bool>()), Times.Once);
             mockCoreSession.Verify(handle => handle.CommitTransaction(It.IsAny<CancellationToken>()), Times.Once);
         }
 
@@ -638,6 +638,7 @@ private Mock<ICoreSessionHandle> CreateCoreSessionMock(
             options = options ?? new CoreSessionOptions();
 
             var mockCoreSession = new Mock<ICoreSessionHandle>();
+            mockCoreSession.As<ICoreSessionInternal>();
             mockCoreSession.Setup(m => m.CurrentTransaction).Returns(new CoreTransaction(It.IsAny<long>(), It.IsAny<TransactionOptions>()));
             mockCoreSession.SetupGet(m => m.Options).Returns(options);
             mockCoreSession.SetupGet(m => m.ServerSession).Returns(serverSession);