Add partial method to the scaffolded DbContext

fixes #12245
dotnet · May 8, 2019 · 7343fa1 · 7343fa1
1 parent cbefe76
commit 7343fa1
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 2 deletions.
diff --git a/src/EFCore.Design/Scaffolding/Internal/CSharpDbContextGenerator.cs b/src/EFCore.Design/Scaffolding/Internal/CSharpDbContextGenerator.cs
@@ -125,6 +125,13 @@ protected virtual void GenerateClass(
                 GenerateOnModelCreating(model, useDataAnnotations);
             }
 
+            _sb.AppendLine();
+
+            using (_sb.Indent())
+            {
+                _sb.AppendLine("partial void OnModelCreatingPartial(ModelBuilder modelBuilder);");
+            }
+
             _sb.AppendLine("}");
         }
 
@@ -319,6 +326,13 @@ protected virtual void GenerateOnModelCreating(
                 }
             }
 
+            _sb.AppendLine();
+
+            using (_sb.Indent())
+            {
+                _sb.AppendLine("OnModelCreatingPartial(modelBuilder);");
+            }
+
             _sb.AppendLine("}");
         }
 

diff --git a/test/EFCore.Design.Tests/Scaffolding/Internal/CSharpDbContextGeneratorTest.cs b/test/EFCore.Design.Tests/Scaffolding/Internal/CSharpDbContextGeneratorTest.cs
@@ -48,7 +48,11 @@ protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
         }
 
         protected override void OnModelCreating(ModelBuilder modelBuilder)
-        {}
+        {
+            OnModelCreatingPartial(modelBuilder);
+        }
+
+        partial void OnModelCreatingPartial(ModelBuilder modelBuilder);
     }
 }
 ",
@@ -98,7 +102,11 @@ protected override void OnConfiguring(DbContextOptionsBuilder optionsBuilder)
         }
 
         protected override void OnModelCreating(ModelBuilder modelBuilder)
-        {}
+        {
+            OnModelCreatingPartial(modelBuilder);
+        }
+
+        partial void OnModelCreatingPartial(ModelBuilder modelBuilder);
     }
 }
 ",