Little tweaks to the eject for dirty sessions

Author: jeremydmiller

src/Marten/Internal/Sessions/DocumentSessionBase.Deletes.cs

@@ -27,7 +27,8 @@ public void Delete<T>(T entity) where T : notnull
         _workTracker.Add(deletion);
 
         documentStorage.Eject(this, entity);
-        ChangeTrackers.RemoveAll(x => ReferenceEquals(entity, x.Document));
+
+        ChangeTrackers.RemoveAll(x => x.Document is T t && documentStorage.IdentityFor(entity).Equals(documentStorage.IdentityFor(t)));
     }
 
     public void Delete<T>(int id) where T : notnull

src/DocumentDbTests/SessionMechanics/reset_all_data_usage_ihost.cs renamed to src/StressTests/reset_all_data_usage_ihost.cs

@@ -1,5 +1,8 @@
+using System.Threading;
 using System.Threading.Tasks;
 using Marten;
+using Marten.Schema;
+using Marten.Testing.Documents;
 using Marten.Testing.Harness;
 using Microsoft.Extensions.Hosting;
 using Xunit;
@@ -35,7 +38,7 @@ public async Task can_reset_all_data_on_ihost()
                                 opts.Logger(new TestOutputMartenLogger(_output));
                             }
                         )
-                        .InitializeWith(new reset_all_data_usage.Users());
+                        .InitializeWith(new Users());
                 }
             )
             .StartAsync();
@@ -61,7 +64,7 @@ public async Task can_reset_all_data_on_ihost_for_specific_marten_database()
                                 opts.Logger(new TestOutputMartenLogger(_output));
                             }
                         )
-                        .InitializeWith(new reset_all_data_usage.Users());
+                        .InitializeWith(new Users());
                 }
             )
             .StartAsync();
@@ -71,3 +74,17 @@ public async Task can_reset_all_data_on_ihost_for_specific_marten_database()
         #endregion
     }
 }
+
+public class Users : IInitialData
+{
+    public Task Populate(IDocumentStore store, CancellationToken cancellation)
+    {
+        var users = new User[]
+        {
+            new User { UserName = "one" }, new User { UserName = "two" }, new User { UserName = "three" },
+            new User { UserName = "four" },
+        };
+
+        return store.BulkInsertDocumentsAsync(users, cancellation: cancellation);
+    }
+}