summaryrefslogtreecommitdiff
path: root/tests/model_forms
diff options
context:
space:
mode:
Diffstat (limited to 'tests/model_forms')
-rw-r--r--tests/model_forms/models.py21
-rw-r--r--tests/model_forms/tests.py38
2 files changed, 59 insertions, 0 deletions
diff --git a/tests/model_forms/models.py b/tests/model_forms/models.py
index f9441a4c77..4111fafd69 100644
--- a/tests/model_forms/models.py
+++ b/tests/model_forms/models.py
@@ -521,3 +521,24 @@ class Dice(models.Model):
through=NumbersToDice,
limit_choices_to=models.Q(value__gte=1),
)
+
+
+class ConstraintsModel(models.Model):
+ name = models.CharField(max_length=100)
+ category = models.CharField(max_length=50, default="uncategorized")
+ price = models.DecimalField(max_digits=10, decimal_places=2, default=0)
+
+ class Meta:
+ constraints = [
+ models.UniqueConstraint(
+ "name",
+ "category",
+ name="unique_name_category",
+ violation_error_message="This product already exists.",
+ ),
+ models.CheckConstraint(
+ condition=models.Q(price__gt=0),
+ name="price_gte_zero",
+ violation_error_message="Price must be greater than zero.",
+ ),
+ ]
diff --git a/tests/model_forms/tests.py b/tests/model_forms/tests.py
index a432f6ce43..837466f07f 100644
--- a/tests/model_forms/tests.py
+++ b/tests/model_forms/tests.py
@@ -40,6 +40,7 @@ from .models import (
Character,
Colour,
ColourfulItem,
+ ConstraintsModel,
CustomErrorMessage,
CustomFF,
CustomFieldForExclusionModel,
@@ -3718,3 +3719,40 @@ class ModelToDictTests(TestCase):
# If data were a QuerySet, it would be reevaluated here and give "red"
# instead of the original value.
self.assertEqual(data, [blue])
+
+
+class ConstraintValidationTests(TestCase):
+ def test_unique_constraint_refs_excluded_field(self):
+ obj = ConstraintsModel.objects.create(name="product", price="1.00")
+ data = {
+ "id": "",
+ "name": obj.name,
+ "price": "1337.00",
+ "category": obj.category,
+ }
+ ConstraintsModelForm = modelform_factory(ConstraintsModel, fields="__all__")
+ ExcludeCategoryForm = modelform_factory(ConstraintsModel, exclude=["category"])
+ full_form = ConstraintsModelForm(data)
+ exclude_category_form = ExcludeCategoryForm(data)
+ self.assertTrue(exclude_category_form.is_valid())
+ self.assertFalse(full_form.is_valid())
+ self.assertEqual(
+ full_form.errors, {"__all__": ["This product already exists."]}
+ )
+
+ def test_check_constraint_refs_excluded_field(self):
+ data = {
+ "id": "",
+ "name": "priceless",
+ "price": "0.00",
+ "category": "category 1",
+ }
+ ConstraintsModelForm = modelform_factory(ConstraintsModel, fields="__all__")
+ ExcludePriceForm = modelform_factory(ConstraintsModel, exclude=["price"])
+ full_form = ConstraintsModelForm(data)
+ exclude_price_form = ExcludePriceForm(data)
+ self.assertTrue(exclude_price_form.is_valid())
+ self.assertFalse(full_form.is_valid())
+ self.assertEqual(
+ full_form.errors, {"__all__": ["Price must be greater than zero."]}
+ )