From 19e838daa8872ee29fbea0bc471c2a6443f26835 Mon Sep 17 00:00:00 2001 From: James Beith Date: Wed, 7 Sep 2022 13:53:37 +1000 Subject: Fixed #33982 -- Fixed migrations crash when adding model with ExclusionConstraint. Regression in 0e656c02fe945389246f0c08f51c6db4a0849bd2. --- tests/postgres_tests/test_constraints.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) (limited to 'tests/postgres_tests') diff --git a/tests/postgres_tests/test_constraints.py b/tests/postgres_tests/test_constraints.py index a6f7b5be85..2b6df7d5f5 100644 --- a/tests/postgres_tests/test_constraints.py +++ b/tests/postgres_tests/test_constraints.py @@ -10,12 +10,14 @@ from django.db.models import ( F, Func, IntegerField, + Model, Q, UniqueConstraint, ) from django.db.models.fields.json import KeyTextTransform from django.db.models.functions import Cast, Left, Lower from django.test import ignore_warnings, modify_settings, skipUnlessDBFeature +from django.test.utils import isolate_apps from django.utils import timezone from django.utils.deprecation import RemovedInDjango50Warning @@ -1151,6 +1153,29 @@ class ExclusionConstraintTests(PostgreSQLTestCase): editor.add_constraint(Room, constraint) self.assertIn(constraint_name, self.get_constraints(Room._meta.db_table)) + @isolate_apps("postgres_tests") + def test_table_create(self): + constraint_name = "exclusion_equal_number_tc" + + class ModelWithExclusionConstraint(Model): + number = IntegerField() + + class Meta: + app_label = "postgres_tests" + constraints = [ + ExclusionConstraint( + name=constraint_name, + expressions=[("number", RangeOperators.EQUAL)], + ) + ] + + with connection.schema_editor() as editor: + editor.create_model(ModelWithExclusionConstraint) + self.assertIn( + constraint_name, + self.get_constraints(ModelWithExclusionConstraint._meta.db_table), + ) + @modify_settings(INSTALLED_APPS={"append": "django.contrib.postgres"}) class ExclusionConstraintOpclassesDepracationTests(PostgreSQLTestCase): -- cgit v1.3