diff options
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/migrations/test_executor.py | 14 | ||||
| -rw-r--r-- | tests/migrations/test_migrations/0001_initial.py | 7 |
2 files changed, 20 insertions, 1 deletions
diff --git a/tests/migrations/test_executor.py b/tests/migrations/test_executor.py index b6909a15a1..e933f9d8bd 100644 --- a/tests/migrations/test_executor.py +++ b/tests/migrations/test_executor.py @@ -155,6 +155,12 @@ class ExecutorTests(MigrationTestBase): self.assertTableNotExists("migrations_author") self.assertTableNotExists("migrations_tribble") # Run it normally + self.assertEqual( + executor.migration_plan([("migrations", "0001_initial")]), + [ + (executor.loader.graph.nodes["migrations", "0001_initial"], False), + ], + ) executor.migrate([("migrations", "0001_initial")]) # Are the tables there now? self.assertTableExists("migrations_author") @@ -171,9 +177,17 @@ class ExecutorTests(MigrationTestBase): # Make sure that was faked self.assertEqual(state["faked"], True) # Finally, migrate forwards; this should fake-apply our initial migration + executor.loader.build_graph() + self.assertEqual( + executor.migration_plan([("migrations", "0001_initial")]), + [ + (executor.loader.graph.nodes["migrations", "0001_initial"], False), + ], + ) executor.migrate([("migrations", "0001_initial")]) self.assertEqual(state["faked"], True) # And migrate back to clean up the database + executor.loader.build_graph() executor.migrate([("migrations", None)]) self.assertTableNotExists("migrations_author") self.assertTableNotExists("migrations_tribble") diff --git a/tests/migrations/test_migrations/0001_initial.py b/tests/migrations/test_migrations/0001_initial.py index 581d536814..9286d0f02b 100644 --- a/tests/migrations/test_migrations/0001_initial.py +++ b/tests/migrations/test_migrations/0001_initial.py @@ -25,6 +25,11 @@ class Migration(migrations.Migration): ("id", models.AutoField(primary_key=True)), ("fluffy", models.BooleanField(default=True)), ], - ) + ), + + migrations.AlterUniqueTogether( + name='author', + unique_together=set([('name', 'slug')]), + ), ] |