Make check constraints work when creating a table

Follow-up to #14673
dotnet · Aug 11, 2019 · 76787fb · 76787fb
1 parent b0ea397
commit 76787fb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/EFCore.Relational/Migrations/Internal/MigrationsModelDiffer.cs b/src/EFCore.Relational/Migrations/Internal/MigrationsModelDiffer.cs
@@ -621,6 +621,9 @@ protected virtual IEnumerable<MigrationOperation> Add(
             createTableOperation.UniqueConstraints.AddRange(
                 target.GetKeys().Where(k => !k.IsPrimaryKey()).SelectMany(k => Add(k, diffContext))
                     .Cast<AddUniqueConstraintOperation>());
+            createTableOperation.CheckConstraints.AddRange(
+                target.GetCheckConstraints().SelectMany(c => Add(c, diffContext))
+                    .Cast<CreateCheckConstraintOperation>());
 
             foreach (var targetEntityType in target.EntityTypes)
             {

diff --git a/test/EFCore.Relational.Tests/Migrations/Internal/MigrationsModelDifferTest.cs b/test/EFCore.Relational.Tests/Migrations/Internal/MigrationsModelDifferTest.cs
@@ -240,6 +240,7 @@ public void Create_table()
                         x.Property<int?>("ParentAltId");
                         x.HasOne("Node").WithMany().HasForeignKey("ParentAltId");
                         x.HasIndex("ParentAltId");
+                        x.HasCheckConstraint("SomeCheckConstraint", "[Id] > 10");
                     }),
                 upOps =>
                 {
@@ -255,7 +256,7 @@ public void Create_table()
                     Assert.Null(createTableOperation.Columns.First(o => o.Name == "AltId").DefaultValue);
                     Assert.NotNull(createTableOperation.PrimaryKey);
                     Assert.Equal(1, createTableOperation.UniqueConstraints.Count);
-                    Assert.Equal(0, createTableOperation.CheckConstraints.Count);
+                    Assert.Equal(1, createTableOperation.CheckConstraints.Count);
                     Assert.Equal(1, createTableOperation.ForeignKeys.Count);
 
                     Assert.IsType<CreateIndexOperation>(upOps[2]);