diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/migrations/test_optimizer.py | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/tests/migrations/test_optimizer.py b/tests/migrations/test_optimizer.py index 59dd401a3c..6bde378cd9 100644 --- a/tests/migrations/test_optimizer.py +++ b/tests/migrations/test_optimizer.py @@ -129,6 +129,30 @@ class OptimizerTests(SimpleTestCase): ], ) + def test_create_alter_model_managers(self): + self.assertOptimizesTo( + [ + migrations.CreateModel("Foo", fields=[]), + migrations.AlterModelManagers( + name="Foo", + managers=[ + ("objects", models.Manager()), + ("things", models.Manager()), + ], + ), + ], + [ + migrations.CreateModel( + "Foo", + fields=[], + managers=[ + ("objects", models.Manager()), + ("things", models.Manager()), + ], + ), + ], + ) + def test_create_model_and_remove_model_options(self): self.assertOptimizesTo( [ |