From 63afe3a2bfaf97fecff6641137a85296029d5b73 Mon Sep 17 00:00:00 2001 From: Ian Foote Date: Fri, 7 Apr 2017 09:44:28 +0100 Subject: Fixed #28043 -- Prevented AddIndex and RemoveIndex from mutating model state. --- tests/migrations/test_operations.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'tests') diff --git a/tests/migrations/test_operations.py b/tests/migrations/test_operations.py index 7943b59a44..a6f713e321 100644 --- a/tests/migrations/test_operations.py +++ b/tests/migrations/test_operations.py @@ -1489,6 +1489,29 @@ class OperationTests(OperationTestBase): self.unapply_operations("test_rmin", project_state, operations=operations) self.assertIndexExists("test_rmin_pony", ["pink", "weight"]) + def test_add_index_state_forwards(self): + project_state = self.set_up_test_model('test_adinsf') + index = models.Index(fields=['pink'], name='test_adinsf_pony_pink_idx') + old_model = project_state.apps.get_model('test_adinsf', 'Pony') + new_state = project_state.clone() + + operation = migrations.AddIndex('Pony', index) + operation.state_forwards('test_adinsf', new_state) + new_model = new_state.apps.get_model('test_adinsf', 'Pony') + self.assertIsNot(old_model, new_model) + + def test_remove_index_state_forwards(self): + project_state = self.set_up_test_model('test_rminsf') + index = models.Index(fields=['pink'], name='test_rminsf_pony_pink_idx') + migrations.AddIndex('Pony', index).state_forwards('test_rminsf', project_state) + old_model = project_state.apps.get_model('test_rminsf', 'Pony') + new_state = project_state.clone() + + operation = migrations.RemoveIndex('Pony', 'test_rminsf_pony_pink_idx') + operation.state_forwards('test_rminsf', new_state) + new_model = new_state.apps.get_model('test_rminsf', 'Pony') + self.assertIsNot(old_model, new_model) + def test_alter_field_with_index(self): """ Test AlterField operation with an index to ensure indexes created via -- cgit v1.3