Update tests to pass CancellationToken and fix builder usage

Explicitly pass TestContext.Current.CancellationToken to async
database method calls in unit tests for improved reliability.
Add null-forgiving operator to _entityMap in integration tests.
Update SqlServerContainerFixture to use new MsSqlBuilder
constructor with image name.

Author: russlank

src/XpressData.Test/Datalayer/DatabaseConnectionAsyncMockTests.cs

@@ -31,7 +31,7 @@ public async Task SelectAsync_WithValidQuery_ReturnsSelectQueryResult()
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.SelectAsync(query);
+        var result = await mockConnection.Object.SelectAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.NotNull(result);
@@ -84,7 +84,7 @@ public async Task SelectAsync_WithParameters_PassesParametersToConnection()
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.SelectAsync(query);
+        var result = await mockConnection.Object.SelectAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.NotNull(result);
@@ -116,7 +116,7 @@ public async Task SelectScalarAsync_WithValidQuery_ReturnsSelectScalarQueryResul
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.SelectScalarAsync(query);
+        var result = await mockConnection.Object.SelectScalarAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.NotNull(result);
@@ -146,7 +146,7 @@ public async Task InsertAsync_WithValidQuery_ReturnsInsertQueryResult()
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.InsertAsync(query);
+        var result = await mockConnection.Object.InsertAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.NotNull(result);
@@ -177,7 +177,7 @@ public async Task UpdateAsync_WithValidQuery_ReturnsUpdateQueryResult()
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.UpdateAsync(query);
+        var result = await mockConnection.Object.UpdateAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.NotNull(result);
@@ -202,7 +202,7 @@ public async Task UpdateAsync_WhenMultipleRowsAffected_ReturnsCorrectCount()
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.UpdateAsync(query);
+        var result = await mockConnection.Object.UpdateAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.Equal(5, result.AffectedRows);
@@ -229,7 +229,7 @@ public async Task ExecuteAsync_WithValidQuery_ReturnsExecuteQueryResult()
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.ExecuteAsync(query);
+        var result = await mockConnection.Object.ExecuteAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.NotNull(result);
@@ -253,7 +253,7 @@ public async Task ExecuteAsync_WithDDLStatement_ReturnsExecuteQueryResult()
             .ReturnsAsync(expectedResult);
 
         // Act
-        var result = await mockConnection.Object.ExecuteAsync(query);
+        var result = await mockConnection.Object.ExecuteAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.NotNull(result);
@@ -276,7 +276,7 @@ public async Task DeleteDatabaseAsync_WhenCalled_CompletesSuccessfully()
             .Returns(Task.CompletedTask);
 
         // Act
-        await mockConnection.Object.DeleteDatabaseAsync();
+        await mockConnection.Object.DeleteDatabaseAsync(cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         mockConnection.Verify(c => c.DeleteDatabaseAsync(It.IsAny<CancellationToken>()), Times.Once);
@@ -364,7 +364,7 @@ public async Task AsyncOperations_WithTransaction_ReturnsTransactionId()
 
         // Act
         mockConnection.Object.BeginTransaction();
-        var result = await mockConnection.Object.SelectAsync(query);
+        var result = await mockConnection.Object.SelectAsync(query, cancellationToken: TestContext.Current.CancellationToken);
 
         // Assert
         Assert.Equal(transactionId, result.TransactionId);

src/XpressData.Test/Integration/EntityMapIntegrationTests.cs

@@ -487,7 +487,7 @@ public async Task Get_WithEqualFilter_ReturnsMatchingEntities()
         await InsertTestEntityAsync("OtherName");
 
         // Act
-        var results = _entityMap.Get(_connection!, filterExpression: NameEquals("TargetName"));
+        var results = _entityMap!.Get(_connection!, filterExpression: NameEquals("TargetName"));
 
         // Assert
         Assert.Single(results);

src/XpressData.Test/Integration/SqlServerContainerFixture.cs

@@ -31,8 +31,7 @@ public class SqlServerContainerFixture : IAsyncLifetime
     /// </summary>
     public async ValueTask InitializeAsync()
     {
-        _container = new MsSqlBuilder()
-            .WithImage("mcr.microsoft.com/mssql/server:2022-latest")
+        _container = new MsSqlBuilder("mcr.microsoft.com/mssql/server:2022-latest")
             .WithPassword("Strong_Password_123!")
             .Build();