BulkMergeTests

Author: phongnguyend

src/EntityFrameworkCore.SqlServer.SimpleBulks.DbContextExtensionsTests/DbContextExtensions/BulkMergeAsyncTests.cs

@@ -105,7 +105,7 @@ public async Task BulkMerge_Using_Linq_With_Transaction(int length, int insertLe
             });
         }
 
-        await _context.BulkMergeAsync(rows,
+        var result1 = await _context.BulkMergeAsync(rows,
                 row => row.Id,
                 row => new { row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
                 row => new { row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
@@ -115,7 +115,7 @@ await _context.BulkMergeAsync(rows,
                     options.ReturnDbGeneratedId = true;
                 });
 
-        await _context.BulkMergeAsync(compositeKeyRows,
+        var result2 = await _context.BulkMergeAsync(compositeKeyRows,
                 row => new { row.Id1, row.Id2 },
                 row => new { row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
                 row => new { row.Id1, row.Id2, row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
@@ -130,6 +130,14 @@ await _context.BulkMergeAsync(compositeKeyRows,
         var dbRows = _context.SingleKeyRows.AsNoTracking().ToList();
         var dbCompositeKeyRows = _context.CompositeKeyRows.AsNoTracking().ToList();
 
+        Assert.Equal(length + insertLength, result1.AffectedRows);
+        Assert.Equal(insertLength, result1.InsertedRows);
+        Assert.Equal(length, result1.UpdatedRows);
+
+        Assert.Equal(length + insertLength, result2.AffectedRows);
+        Assert.Equal(insertLength, result2.InsertedRows);
+        Assert.Equal(length, result2.UpdatedRows);
+
         for (var i = 0; i < length + insertLength; i++)
         {
             Assert.Equal(rows[i].Id, dbRows[i].Id);
@@ -202,7 +210,7 @@ public async Task BulkMerge_Using_Dynamic_String_With_Transaction(int length, in
             });
         }
 
-        await _context.BulkMergeAsync(rows,
+        var result1 = await _context.BulkMergeAsync(rows,
             "Id",
             ["Column1", "Column2", "Column3", "Season", "SeasonAsString"],
             ["Column1", "Column2", "Column3", "Season", "SeasonAsString"],
@@ -211,7 +219,7 @@ await _context.BulkMergeAsync(rows,
                 options.LogTo = _output.WriteLine;
                 options.ReturnDbGeneratedId = true;
             });
-        await _context.BulkMergeAsync(compositeKeyRows,
+        var result2 = await _context.BulkMergeAsync(compositeKeyRows,
             ["Id1", "Id2"],
             ["Column1", "Column2", "Column3", "Season", "SeasonAsString"],
             ["Id1", "Id2", "Column1", "Column2", "Column3", "Season", "SeasonAsString"],
@@ -226,6 +234,14 @@ await _context.BulkMergeAsync(compositeKeyRows,
         var dbRows = _context.SingleKeyRows.AsNoTracking().ToList();
         var dbCompositeKeyRows = _context.CompositeKeyRows.AsNoTracking().ToList();
 
+        Assert.Equal(length + insertLength, result1.AffectedRows);
+        Assert.Equal(insertLength, result1.InsertedRows);
+        Assert.Equal(length, result1.UpdatedRows);
+
+        Assert.Equal(length + insertLength, result2.AffectedRows);
+        Assert.Equal(insertLength, result2.InsertedRows);
+        Assert.Equal(length, result2.UpdatedRows);
+
         for (var i = 0; i < length + insertLength; i++)
         {
             Assert.Equal(rows[i].Id, dbRows[i].Id);

src/EntityFrameworkCore.SqlServer.SimpleBulks.DbContextExtensionsTests/DbContextExtensions/BulkMergeTests.cs

@@ -105,7 +105,7 @@ public void BulkMerge_Using_Linq_With_Transaction(int length, int insertLength)
             });
         }
 
-        _context.BulkMerge(rows,
+        var result1 = _context.BulkMerge(rows,
                 row => row.Id,
                 row => new { row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
                 row => new { row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
@@ -115,7 +115,7 @@ public void BulkMerge_Using_Linq_With_Transaction(int length, int insertLength)
                     options.ReturnDbGeneratedId = true;
                 });
 
-        _context.BulkMerge(compositeKeyRows,
+        var result2 = _context.BulkMerge(compositeKeyRows,
                 row => new { row.Id1, row.Id2 },
                 row => new { row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
                 row => new { row.Id1, row.Id2, row.Column1, row.Column2, row.Column3, row.Season, row.SeasonAsString },
@@ -130,6 +130,14 @@ public void BulkMerge_Using_Linq_With_Transaction(int length, int insertLength)
         var dbRows = _context.SingleKeyRows.AsNoTracking().ToList();
         var dbCompositeKeyRows = _context.CompositeKeyRows.AsNoTracking().ToList();
 
+        Assert.Equal(length + insertLength, result1.AffectedRows);
+        Assert.Equal(insertLength, result1.InsertedRows);
+        Assert.Equal(length, result1.UpdatedRows);
+
+        Assert.Equal(length + insertLength, result2.AffectedRows);
+        Assert.Equal(insertLength, result2.InsertedRows);
+        Assert.Equal(length, result2.UpdatedRows);
+
         for (int i = 0; i < length + insertLength; i++)
         {
             Assert.Equal(rows[i].Id, dbRows[i].Id);
@@ -202,7 +210,7 @@ public void BulkMerge_Using_Dynamic_String_With_Transaction(int length, int inse
             });
         }
 
-        _context.BulkMerge(rows,
+        var result1 = _context.BulkMerge(rows,
             "Id",
             ["Column1", "Column2", "Column3", "Season", "SeasonAsString"],
             ["Column1", "Column2", "Column3", "Season", "SeasonAsString"],
@@ -211,7 +219,7 @@ public void BulkMerge_Using_Dynamic_String_With_Transaction(int length, int inse
                 options.LogTo = _output.WriteLine;
                 options.ReturnDbGeneratedId = true;
             });
-        _context.BulkMerge(compositeKeyRows,
+        var result2 = _context.BulkMerge(compositeKeyRows,
             ["Id1", "Id2"],
             ["Column1", "Column2", "Column3", "Season", "SeasonAsString"],
             ["Id1", "Id2", "Column1", "Column2", "Column3", "Season", "SeasonAsString"],
@@ -226,6 +234,14 @@ public void BulkMerge_Using_Dynamic_String_With_Transaction(int length, int inse
         var dbRows = _context.SingleKeyRows.AsNoTracking().ToList();
         var dbCompositeKeyRows = _context.CompositeKeyRows.AsNoTracking().ToList();
 
+        Assert.Equal(length + insertLength, result1.AffectedRows);
+        Assert.Equal(insertLength, result1.InsertedRows);
+        Assert.Equal(length, result1.UpdatedRows);
+
+        Assert.Equal(length + insertLength, result2.AffectedRows);
+        Assert.Equal(insertLength, result2.InsertedRows);
+        Assert.Equal(length, result2.UpdatedRows);
+
         for (int i = 0; i < length + insertLength; i++)
         {
             Assert.Equal(rows[i].Id, dbRows[i].Id);