|
@@ -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.
|