Additional test for cascading changes

Author: koenbeuk

test/EntityFrameworkCore.Triggered.IntegrationTests/CascadingSoftDeletes/ApplicationDbContext.cs

@@ -0,0 +1,43 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes.Models;
+using EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes.Triggers;
+using Microsoft.EntityFrameworkCore;
+
+namespace EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes
+{
+    public class ApplicationDbContext : DbContext
+    {
+        readonly string _databaseName;
+
+        public ApplicationDbContext(string databaseName)
+        {
+            _databaseName = databaseName;
+        }
+
+        protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
+        {
+            optionsBuilder
+                .UseInMemoryDatabase(_databaseName)
+                .UseTriggers(triggerOptions => {
+                    triggerOptions.AddTrigger<SoftDelete>();
+                });
+        }
+
+        protected override void OnModelCreating(ModelBuilder modelBuilder)
+        {
+            modelBuilder.Entity<Branch>()
+                .HasOne(x => x.Parent)
+                .WithMany(x => x.Children)
+                .HasForeignKey(x => x.ParentId)
+                .OnDelete(DeleteBehavior.Cascade);
+        }
+
+        public DbSet<Branch> Branches => Set<Branch>();
+    }
+
+
+}

test/EntityFrameworkCore.Triggered.IntegrationTests/CascadingSoftDeletes/CascadingSoftDeletesTestScenario.cs

@@ -0,0 +1,57 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes.Models;
+using ScenarioTests;
+using Xunit;
+
+namespace EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes
+{
+    public partial class CascadingSoftDeletesTestScenario
+    {
+        [Scenario(NamingPolicy = ScenarioTestMethodNamingPolicy.Test)]
+        public void Scenario(ScenarioContext scenario)
+        {
+            var dbContext = new ApplicationDbContext(scenario.TargetName);
+
+            var level0 = new Branch();
+            var level1 = new Branch() { Parent = level0 };
+            var level2 = new Branch() { Parent = level1 };
+
+            dbContext.AddRange(level0, level1, level2);
+            {
+                var result = dbContext.SaveChanges();
+                Assert.Equal(3, result);
+            }
+
+            scenario.Fact("1. Soft delete works on the depest level", () => {
+                dbContext.Remove(level2);
+                var result = dbContext.SaveChanges();
+
+                Assert.Equal(1, result);
+                Assert.NotNull(level2.DeletedOn);
+            });
+
+            scenario.Fact("2. Soft delete cascades at least 1 level deep", () => {
+                dbContext.Remove(level1);
+                var result = dbContext.SaveChanges();
+
+                Assert.Equal(2, result);
+                Assert.NotNull(level1.DeletedOn);
+                Assert.NotNull(level2.DeletedOn);
+            });
+
+            scenario.Fact("3. Soft delete cascades multiple levels deep", () => {
+                dbContext.Remove(level0);
+                var result = dbContext.SaveChanges();
+
+                Assert.Equal(3, result);
+                Assert.NotNull(level0.DeletedOn);
+                Assert.NotNull(level1.DeletedOn);
+                Assert.NotNull(level2.DeletedOn);
+            });
+        }
+    }
+}

test/EntityFrameworkCore.Triggered.IntegrationTests/CascadingSoftDeletes/Models/Branch.cs

@@ -0,0 +1,20 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using Microsoft.VisualBasic;
+
+namespace EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes.Models
+{
+    public class Branch
+    {
+        public int Id { get; set; }
+        public int? ParentId { get; set; }
+        public DateTime? DeletedOn { get; set; }
+
+        public Branch Parent { get; set; }
+        public ICollection<Branch> Children { get; set; }
+        
+    }
+}

test/EntityFrameworkCore.Triggered.IntegrationTests/CascadingSoftDeletes/Triggers/SoftDelete.cs

@@ -0,0 +1,29 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+using EntityFrameworkCore.Triggered.Extensions;
+using EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes.Models;
+
+namespace EntityFrameworkCore.Triggered.IntegrationTests.CascadingSoftDeletes.Triggers
+{
+    public class SoftDelete : Trigger<Branch>
+    {
+        readonly ApplicationDbContext _applicationDbContext;
+
+        public SoftDelete(ApplicationDbContext applicationDbContext)
+        {
+            _applicationDbContext = applicationDbContext;
+        }
+
+        public override void BeforeSave(ITriggerContext<Branch> context)
+        {
+            if (context.ChangeType is ChangeType.Deleted)
+            {
+                _applicationDbContext.Entry(context.Entity).State = Microsoft.EntityFrameworkCore.EntityState.Modified;
+                context.Entity.DeletedOn = DateTime.UtcNow;
+            }
+        }
+    }
+}