mirror of
https://github.com/django/django.git
synced 2025-10-24 06:06:09 +00:00
Fixed #20581 -- Added support for deferrable unique constraints.
This commit is contained in:
Ian Foote
committed by
Mariusz Felisiak
Mariusz Felisiak
parent
555e3a848e
commit
c226c6cb32
@@ -393,6 +393,60 @@ class OperationTests(OperationTestBase):
|
||||
self.assertEqual(definition[1], [])
|
||||
self.assertEqual(definition[2]['options']['constraints'], [partial_unique_constraint])
|
||||
|
||||
def test_create_model_with_deferred_unique_constraint(self):
|
||||
deferred_unique_constraint = models.UniqueConstraint(
|
||||
fields=['pink'],
|
||||
name='deferrable_pink_constraint',
|
||||
deferrable=models.Deferrable.DEFERRED,
|
||||
)
|
||||
operation = migrations.CreateModel(
|
||||
'Pony',
|
||||
[
|
||||
('id', models.AutoField(primary_key=True)),
|
||||
('pink', models.IntegerField(default=3)),
|
||||
],
|
||||
options={'constraints': [deferred_unique_constraint]},
|
||||
)
|
||||
project_state = ProjectState()
|
||||
new_state = project_state.clone()
|
||||
operation.state_forwards('test_crmo', new_state)
|
||||
self.assertEqual(len(new_state.models['test_crmo', 'pony'].options['constraints']), 1)
|
||||
self.assertTableNotExists('test_crmo_pony')
|
||||
# Create table.
|
||||
with connection.schema_editor() as editor:
|
||||
operation.database_forwards('test_crmo', editor, project_state, new_state)
|
||||
self.assertTableExists('test_crmo_pony')
|
||||
Pony = new_state.apps.get_model('test_crmo', 'Pony')
|
||||
Pony.objects.create(pink=1)
|
||||
if connection.features.supports_deferrable_unique_constraints:
|
||||
# Unique constraint is deferred.
|
||||
with transaction.atomic():
|
||||
obj = Pony.objects.create(pink=1)
|
||||
obj.pink = 2
|
||||
obj.save()
|
||||
# Constraint behavior can be changed with SET CONSTRAINTS.
|
||||
with self.assertRaises(IntegrityError):
|
||||
with transaction.atomic(), connection.cursor() as cursor:
|
||||
quoted_name = connection.ops.quote_name(deferred_unique_constraint.name)
|
||||
cursor.execute('SET CONSTRAINTS %s IMMEDIATE' % quoted_name)
|
||||
obj = Pony.objects.create(pink=1)
|
||||
obj.pink = 3
|
||||
obj.save()
|
||||
else:
|
||||
Pony.objects.create(pink=1)
|
||||
# Reversal.
|
||||
with connection.schema_editor() as editor:
|
||||
operation.database_backwards('test_crmo', editor, new_state, project_state)
|
||||
self.assertTableNotExists('test_crmo_pony')
|
||||
# Deconstruction.
|
||||
definition = operation.deconstruct()
|
||||
self.assertEqual(definition[0], 'CreateModel')
|
||||
self.assertEqual(definition[1], [])
|
||||
self.assertEqual(
|
||||
definition[2]['options']['constraints'],
|
||||
[deferred_unique_constraint],
|
||||
)
|
||||
|
||||
def test_create_model_managers(self):
|
||||
"""
|
||||
The managers on a model are set.
|
||||
@@ -2046,6 +2100,110 @@ class OperationTests(OperationTestBase):
|
||||
'name': 'test_constraint_pony_pink_for_weight_gt_5_uniq',
|
||||
})
|
||||
|
||||
def test_add_deferred_unique_constraint(self):
|
||||
app_label = 'test_adddeferred_uc'
|
||||
project_state = self.set_up_test_model(app_label)
|
||||
deferred_unique_constraint = models.UniqueConstraint(
|
||||
fields=['pink'],
|
||||
name='deferred_pink_constraint_add',
|
||||
deferrable=models.Deferrable.DEFERRED,
|
||||
)
|
||||
operation = migrations.AddConstraint('Pony', deferred_unique_constraint)
|
||||
self.assertEqual(
|
||||
operation.describe(),
|
||||
'Create constraint deferred_pink_constraint_add on model Pony',
|
||||
)
|
||||
# Add constraint.
|
||||
new_state = project_state.clone()
|
||||
operation.state_forwards(app_label, new_state)
|
||||
self.assertEqual(len(new_state.models[app_label, 'pony'].options['constraints']), 1)
|
||||
Pony = new_state.apps.get_model(app_label, 'Pony')
|
||||
self.assertEqual(len(Pony._meta.constraints), 1)
|
||||
with connection.schema_editor() as editor:
|
||||
operation.database_forwards(app_label, editor, project_state, new_state)
|
||||
Pony.objects.create(pink=1, weight=4.0)
|
||||
if connection.features.supports_deferrable_unique_constraints:
|
||||
# Unique constraint is deferred.
|
||||
with transaction.atomic():
|
||||
obj = Pony.objects.create(pink=1, weight=4.0)
|
||||
obj.pink = 2
|
||||
obj.save()
|
||||
# Constraint behavior can be changed with SET CONSTRAINTS.
|
||||
with self.assertRaises(IntegrityError):
|
||||
with transaction.atomic(), connection.cursor() as cursor:
|
||||
quoted_name = connection.ops.quote_name(deferred_unique_constraint.name)
|
||||
cursor.execute('SET CONSTRAINTS %s IMMEDIATE' % quoted_name)
|
||||
obj = Pony.objects.create(pink=1, weight=4.0)
|
||||
obj.pink = 3
|
||||
obj.save()
|
||||
else:
|
||||
Pony.objects.create(pink=1, weight=4.0)
|
||||
# Reversal.
|
||||
with connection.schema_editor() as editor:
|
||||
operation.database_backwards(app_label, editor, new_state, project_state)
|
||||
# Constraint doesn't work.
|
||||
Pony.objects.create(pink=1, weight=4.0)
|
||||
# Deconstruction.
|
||||
definition = operation.deconstruct()
|
||||
self.assertEqual(definition[0], 'AddConstraint')
|
||||
self.assertEqual(definition[1], [])
|
||||
self.assertEqual(
|
||||
definition[2],
|
||||
{'model_name': 'Pony', 'constraint': deferred_unique_constraint},
|
||||
)
|
||||
|
||||
def test_remove_deferred_unique_constraint(self):
|
||||
app_label = 'test_removedeferred_uc'
|
||||
deferred_unique_constraint = models.UniqueConstraint(
|
||||
fields=['pink'],
|
||||
name='deferred_pink_constraint_rm',
|
||||
deferrable=models.Deferrable.DEFERRED,
|
||||
)
|
||||
project_state = self.set_up_test_model(app_label, constraints=[deferred_unique_constraint])
|
||||
operation = migrations.RemoveConstraint('Pony', deferred_unique_constraint.name)
|
||||
self.assertEqual(
|
||||
operation.describe(),
|
||||
'Remove constraint deferred_pink_constraint_rm from model Pony',
|
||||
)
|
||||
# Remove constraint.
|
||||
new_state = project_state.clone()
|
||||
operation.state_forwards(app_label, new_state)
|
||||
self.assertEqual(len(new_state.models[app_label, 'pony'].options['constraints']), 0)
|
||||
Pony = new_state.apps.get_model(app_label, 'Pony')
|
||||
self.assertEqual(len(Pony._meta.constraints), 0)
|
||||
with connection.schema_editor() as editor:
|
||||
operation.database_forwards(app_label, editor, project_state, new_state)
|
||||
# Constraint doesn't work.
|
||||
Pony.objects.create(pink=1, weight=4.0)
|
||||
Pony.objects.create(pink=1, weight=4.0).delete()
|
||||
# Reversal.
|
||||
with connection.schema_editor() as editor:
|
||||
operation.database_backwards(app_label, editor, new_state, project_state)
|
||||
if connection.features.supports_deferrable_unique_constraints:
|
||||
# Unique constraint is deferred.
|
||||
with transaction.atomic():
|
||||
obj = Pony.objects.create(pink=1, weight=4.0)
|
||||
obj.pink = 2
|
||||
obj.save()
|
||||
# Constraint behavior can be changed with SET CONSTRAINTS.
|
||||
with self.assertRaises(IntegrityError):
|
||||
with transaction.atomic(), connection.cursor() as cursor:
|
||||
quoted_name = connection.ops.quote_name(deferred_unique_constraint.name)
|
||||
cursor.execute('SET CONSTRAINTS %s IMMEDIATE' % quoted_name)
|
||||
obj = Pony.objects.create(pink=1, weight=4.0)
|
||||
obj.pink = 3
|
||||
obj.save()
|
||||
else:
|
||||
Pony.objects.create(pink=1, weight=4.0)
|
||||
# Deconstruction.
|
||||
definition = operation.deconstruct()
|
||||
self.assertEqual(definition[0], 'RemoveConstraint')
|
||||
self.assertEqual(definition[1], [])
|
||||
self.assertEqual(definition[2], {
|
||||
'model_name': 'Pony',
|
||||
'name': 'deferred_pink_constraint_rm',
|
||||
})
|
||||
|
||||
def test_alter_model_options(self):
|
||||
"""
|
||||
Tests the AlterModelOptions operation.
|
||||
|
||||
Reference in New Issue
Block a user