Skip to content

Commit 4e563f8

Browse files
committed
added a test for running a drop default migration in reverse
1 parent 3b572e1 commit 4e563f8

File tree

1 file changed

+25
-2
lines changed

1 file changed

+25
-2
lines changed

tests/apps/migrations/auto/test_migration_manager.py

Lines changed: 25 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,6 @@ def test_alter_column_drop_default(self):
351351
old_params={"default": None},
352352
)
353353
asyncio.run(manager_1.run())
354-
355354
self.assertEqual(
356355
self._get_column_default(),
357356
[{"column_default": "'Mr Manager'::character varying"}],
@@ -364,10 +363,34 @@ def test_alter_column_drop_default(self):
364363
tablename="manager",
365364
column_name="name",
366365
params={"default": None},
367-
old_params={"default": "My Manager"},
366+
old_params={"default": "Mr Manager"},
368367
)
369368
asyncio.run(manager_2.run())
369+
self.assertEqual(
370+
self._get_column_default(), [{"column_default": None}],
371+
)
372+
373+
# And add it back once more to be sure.
374+
manager_3 = manager_1
375+
asyncio.run(manager_3.run())
376+
self.assertEqual(
377+
self._get_column_default(),
378+
[{"column_default": "'Mr Manager'::character varying"}],
379+
)
380+
381+
# Run them all backwards
382+
asyncio.run(manager_3.run_backwards())
383+
self.assertEqual(
384+
self._get_column_default(), [{"column_default": None}],
385+
)
386+
387+
asyncio.run(manager_2.run_backwards())
388+
self.assertEqual(
389+
self._get_column_default(),
390+
[{"column_default": "'Mr Manager'::character varying"}],
391+
)
370392

393+
asyncio.run(manager_1.run_backwards())
371394
self.assertEqual(
372395
self._get_column_default(), [{"column_default": None}],
373396
)

0 commit comments

Comments
 (0)