@@ -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