12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460 |
- from unittest import mock
- from django.core.exceptions import ValidationError
- from django.db import IntegrityError, connection, models
- from django.db.models import F
- from django.db.models.constraints import BaseConstraint, UniqueConstraint
- from django.db.models.functions import Abs, Lower, Sqrt, Upper
- from django.db.transaction import atomic
- from django.test import SimpleTestCase, TestCase, skipIfDBFeature, skipUnlessDBFeature
- from django.test.utils import ignore_warnings
- from django.utils.deprecation import RemovedInDjango60Warning
- from .models import (
- ChildModel,
- ChildUniqueConstraintProduct,
- GeneratedFieldStoredProduct,
- GeneratedFieldVirtualProduct,
- JSONFieldModel,
- ModelWithDatabaseDefault,
- Product,
- UniqueConstraintConditionProduct,
- UniqueConstraintDeferrable,
- UniqueConstraintInclude,
- UniqueConstraintProduct,
- )
- def get_constraints(table):
- with connection.cursor() as cursor:
- return connection.introspection.get_constraints(cursor, table)
- class BaseConstraintTests(SimpleTestCase):
- def test_constraint_sql(self):
- c = BaseConstraint(name="name")
- msg = "This method must be implemented by a subclass."
- with self.assertRaisesMessage(NotImplementedError, msg):
- c.constraint_sql(None, None)
- def test_contains_expressions(self):
- c = BaseConstraint(name="name")
- self.assertIs(c.contains_expressions, False)
- def test_create_sql(self):
- c = BaseConstraint(name="name")
- msg = "This method must be implemented by a subclass."
- with self.assertRaisesMessage(NotImplementedError, msg):
- c.create_sql(None, None)
- def test_remove_sql(self):
- c = BaseConstraint(name="name")
- msg = "This method must be implemented by a subclass."
- with self.assertRaisesMessage(NotImplementedError, msg):
- c.remove_sql(None, None)
- def test_validate(self):
- c = BaseConstraint(name="name")
- msg = "This method must be implemented by a subclass."
- with self.assertRaisesMessage(NotImplementedError, msg):
- c.validate(None, None)
- def test_default_violation_error_message(self):
- c = BaseConstraint(name="name")
- self.assertEqual(
- c.get_violation_error_message(), "Constraint “name” is violated."
- )
- def test_custom_violation_error_message(self):
- c = BaseConstraint(
- name="base_name", violation_error_message="custom %(name)s message"
- )
- self.assertEqual(c.get_violation_error_message(), "custom base_name message")
- def test_custom_violation_error_message_clone(self):
- constraint = BaseConstraint(
- name="base_name",
- violation_error_message="custom %(name)s message",
- ).clone()
- self.assertEqual(
- constraint.get_violation_error_message(),
- "custom base_name message",
- )
- def test_custom_violation_code_message(self):
- c = BaseConstraint(name="base_name", violation_error_code="custom_code")
- self.assertEqual(c.violation_error_code, "custom_code")
- def test_deconstruction(self):
- constraint = BaseConstraint(
- name="base_name",
- violation_error_message="custom %(name)s message",
- violation_error_code="custom_code",
- )
- 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",
- "violation_error_code": "custom_code",
- },
- )
- def test_name_required(self):
- msg = (
- "BaseConstraint.__init__() missing 1 required keyword-only argument: 'name'"
- )
- with self.assertRaisesMessage(TypeError, msg):
- BaseConstraint()
- class CheckConstraintTests(TestCase):
- def test_eq(self):
- check1 = models.Q(price__gt=models.F("discounted_price"))
- check2 = models.Q(price__lt=models.F("discounted_price"))
- self.assertEqual(
- models.CheckConstraint(condition=check1, name="price"),
- models.CheckConstraint(condition=check1, name="price"),
- )
- self.assertEqual(
- models.CheckConstraint(condition=check1, name="price"), mock.ANY
- )
- self.assertNotEqual(
- models.CheckConstraint(condition=check1, name="price"),
- models.CheckConstraint(condition=check1, name="price2"),
- )
- self.assertNotEqual(
- models.CheckConstraint(condition=check1, name="price"),
- models.CheckConstraint(condition=check2, name="price"),
- )
- self.assertNotEqual(models.CheckConstraint(condition=check1, name="price"), 1)
- self.assertNotEqual(
- models.CheckConstraint(condition=check1, name="price"),
- models.CheckConstraint(
- condition=check1, name="price", violation_error_message="custom error"
- ),
- )
- self.assertNotEqual(
- models.CheckConstraint(
- condition=check1, name="price", violation_error_message="custom error"
- ),
- models.CheckConstraint(
- condition=check1,
- name="price",
- violation_error_message="other custom error",
- ),
- )
- self.assertEqual(
- models.CheckConstraint(
- condition=check1, name="price", violation_error_message="custom error"
- ),
- models.CheckConstraint(
- condition=check1, name="price", violation_error_message="custom error"
- ),
- )
- self.assertNotEqual(
- models.CheckConstraint(condition=check1, name="price"),
- models.CheckConstraint(
- condition=check1, name="price", violation_error_code="custom_code"
- ),
- )
- self.assertEqual(
- models.CheckConstraint(
- condition=check1, name="price", violation_error_code="custom_code"
- ),
- models.CheckConstraint(
- condition=check1, name="price", violation_error_code="custom_code"
- ),
- )
- def test_repr(self):
- constraint = models.CheckConstraint(
- condition=models.Q(price__gt=models.F("discounted_price")),
- name="price_gt_discounted_price",
- )
- self.assertEqual(
- repr(constraint),
- "<CheckConstraint: condition=(AND: ('price__gt', F(discounted_price))) "
- "name='price_gt_discounted_price'>",
- )
- def test_repr_with_violation_error_message(self):
- constraint = models.CheckConstraint(
- condition=models.Q(price__lt=1),
- name="price_lt_one",
- violation_error_message="More than 1",
- )
- self.assertEqual(
- repr(constraint),
- "<CheckConstraint: condition=(AND: ('price__lt', 1)) name='price_lt_one' "
- "violation_error_message='More than 1'>",
- )
- def test_repr_with_violation_error_code(self):
- constraint = models.CheckConstraint(
- condition=models.Q(price__lt=1),
- name="price_lt_one",
- violation_error_code="more_than_one",
- )
- self.assertEqual(
- repr(constraint),
- "<CheckConstraint: condition=(AND: ('price__lt', 1)) name='price_lt_one' "
- "violation_error_code='more_than_one'>",
- )
- def test_invalid_check_types(self):
- msg = "CheckConstraint.condition must be a Q instance or boolean expression."
- with self.assertRaisesMessage(TypeError, msg):
- models.CheckConstraint(condition=models.F("discounted_price"), name="check")
- def test_deconstruction(self):
- check = models.Q(price__gt=models.F("discounted_price"))
- name = "price_gt_discounted_price"
- constraint = models.CheckConstraint(condition=check, name=name)
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.CheckConstraint")
- self.assertEqual(args, ())
- self.assertEqual(kwargs, {"condition": check, "name": name})
- @skipUnlessDBFeature("supports_table_check_constraints")
- def test_database_constraint(self):
- Product.objects.create(price=10, discounted_price=5)
- with self.assertRaises(IntegrityError):
- Product.objects.create(price=10, discounted_price=20)
- @skipUnlessDBFeature("supports_table_check_constraints")
- def test_database_constraint_unicode(self):
- Product.objects.create(price=10, discounted_price=5, unit="μg/mL")
- with self.assertRaises(IntegrityError):
- Product.objects.create(price=10, discounted_price=7, unit="l")
- @skipUnlessDBFeature(
- "supports_table_check_constraints", "can_introspect_check_constraints"
- )
- def test_name(self):
- constraints = get_constraints(Product._meta.db_table)
- for expected_name in (
- "price_gt_discounted_price",
- "constraints_product_price_gt_0",
- ):
- with self.subTest(expected_name):
- self.assertIn(expected_name, constraints)
- @skipUnlessDBFeature(
- "supports_table_check_constraints", "can_introspect_check_constraints"
- )
- def test_abstract_name(self):
- constraints = get_constraints(ChildModel._meta.db_table)
- self.assertIn("constraints_childmodel_adult", constraints)
- def test_validate(self):
- check = models.Q(price__gt=models.F("discounted_price"))
- constraint = models.CheckConstraint(condition=check, name="price")
- # Invalid product.
- invalid_product = Product(price=10, discounted_price=42)
- with self.assertRaises(ValidationError):
- constraint.validate(Product, invalid_product)
- with self.assertRaises(ValidationError):
- constraint.validate(Product, invalid_product, exclude={"unit"})
- # Fields used by the check constraint are excluded.
- constraint.validate(Product, invalid_product, exclude={"price"})
- constraint.validate(Product, invalid_product, exclude={"discounted_price"})
- constraint.validate(
- Product,
- invalid_product,
- exclude={"discounted_price", "price"},
- )
- # Valid product.
- constraint.validate(Product, Product(price=10, discounted_price=5))
- def test_validate_custom_error(self):
- check = models.Q(price__gt=models.F("discounted_price"))
- constraint = models.CheckConstraint(
- condition=check,
- name="price",
- violation_error_message="discount is fake",
- violation_error_code="fake_discount",
- )
- # Invalid product.
- invalid_product = Product(price=10, discounted_price=42)
- msg = "discount is fake"
- with self.assertRaisesMessage(ValidationError, msg) as cm:
- constraint.validate(Product, invalid_product)
- self.assertEqual(cm.exception.code, "fake_discount")
- def test_validate_boolean_expressions(self):
- constraint = models.CheckConstraint(
- condition=models.expressions.ExpressionWrapper(
- models.Q(price__gt=500) | models.Q(price__lt=500),
- output_field=models.BooleanField(),
- ),
- name="price_neq_500_wrap",
- )
- msg = f"Constraint “{constraint.name}” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(Product, Product(price=500, discounted_price=5))
- constraint.validate(Product, Product(price=501, discounted_price=5))
- constraint.validate(Product, Product(price=499, discounted_price=5))
- def test_validate_rawsql_expressions_noop(self):
- constraint = models.CheckConstraint(
- condition=models.expressions.RawSQL(
- "price < %s OR price > %s",
- (500, 500),
- output_field=models.BooleanField(),
- ),
- name="price_neq_500_raw",
- )
- # RawSQL can not be checked and is always considered valid.
- constraint.validate(Product, Product(price=500, discounted_price=5))
- constraint.validate(Product, Product(price=501, discounted_price=5))
- constraint.validate(Product, Product(price=499, discounted_price=5))
- @skipUnlessDBFeature("supports_comparing_boolean_expr")
- def test_validate_nullable_field_with_none(self):
- # Nullable fields should be considered valid on None values.
- constraint = models.CheckConstraint(
- condition=models.Q(price__gte=0),
- name="positive_price",
- )
- constraint.validate(Product, Product())
- @skipIfDBFeature("supports_comparing_boolean_expr")
- def test_validate_nullable_field_with_isnull(self):
- constraint = models.CheckConstraint(
- condition=models.Q(price__gte=0) | models.Q(price__isnull=True),
- name="positive_price",
- )
- constraint.validate(Product, Product())
- @skipUnlessDBFeature("supports_json_field")
- def test_validate_nullable_jsonfield(self):
- is_null_constraint = models.CheckConstraint(
- condition=models.Q(data__isnull=True),
- name="nullable_data",
- )
- is_not_null_constraint = models.CheckConstraint(
- condition=models.Q(data__isnull=False),
- name="nullable_data",
- )
- is_null_constraint.validate(JSONFieldModel, JSONFieldModel(data=None))
- msg = f"Constraint “{is_null_constraint.name}” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- is_null_constraint.validate(JSONFieldModel, JSONFieldModel(data={}))
- msg = f"Constraint “{is_not_null_constraint.name}” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- is_not_null_constraint.validate(JSONFieldModel, JSONFieldModel(data=None))
- is_not_null_constraint.validate(JSONFieldModel, JSONFieldModel(data={}))
- def test_validate_pk_field(self):
- constraint_with_pk = models.CheckConstraint(
- condition=~models.Q(pk=models.F("age")),
- name="pk_not_age_check",
- )
- constraint_with_pk.validate(ChildModel, ChildModel(pk=1, age=2))
- msg = f"Constraint “{constraint_with_pk.name}” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint_with_pk.validate(ChildModel, ChildModel(pk=1, age=1))
- with self.assertRaisesMessage(ValidationError, msg):
- constraint_with_pk.validate(ChildModel, ChildModel(id=1, age=1))
- constraint_with_pk.validate(ChildModel, ChildModel(pk=1, age=1), exclude={"pk"})
- @skipUnlessDBFeature("supports_json_field")
- def test_validate_jsonfield_exact(self):
- data = {"release": "5.0.2", "version": "stable"}
- json_exact_constraint = models.CheckConstraint(
- condition=models.Q(data__version="stable"),
- name="only_stable_version",
- )
- json_exact_constraint.validate(JSONFieldModel, JSONFieldModel(data=data))
- data = {"release": "5.0.2", "version": "not stable"}
- msg = f"Constraint “{json_exact_constraint.name}” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- json_exact_constraint.validate(JSONFieldModel, JSONFieldModel(data=data))
- @skipUnlessDBFeature("supports_stored_generated_columns")
- def test_validate_generated_field_stored(self):
- self.assertGeneratedFieldIsValidated(model=GeneratedFieldStoredProduct)
- @skipUnlessDBFeature("supports_virtual_generated_columns")
- def test_validate_generated_field_virtual(self):
- self.assertGeneratedFieldIsValidated(model=GeneratedFieldVirtualProduct)
- def assertGeneratedFieldIsValidated(self, model):
- constraint = models.CheckConstraint(
- condition=models.Q(rebate__range=(0, 100)), name="bounded_rebate"
- )
- constraint.validate(model, model(price=50, discounted_price=20))
- invalid_product = model(price=1200, discounted_price=500)
- msg = f"Constraint “{constraint.name}” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(model, invalid_product)
- # Excluding referenced or generated fields should skip validation.
- constraint.validate(model, invalid_product, exclude={"price"})
- constraint.validate(model, invalid_product, exclude={"rebate"})
- def test_check_deprecation(self):
- msg = "CheckConstraint.check is deprecated in favor of `.condition`."
- condition = models.Q(foo="bar")
- with self.assertWarnsMessage(RemovedInDjango60Warning, msg) as ctx:
- constraint = models.CheckConstraint(name="constraint", check=condition)
- self.assertEqual(ctx.filename, __file__)
- with self.assertWarnsMessage(RemovedInDjango60Warning, msg) as ctx:
- self.assertIs(constraint.check, condition)
- self.assertEqual(ctx.filename, __file__)
- other_condition = models.Q(something="else")
- with self.assertWarnsMessage(RemovedInDjango60Warning, msg) as ctx:
- constraint.check = other_condition
- self.assertEqual(ctx.filename, __file__)
- with self.assertWarnsMessage(RemovedInDjango60Warning, msg) as ctx:
- self.assertIs(constraint.check, other_condition)
- self.assertEqual(ctx.filename, __file__)
- def test_database_default(self):
- models.CheckConstraint(
- condition=models.Q(field_with_db_default="field_with_db_default"),
- name="check_field_with_db_default",
- ).validate(ModelWithDatabaseDefault, ModelWithDatabaseDefault())
- # Ensure that a check also does not silently pass with either
- # FieldError or DatabaseError when checking with a db_default.
- with self.assertRaises(ValidationError):
- models.CheckConstraint(
- condition=models.Q(
- field_with_db_default="field_with_db_default", field="field"
- ),
- name="check_field_with_db_default_2",
- ).validate(
- ModelWithDatabaseDefault, ModelWithDatabaseDefault(field="not-field")
- )
- with self.assertRaises(ValidationError):
- models.CheckConstraint(
- condition=models.Q(field_with_db_default="field_with_db_default"),
- name="check_field_with_db_default",
- ).validate(
- ModelWithDatabaseDefault,
- ModelWithDatabaseDefault(field_with_db_default="other value"),
- )
- class UniqueConstraintTests(TestCase):
- @classmethod
- def setUpTestData(cls):
- cls.p1 = UniqueConstraintProduct.objects.create(name="p1", color="red")
- cls.p2 = UniqueConstraintProduct.objects.create(name="p2")
- def test_eq(self):
- self.assertEqual(
- models.UniqueConstraint(fields=["foo", "bar"], name="unique"),
- models.UniqueConstraint(fields=["foo", "bar"], name="unique"),
- )
- self.assertEqual(
- models.UniqueConstraint(fields=["foo", "bar"], name="unique"),
- mock.ANY,
- )
- self.assertNotEqual(
- models.UniqueConstraint(fields=["foo", "bar"], name="unique"),
- models.UniqueConstraint(fields=["foo", "bar"], name="unique2"),
- )
- self.assertNotEqual(
- models.UniqueConstraint(fields=["foo", "bar"], name="unique"),
- models.UniqueConstraint(fields=["foo", "baz"], name="unique"),
- )
- 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",
- ),
- )
- self.assertNotEqual(
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- violation_error_code="custom_error",
- ),
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- violation_error_code="other_custom_error",
- ),
- )
- self.assertEqual(
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- violation_error_code="custom_error",
- ),
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- violation_error_code="custom_error",
- ),
- )
- def test_eq_with_condition(self):
- self.assertEqual(
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- condition=models.Q(foo=models.F("bar")),
- ),
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- condition=models.Q(foo=models.F("bar")),
- ),
- )
- self.assertNotEqual(
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- condition=models.Q(foo=models.F("bar")),
- ),
- models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- condition=models.Q(foo=models.F("baz")),
- ),
- )
- def test_eq_with_deferrable(self):
- constraint_1 = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- deferrable=models.Deferrable.DEFERRED,
- )
- constraint_2 = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique",
- deferrable=models.Deferrable.IMMEDIATE,
- )
- self.assertEqual(constraint_1, constraint_1)
- self.assertNotEqual(constraint_1, constraint_2)
- def test_eq_with_include(self):
- constraint_1 = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="include",
- include=["baz_1"],
- )
- constraint_2 = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="include",
- include=["baz_2"],
- )
- self.assertEqual(constraint_1, constraint_1)
- self.assertNotEqual(constraint_1, constraint_2)
- def test_eq_with_opclasses(self):
- constraint_1 = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="opclasses",
- opclasses=["text_pattern_ops", "varchar_pattern_ops"],
- )
- constraint_2 = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="opclasses",
- opclasses=["varchar_pattern_ops", "text_pattern_ops"],
- )
- self.assertEqual(constraint_1, constraint_1)
- self.assertNotEqual(constraint_1, constraint_2)
- def test_eq_with_expressions(self):
- constraint = models.UniqueConstraint(
- Lower("title"),
- F("author"),
- name="book_func_uq",
- )
- same_constraint = models.UniqueConstraint(
- Lower("title"),
- "author",
- name="book_func_uq",
- )
- another_constraint = models.UniqueConstraint(
- Lower("title"),
- name="book_func_uq",
- )
- self.assertEqual(constraint, same_constraint)
- self.assertEqual(constraint, mock.ANY)
- self.assertNotEqual(constraint, another_constraint)
- def test_eq_with_nulls_distinct(self):
- constraint_1 = models.UniqueConstraint(
- Lower("title"),
- nulls_distinct=False,
- name="book_func_nulls_distinct_uq",
- )
- constraint_2 = models.UniqueConstraint(
- Lower("title"),
- nulls_distinct=True,
- name="book_func_nulls_distinct_uq",
- )
- constraint_3 = models.UniqueConstraint(
- Lower("title"),
- name="book_func_nulls_distinct_uq",
- )
- 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_2, constraint_3)
- def test_repr(self):
- fields = ["foo", "bar"]
- name = "unique_fields"
- constraint = models.UniqueConstraint(fields=fields, name=name)
- self.assertEqual(
- repr(constraint),
- "<UniqueConstraint: fields=('foo', 'bar') name='unique_fields'>",
- )
- def test_repr_with_condition(self):
- constraint = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique_fields",
- condition=models.Q(foo=models.F("bar")),
- )
- self.assertEqual(
- repr(constraint),
- "<UniqueConstraint: fields=('foo', 'bar') name='unique_fields' "
- "condition=(AND: ('foo', F(bar)))>",
- )
- def test_repr_with_deferrable(self):
- constraint = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="unique_fields",
- deferrable=models.Deferrable.IMMEDIATE,
- )
- self.assertEqual(
- repr(constraint),
- "<UniqueConstraint: fields=('foo', 'bar') name='unique_fields' "
- "deferrable=Deferrable.IMMEDIATE>",
- )
- def test_repr_with_include(self):
- constraint = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="include_fields",
- include=["baz_1", "baz_2"],
- )
- self.assertEqual(
- repr(constraint),
- "<UniqueConstraint: fields=('foo', 'bar') name='include_fields' "
- "include=('baz_1', 'baz_2')>",
- )
- def test_repr_with_opclasses(self):
- constraint = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="opclasses_fields",
- opclasses=["text_pattern_ops", "varchar_pattern_ops"],
- )
- self.assertEqual(
- repr(constraint),
- "<UniqueConstraint: fields=('foo', 'bar') name='opclasses_fields' "
- "opclasses=['text_pattern_ops', 'varchar_pattern_ops']>",
- )
- def test_repr_with_nulls_distinct(self):
- constraint = models.UniqueConstraint(
- fields=["foo", "bar"],
- name="nulls_distinct_fields",
- nulls_distinct=False,
- )
- self.assertEqual(
- repr(constraint),
- "<UniqueConstraint: fields=('foo', 'bar') name='nulls_distinct_fields' "
- "nulls_distinct=False>",
- )
- def test_repr_with_expressions(self):
- constraint = models.UniqueConstraint(
- Lower("title"),
- F("author"),
- name="book_func_uq",
- )
- self.assertEqual(
- repr(constraint),
- "<UniqueConstraint: expressions=(Lower(F(title)), F(author)) "
- "name='book_func_uq'>",
- )
- def test_repr_with_violation_error_message(self):
- constraint = models.UniqueConstraint(
- models.F("baz__lower"),
- name="unique_lower_baz",
- violation_error_message="BAZ",
- )
- self.assertEqual(
- repr(constraint),
- (
- "<UniqueConstraint: expressions=(F(baz__lower),) "
- "name='unique_lower_baz' violation_error_message='BAZ'>"
- ),
- )
- def test_repr_with_violation_error_code(self):
- constraint = models.UniqueConstraint(
- models.F("baz__lower"),
- name="unique_lower_baz",
- violation_error_code="baz",
- )
- self.assertEqual(
- repr(constraint),
- (
- "<UniqueConstraint: expressions=(F(baz__lower),) "
- "name='unique_lower_baz' violation_error_code='baz'>"
- ),
- )
- def test_deconstruction(self):
- fields = ["foo", "bar"]
- name = "unique_fields"
- constraint = models.UniqueConstraint(fields=fields, name=name)
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.UniqueConstraint")
- self.assertEqual(args, ())
- self.assertEqual(kwargs, {"fields": tuple(fields), "name": name})
- def test_deconstruction_with_condition(self):
- fields = ["foo", "bar"]
- name = "unique_fields"
- condition = models.Q(foo=models.F("bar"))
- constraint = models.UniqueConstraint(
- fields=fields, name=name, condition=condition
- )
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.UniqueConstraint")
- self.assertEqual(args, ())
- self.assertEqual(
- kwargs, {"fields": tuple(fields), "name": name, "condition": condition}
- )
- def test_deconstruction_with_deferrable(self):
- fields = ["foo"]
- name = "unique_fields"
- constraint = models.UniqueConstraint(
- fields=fields,
- name=name,
- deferrable=models.Deferrable.DEFERRED,
- )
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.UniqueConstraint")
- self.assertEqual(args, ())
- self.assertEqual(
- kwargs,
- {
- "fields": tuple(fields),
- "name": name,
- "deferrable": models.Deferrable.DEFERRED,
- },
- )
- def test_deconstruction_with_include(self):
- fields = ["foo", "bar"]
- name = "unique_fields"
- include = ["baz_1", "baz_2"]
- constraint = models.UniqueConstraint(fields=fields, name=name, include=include)
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.UniqueConstraint")
- self.assertEqual(args, ())
- self.assertEqual(
- kwargs,
- {
- "fields": tuple(fields),
- "name": name,
- "include": tuple(include),
- },
- )
- def test_deconstruction_with_opclasses(self):
- fields = ["foo", "bar"]
- name = "unique_fields"
- opclasses = ["varchar_pattern_ops", "text_pattern_ops"]
- constraint = models.UniqueConstraint(
- fields=fields, name=name, opclasses=opclasses
- )
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.UniqueConstraint")
- self.assertEqual(args, ())
- self.assertEqual(
- kwargs,
- {
- "fields": tuple(fields),
- "name": name,
- "opclasses": opclasses,
- },
- )
- def test_deconstruction_with_nulls_distinct(self):
- fields = ["foo", "bar"]
- name = "unique_fields"
- constraint = models.UniqueConstraint(
- fields=fields, name=name, nulls_distinct=True
- )
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.UniqueConstraint")
- self.assertEqual(args, ())
- self.assertEqual(
- kwargs,
- {
- "fields": tuple(fields),
- "name": name,
- "nulls_distinct": True,
- },
- )
- def test_deconstruction_with_expressions(self):
- name = "unique_fields"
- constraint = models.UniqueConstraint(Lower("title"), name=name)
- path, args, kwargs = constraint.deconstruct()
- self.assertEqual(path, "django.db.models.UniqueConstraint")
- self.assertEqual(args, (Lower("title"),))
- self.assertEqual(kwargs, {"name": name})
- def test_database_constraint(self):
- with self.assertRaises(IntegrityError):
- UniqueConstraintProduct.objects.create(
- name=self.p1.name, color=self.p1.color
- )
- @skipUnlessDBFeature("supports_partial_indexes")
- def test_database_constraint_with_condition(self):
- UniqueConstraintConditionProduct.objects.create(name="p1")
- UniqueConstraintConditionProduct.objects.create(name="p2")
- with self.assertRaises(IntegrityError):
- UniqueConstraintConditionProduct.objects.create(name="p1")
- def test_model_validation(self):
- msg = "Unique constraint product with this Name and Color already exists."
- with self.assertRaisesMessage(ValidationError, msg):
- UniqueConstraintProduct(
- name=self.p1.name, color=self.p1.color
- ).validate_constraints()
- @skipUnlessDBFeature("supports_partial_indexes")
- def test_model_validation_with_condition(self):
- """
- Partial unique constraints are not ignored by
- Model.validate_constraints().
- """
- obj1 = UniqueConstraintConditionProduct.objects.create(name="p1", color="red")
- obj2 = UniqueConstraintConditionProduct.objects.create(name="p2")
- UniqueConstraintConditionProduct(
- name=obj1.name, color="blue"
- ).validate_constraints()
- msg = "Constraint “name_without_color_uniq” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- UniqueConstraintConditionProduct(name=obj2.name).validate_constraints()
- def test_model_validation_constraint_no_code_error(self):
- class ValidateNoCodeErrorConstraint(UniqueConstraint):
- def validate(self, model, instance, **kwargs):
- raise ValidationError({"name": ValidationError("Already exists.")})
- class NoCodeErrorConstraintModel(models.Model):
- name = models.CharField(max_length=255)
- class Meta:
- constraints = [
- ValidateNoCodeErrorConstraint(
- Lower("name"),
- name="custom_validate_no_code_error",
- )
- ]
- msg = "{'name': ['Already exists.']}"
- with self.assertRaisesMessage(ValidationError, msg):
- NoCodeErrorConstraintModel(name="test").validate_constraints()
- def test_validate(self):
- constraint = UniqueConstraintProduct._meta.constraints[0]
- msg = "Unique constraint product with this Name and Color already exists."
- non_unique_product = UniqueConstraintProduct(
- name=self.p1.name, color=self.p1.color
- )
- with self.assertRaisesMessage(ValidationError, msg) as cm:
- constraint.validate(UniqueConstraintProduct, non_unique_product)
- self.assertEqual(cm.exception.code, "unique_together")
- # Null values are ignored.
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p2.name, color=None),
- )
- # Existing instances have their existing row excluded.
- constraint.validate(UniqueConstraintProduct, self.p1)
- # Unique fields are excluded.
- constraint.validate(
- UniqueConstraintProduct,
- non_unique_product,
- exclude={"name"},
- )
- constraint.validate(
- UniqueConstraintProduct,
- non_unique_product,
- exclude={"color"},
- )
- constraint.validate(
- UniqueConstraintProduct,
- non_unique_product,
- exclude={"name", "color"},
- )
- # Validation on a child instance.
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(
- UniqueConstraintProduct,
- ChildUniqueConstraintProduct(name=self.p1.name, color=self.p1.color),
- )
- def test_validate_unique_custom_code_and_message(self):
- product = UniqueConstraintProduct.objects.create(
- name="test", color="red", age=42
- )
- code = "custom_code"
- message = "Custom message"
- multiple_fields_constraint = models.UniqueConstraint(
- fields=["color", "age"],
- name="color_age_uniq",
- violation_error_code=code,
- violation_error_message=message,
- )
- single_field_constraint = models.UniqueConstraint(
- fields=["color"],
- name="color_uniq",
- violation_error_code=code,
- violation_error_message=message,
- )
- with self.assertRaisesMessage(ValidationError, message) as cm:
- multiple_fields_constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(
- name="new-test", color=product.color, age=product.age
- ),
- )
- self.assertEqual(cm.exception.code, code)
- with self.assertRaisesMessage(ValidationError, message) as cm:
- single_field_constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name="new-test", color=product.color),
- )
- self.assertEqual(cm.exception.code, code)
- @skipUnlessDBFeature("supports_table_check_constraints")
- def test_validate_fields_unattached(self):
- Product.objects.create(price=42)
- constraint = models.UniqueConstraint(fields=["price"], name="uniq_prices")
- msg = "Product with this Price already exists."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(Product, Product(price=42))
- @skipUnlessDBFeature("supports_partial_indexes")
- def test_validate_condition(self):
- p1 = UniqueConstraintConditionProduct.objects.create(name="p1")
- constraint = UniqueConstraintConditionProduct._meta.constraints[0]
- msg = "Constraint “name_without_color_uniq” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(
- UniqueConstraintConditionProduct,
- UniqueConstraintConditionProduct(name=p1.name, color=None),
- )
- # Values not matching condition are ignored.
- constraint.validate(
- UniqueConstraintConditionProduct,
- UniqueConstraintConditionProduct(name=p1.name, color="anything-but-none"),
- )
- # Existing instances have their existing row excluded.
- constraint.validate(UniqueConstraintConditionProduct, p1)
- # Unique field is excluded.
- constraint.validate(
- UniqueConstraintConditionProduct,
- UniqueConstraintConditionProduct(name=p1.name, color=None),
- exclude={"name"},
- )
- @skipUnlessDBFeature("supports_partial_indexes")
- def test_validate_condition_custom_error(self):
- p1 = UniqueConstraintConditionProduct.objects.create(name="p1")
- constraint = models.UniqueConstraint(
- fields=["name"],
- name="name_without_color_uniq",
- condition=models.Q(color__isnull=True),
- violation_error_code="custom_code",
- violation_error_message="Custom message",
- )
- msg = "Custom message"
- with self.assertRaisesMessage(ValidationError, msg) as cm:
- constraint.validate(
- UniqueConstraintConditionProduct,
- UniqueConstraintConditionProduct(name=p1.name, color=None),
- )
- self.assertEqual(cm.exception.code, "custom_code")
- def test_validate_expression(self):
- constraint = models.UniqueConstraint(Lower("name"), name="name_lower_uniq")
- msg = "Constraint “name_lower_uniq” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p1.name.upper()),
- )
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name="another-name"),
- )
- # Existing instances have their existing row excluded.
- constraint.validate(UniqueConstraintProduct, self.p1)
- # Unique field is excluded.
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p1.name.upper()),
- exclude={"name"},
- )
- def test_validate_ordered_expression(self):
- constraint = models.UniqueConstraint(
- Lower("name").desc(), name="name_lower_uniq_desc"
- )
- msg = "Constraint “name_lower_uniq_desc” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p1.name.upper()),
- )
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name="another-name"),
- )
- # Existing instances have their existing row excluded.
- constraint.validate(UniqueConstraintProduct, self.p1)
- # Unique field is excluded.
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p1.name.upper()),
- exclude={"name"},
- )
- def test_validate_expression_condition(self):
- constraint = models.UniqueConstraint(
- Lower("name"),
- name="name_lower_without_color_uniq",
- condition=models.Q(color__isnull=True),
- )
- non_unique_product = UniqueConstraintProduct(name=self.p2.name.upper())
- msg = "Constraint “name_lower_without_color_uniq” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(UniqueConstraintProduct, non_unique_product)
- # Values not matching condition are ignored.
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p1.name, color=self.p1.color),
- )
- # Existing instances have their existing row excluded.
- constraint.validate(UniqueConstraintProduct, self.p2)
- # Unique field is excluded.
- constraint.validate(
- UniqueConstraintProduct,
- non_unique_product,
- exclude={"name"},
- )
- # Field from a condition is excluded.
- constraint.validate(
- UniqueConstraintProduct,
- non_unique_product,
- exclude={"color"},
- )
- def test_validate_expression_str(self):
- constraint = models.UniqueConstraint("name", name="name_uniq")
- msg = "Constraint “name_uniq” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p1.name),
- )
- constraint.validate(
- UniqueConstraintProduct,
- UniqueConstraintProduct(name=self.p1.name),
- exclude={"name"},
- )
- @skipUnlessDBFeature("supports_stored_generated_columns")
- def test_validate_expression_generated_field_stored(self):
- self.assertGeneratedFieldWithExpressionIsValidated(
- model=GeneratedFieldStoredProduct
- )
- @skipUnlessDBFeature("supports_virtual_generated_columns")
- def test_validate_expression_generated_field_virtual(self):
- self.assertGeneratedFieldWithExpressionIsValidated(
- model=GeneratedFieldVirtualProduct
- )
- def assertGeneratedFieldWithExpressionIsValidated(self, model):
- constraint = UniqueConstraint(Sqrt("rebate"), name="unique_rebate_sqrt")
- model.objects.create(price=100, discounted_price=84)
- valid_product = model(price=100, discounted_price=75)
- constraint.validate(model, valid_product)
- invalid_product = model(price=20, discounted_price=4)
- with self.assertRaisesMessage(
- ValidationError, f"Constraint “{constraint.name}” is violated."
- ):
- constraint.validate(model, invalid_product)
- # Excluding referenced or generated fields should skip validation.
- constraint.validate(model, invalid_product, exclude={"rebate"})
- constraint.validate(model, invalid_product, exclude={"price"})
- @skipUnlessDBFeature("supports_stored_generated_columns")
- def test_validate_fields_generated_field_stored(self):
- self.assertGeneratedFieldWithFieldsIsValidated(
- model=GeneratedFieldStoredProduct
- )
- @skipUnlessDBFeature("supports_virtual_generated_columns")
- def test_validate_fields_generated_field_virtual(self):
- self.assertGeneratedFieldWithFieldsIsValidated(
- model=GeneratedFieldVirtualProduct
- )
- def assertGeneratedFieldWithFieldsIsValidated(self, model):
- constraint = models.UniqueConstraint(
- fields=["lower_name"], name="lower_name_unique"
- )
- model.objects.create(name="Box")
- constraint.validate(model, model(name="Case"))
- invalid_product = model(name="BOX")
- msg = str(invalid_product.unique_error_message(model, ["lower_name"]))
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(model, invalid_product)
- # Excluding referenced or generated fields should skip validation.
- constraint.validate(model, invalid_product, exclude={"lower_name"})
- constraint.validate(model, invalid_product, exclude={"name"})
- @skipUnlessDBFeature("supports_stored_generated_columns")
- def test_validate_fields_generated_field_stored_nulls_distinct(self):
- self.assertGeneratedFieldNullsDistinctIsValidated(
- model=GeneratedFieldStoredProduct
- )
- @skipUnlessDBFeature("supports_virtual_generated_columns")
- def test_validate_fields_generated_field_virtual_nulls_distinct(self):
- self.assertGeneratedFieldNullsDistinctIsValidated(
- model=GeneratedFieldVirtualProduct
- )
- def assertGeneratedFieldNullsDistinctIsValidated(self, model):
- constraint = models.UniqueConstraint(
- fields=["lower_name"],
- name="lower_name_unique_nulls_distinct",
- nulls_distinct=False,
- )
- model.objects.create(name=None)
- valid_product = model(name="Box")
- constraint.validate(model, valid_product)
- invalid_product = model(name=None)
- msg = str(invalid_product.unique_error_message(model, ["lower_name"]))
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(model, invalid_product)
- @skipUnlessDBFeature("supports_table_check_constraints")
- def test_validate_nullable_textfield_with_isnull_true(self):
- is_null_constraint = models.UniqueConstraint(
- "price",
- "discounted_price",
- condition=models.Q(unit__isnull=True),
- name="uniq_prices_no_unit",
- )
- is_not_null_constraint = models.UniqueConstraint(
- "price",
- "discounted_price",
- condition=models.Q(unit__isnull=False),
- name="uniq_prices_unit",
- )
- Product.objects.create(price=2, discounted_price=1)
- Product.objects.create(price=4, discounted_price=3, unit="ng/mL")
- msg = "Constraint “uniq_prices_no_unit” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- is_null_constraint.validate(
- Product, Product(price=2, discounted_price=1, unit=None)
- )
- is_null_constraint.validate(
- Product, Product(price=2, discounted_price=1, unit="ng/mL")
- )
- is_null_constraint.validate(Product, Product(price=4, discounted_price=3))
- msg = "Constraint “uniq_prices_unit” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- is_not_null_constraint.validate(
- Product,
- Product(price=4, discounted_price=3, unit="μg/mL"),
- )
- is_not_null_constraint.validate(Product, Product(price=4, discounted_price=3))
- is_not_null_constraint.validate(Product, Product(price=2, discounted_price=1))
- @skipUnlessDBFeature("supports_table_check_constraints")
- def test_validate_nulls_distinct_fields(self):
- Product.objects.create(price=42)
- constraint = models.UniqueConstraint(
- fields=["price"],
- nulls_distinct=False,
- name="uniq_prices_nulls_distinct",
- )
- constraint.validate(Product, Product(price=None))
- Product.objects.create(price=None)
- msg = "Product with this Price already exists."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(Product, Product(price=None))
- @skipUnlessDBFeature("supports_table_check_constraints")
- def test_validate_nulls_distinct_expressions(self):
- Product.objects.create(price=42)
- constraint = models.UniqueConstraint(
- Abs("price"),
- nulls_distinct=False,
- name="uniq_prices_nulls_distinct",
- )
- constraint.validate(Product, Product(price=None))
- Product.objects.create(price=None)
- msg = f"Constraint “{constraint.name}” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- constraint.validate(Product, Product(price=None))
- def test_name(self):
- constraints = get_constraints(UniqueConstraintProduct._meta.db_table)
- expected_name = "name_color_uniq"
- self.assertIn(expected_name, constraints)
- def test_condition_must_be_q(self):
- with self.assertRaisesMessage(
- ValueError, "UniqueConstraint.condition must be a Q instance."
- ):
- models.UniqueConstraint(name="uniq", fields=["name"], condition="invalid")
- @skipUnlessDBFeature("supports_deferrable_unique_constraints")
- def test_initially_deferred_database_constraint(self):
- obj_1 = UniqueConstraintDeferrable.objects.create(name="p1", shelf="front")
- obj_2 = UniqueConstraintDeferrable.objects.create(name="p2", shelf="back")
- def swap():
- obj_1.name, obj_2.name = obj_2.name, obj_1.name
- obj_1.save()
- obj_2.save()
- swap()
- # Behavior can be changed with SET CONSTRAINTS.
- with self.assertRaises(IntegrityError):
- with atomic(), connection.cursor() as cursor:
- constraint_name = connection.ops.quote_name("name_init_deferred_uniq")
- cursor.execute("SET CONSTRAINTS %s IMMEDIATE" % constraint_name)
- swap()
- @skipUnlessDBFeature("supports_deferrable_unique_constraints")
- def test_initially_immediate_database_constraint(self):
- obj_1 = UniqueConstraintDeferrable.objects.create(name="p1", shelf="front")
- obj_2 = UniqueConstraintDeferrable.objects.create(name="p2", shelf="back")
- obj_1.shelf, obj_2.shelf = obj_2.shelf, obj_1.shelf
- with self.assertRaises(IntegrityError), atomic():
- obj_1.save()
- # Behavior can be changed with SET CONSTRAINTS.
- with connection.cursor() as cursor:
- constraint_name = connection.ops.quote_name("sheld_init_immediate_uniq")
- cursor.execute("SET CONSTRAINTS %s DEFERRED" % constraint_name)
- obj_1.save()
- obj_2.save()
- def test_deferrable_with_condition(self):
- message = "UniqueConstraint with conditions cannot be deferred."
- with self.assertRaisesMessage(ValueError, message):
- models.UniqueConstraint(
- fields=["name"],
- name="name_without_color_unique",
- condition=models.Q(color__isnull=True),
- deferrable=models.Deferrable.DEFERRED,
- )
- def test_deferrable_with_include(self):
- message = "UniqueConstraint with include fields cannot be deferred."
- with self.assertRaisesMessage(ValueError, message):
- models.UniqueConstraint(
- fields=["name"],
- name="name_inc_color_color_unique",
- include=["color"],
- deferrable=models.Deferrable.DEFERRED,
- )
- def test_deferrable_with_opclasses(self):
- message = "UniqueConstraint with opclasses cannot be deferred."
- with self.assertRaisesMessage(ValueError, message):
- models.UniqueConstraint(
- fields=["name"],
- name="name_text_pattern_ops_unique",
- opclasses=["text_pattern_ops"],
- deferrable=models.Deferrable.DEFERRED,
- )
- def test_deferrable_with_expressions(self):
- message = "UniqueConstraint with expressions cannot be deferred."
- with self.assertRaisesMessage(ValueError, message):
- models.UniqueConstraint(
- Lower("name"),
- name="deferred_expression_unique",
- deferrable=models.Deferrable.DEFERRED,
- )
- def test_invalid_defer_argument(self):
- message = "UniqueConstraint.deferrable must be a Deferrable instance."
- with self.assertRaisesMessage(TypeError, message):
- models.UniqueConstraint(
- fields=["name"],
- name="name_invalid",
- deferrable="invalid",
- )
- @skipUnlessDBFeature(
- "supports_table_check_constraints",
- "supports_covering_indexes",
- )
- def test_include_database_constraint(self):
- UniqueConstraintInclude.objects.create(name="p1", color="red")
- with self.assertRaises(IntegrityError):
- UniqueConstraintInclude.objects.create(name="p1", color="blue")
- def test_invalid_include_argument(self):
- msg = "UniqueConstraint.include must be a list or tuple."
- with self.assertRaisesMessage(TypeError, msg):
- models.UniqueConstraint(
- name="uniq_include",
- fields=["field"],
- include="other",
- )
- def test_invalid_opclasses_argument(self):
- msg = "UniqueConstraint.opclasses must be a list or tuple."
- with self.assertRaisesMessage(TypeError, msg):
- models.UniqueConstraint(
- name="uniq_opclasses",
- fields=["field"],
- opclasses="jsonb_path_ops",
- )
- def test_invalid_nulls_distinct_argument(self):
- msg = "UniqueConstraint.nulls_distinct must be a bool."
- with self.assertRaisesMessage(TypeError, msg):
- models.UniqueConstraint(
- name="uniq_opclasses", fields=["field"], nulls_distinct="NULLS DISTINCT"
- )
- def test_opclasses_and_fields_same_length(self):
- msg = (
- "UniqueConstraint.fields and UniqueConstraint.opclasses must have "
- "the same number of elements."
- )
- with self.assertRaisesMessage(ValueError, msg):
- models.UniqueConstraint(
- name="uniq_opclasses",
- fields=["field"],
- opclasses=["foo", "bar"],
- )
- def test_requires_field_or_expression(self):
- msg = (
- "At least one field or expression is required to define a unique "
- "constraint."
- )
- with self.assertRaisesMessage(ValueError, msg):
- models.UniqueConstraint(name="name")
- def test_expressions_and_fields_mutually_exclusive(self):
- msg = "UniqueConstraint.fields and expressions are mutually exclusive."
- with self.assertRaisesMessage(ValueError, msg):
- models.UniqueConstraint(Lower("field_1"), fields=["field_2"], name="name")
- def test_expressions_with_opclasses(self):
- msg = (
- "UniqueConstraint.opclasses cannot be used with expressions. Use "
- "django.contrib.postgres.indexes.OpClass() instead."
- )
- with self.assertRaisesMessage(ValueError, msg):
- models.UniqueConstraint(
- Lower("field"),
- name="test_func_opclass",
- opclasses=["jsonb_path_ops"],
- )
- def test_requires_name(self):
- msg = "A unique constraint must be named."
- with self.assertRaisesMessage(ValueError, msg):
- models.UniqueConstraint(fields=["field"])
- def test_database_default(self):
- models.UniqueConstraint(
- fields=["field_with_db_default"], name="unique_field_with_db_default"
- ).validate(ModelWithDatabaseDefault, ModelWithDatabaseDefault())
- models.UniqueConstraint(
- Upper("field_with_db_default"),
- name="unique_field_with_db_default_expression",
- ).validate(ModelWithDatabaseDefault, ModelWithDatabaseDefault())
- ModelWithDatabaseDefault.objects.create()
- msg = (
- "Model with database default with this Field with db default already "
- "exists."
- )
- with self.assertRaisesMessage(ValidationError, msg):
- models.UniqueConstraint(
- fields=["field_with_db_default"], name="unique_field_with_db_default"
- ).validate(ModelWithDatabaseDefault, ModelWithDatabaseDefault())
- msg = "Constraint “unique_field_with_db_default_expression” is violated."
- with self.assertRaisesMessage(ValidationError, msg):
- models.UniqueConstraint(
- Upper("field_with_db_default"),
- name="unique_field_with_db_default_expression",
- ).validate(ModelWithDatabaseDefault, ModelWithDatabaseDefault())
|