Skip to content

Commit ada83fd

Browse files
authored
Fixed isolation of test_migrate_fake_split_initial.
1 parent 8b2a30f commit ada83fd

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

tests/migrations/test_commands.py

+14-12
Original file line numberDiff line numberDiff line change
@@ -229,18 +229,20 @@ def test_migrate_fake_split_initial(self):
229229
"""
230230
Split initial migrations can be faked with --fake-initial.
231231
"""
232-
call_command("migrate", "migrations", "0002", verbosity=0)
233-
call_command("migrate", "migrations", "zero", fake=True, verbosity=0)
234-
out = io.StringIO()
235-
with mock.patch('django.core.management.color.supports_color', lambda *args: False):
236-
call_command("migrate", "migrations", "0002", fake_initial=True, stdout=out, verbosity=1)
237-
value = out.getvalue().lower()
238-
self.assertIn("migrations.0001_initial... faked", value)
239-
self.assertIn("migrations.0002_second... faked", value)
240-
# Fake an apply
241-
call_command("migrate", "migrations", fake=True, verbosity=0)
242-
# Unmigrate everything
243-
call_command("migrate", "migrations", "zero", verbosity=0)
232+
try:
233+
call_command('migrate', 'migrations', '0002', verbosity=0)
234+
call_command('migrate', 'migrations', 'zero', fake=True, verbosity=0)
235+
out = io.StringIO()
236+
with mock.patch('django.core.management.color.supports_color', lambda *args: False):
237+
call_command('migrate', 'migrations', '0002', fake_initial=True, stdout=out, verbosity=1)
238+
value = out.getvalue().lower()
239+
self.assertIn('migrations.0001_initial... faked', value)
240+
self.assertIn('migrations.0002_second... faked', value)
241+
finally:
242+
# Fake an apply.
243+
call_command('migrate', 'migrations', fake=True, verbosity=0)
244+
# Unmigrate everything.
245+
call_command('migrate', 'migrations', 'zero', verbosity=0)
244246

245247
@override_settings(MIGRATION_MODULES={"migrations": "migrations.test_migrations_conflict"})
246248
def test_migrate_conflict_exit(self):

0 commit comments

Comments
 (0)