From ccbf714ebeff51d1370789e5e487a978d0e2dbfb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?St=C3=A9phane=20=22Twidi=22=20Angel?= Date: Thu, 7 Jul 2022 04:26:49 +0200 Subject: Fixed #33829 -- Made BaseConstraint.deconstruct() and equality handle violation_error_message. Regression in 667105877e6723c6985399803a364848891513cc. --- tests/constraints/tests.py | 77 ++++++++++++++++++++++++++++++++ tests/postgres_tests/test_constraints.py | 22 +++++++++ 2 files changed, 99 insertions(+) (limited to 'tests') diff --git a/tests/constraints/tests.py b/tests/constraints/tests.py index d9e377438e..4032b418b4 100644 --- a/tests/constraints/tests.py +++ b/tests/constraints/tests.py @@ -65,6 +65,29 @@ class BaseConstraintTests(SimpleTestCase): ) self.assertEqual(c.get_violation_error_message(), "custom base_name message") + def test_custom_violation_error_message_clone(self): + constraint = BaseConstraint( + "base_name", + violation_error_message="custom %(name)s message", + ).clone() + self.assertEqual( + constraint.get_violation_error_message(), + "custom base_name message", + ) + + def test_deconstruction(self): + constraint = BaseConstraint( + "base_name", + violation_error_message="custom %(name)s message", + ) + path, args, kwargs = constraint.deconstruct() + self.assertEqual(path, "django.db.models.BaseConstraint") + self.assertEqual(args, ()) + self.assertEqual( + kwargs, + {"name": "base_name", "violation_error_message": "custom %(name)s message"}, + ) + class CheckConstraintTests(TestCase): def test_eq(self): @@ -84,6 +107,28 @@ class CheckConstraintTests(TestCase): models.CheckConstraint(check=check2, name="price"), ) self.assertNotEqual(models.CheckConstraint(check=check1, name="price"), 1) + self.assertNotEqual( + models.CheckConstraint(check=check1, name="price"), + models.CheckConstraint( + check=check1, name="price", violation_error_message="custom error" + ), + ) + self.assertNotEqual( + models.CheckConstraint( + check=check1, name="price", violation_error_message="custom error" + ), + models.CheckConstraint( + check=check1, name="price", violation_error_message="other custom error" + ), + ) + self.assertEqual( + models.CheckConstraint( + check=check1, name="price", violation_error_message="custom error" + ), + models.CheckConstraint( + check=check1, name="price", violation_error_message="custom error" + ), + ) def test_repr(self): constraint = models.CheckConstraint( @@ -216,6 +261,38 @@ class UniqueConstraintTests(TestCase): self.assertNotEqual( models.UniqueConstraint(fields=["foo", "bar"], name="unique"), 1 ) + self.assertNotEqual( + models.UniqueConstraint(fields=["foo", "bar"], name="unique"), + models.UniqueConstraint( + fields=["foo", "bar"], + name="unique", + violation_error_message="custom error", + ), + ) + self.assertNotEqual( + models.UniqueConstraint( + fields=["foo", "bar"], + name="unique", + violation_error_message="custom error", + ), + models.UniqueConstraint( + fields=["foo", "bar"], + name="unique", + violation_error_message="other custom error", + ), + ) + self.assertEqual( + models.UniqueConstraint( + fields=["foo", "bar"], + name="unique", + violation_error_message="custom error", + ), + models.UniqueConstraint( + fields=["foo", "bar"], + name="unique", + violation_error_message="custom error", + ), + ) def test_eq_with_condition(self): self.assertEqual( diff --git a/tests/postgres_tests/test_constraints.py b/tests/postgres_tests/test_constraints.py index d36c6fd9ed..a33c485a36 100644 --- a/tests/postgres_tests/test_constraints.py +++ b/tests/postgres_tests/test_constraints.py @@ -444,17 +444,39 @@ class ExclusionConstraintTests(PostgreSQLTestCase): ) self.assertNotEqual(constraint_2, constraint_9) self.assertNotEqual(constraint_7, constraint_8) + + constraint_10 = ExclusionConstraint( + name="exclude_overlapping", + expressions=[ + (F("datespan"), RangeOperators.OVERLAPS), + (F("room"), RangeOperators.EQUAL), + ], + condition=Q(cancelled=False), + violation_error_message="custom error", + ) + constraint_11 = ExclusionConstraint( + name="exclude_overlapping", + expressions=[ + (F("datespan"), RangeOperators.OVERLAPS), + (F("room"), RangeOperators.EQUAL), + ], + condition=Q(cancelled=False), + violation_error_message="other custom error", + ) self.assertEqual(constraint_1, constraint_1) self.assertEqual(constraint_1, mock.ANY) self.assertNotEqual(constraint_1, constraint_2) self.assertNotEqual(constraint_1, constraint_3) self.assertNotEqual(constraint_1, constraint_4) + self.assertNotEqual(constraint_1, constraint_10) self.assertNotEqual(constraint_2, constraint_3) self.assertNotEqual(constraint_2, constraint_4) self.assertNotEqual(constraint_2, constraint_7) self.assertNotEqual(constraint_4, constraint_5) self.assertNotEqual(constraint_5, constraint_6) self.assertNotEqual(constraint_1, object()) + self.assertNotEqual(constraint_10, constraint_11) + self.assertEqual(constraint_10, constraint_10) def test_deconstruct(self): constraint = ExclusionConstraint( -- cgit v1.2.1