diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/migrations/test_operations.py | 23 |
1 files changed, 23 insertions, 0 deletions
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 |