mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-10-24 06:06:09 +00:00
Code Issues 32 Releases Wiki Activity

Refs #23919 -- Replaced tempfile.mkdtemp() with TemporaryDirectory() context manager.

Browse Source
This commit is contained in:
Chillar Anand
2017-01-27 00:24:16 +05:30
committed by Tim Graham
parent fee42fd99e
commit 6478e07a62
5 changed files with 34 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/migrations/test_base.py
View File

@@ -98,8 +98,7 @@ class MigrationTestBase(TransactionTestCase):
Returns the filesystem path to the temporary migrations module.
"""
temp_dir = tempfile.mkdtemp()
try:
with tempfile.TemporaryDirectory() as temp_dir:
target_dir = tempfile.mkdtemp(dir=temp_dir)
with open(os.path.join(target_dir, '__init__.py'), 'w'):
pass
@@ -119,6 +118,3 @@ class MigrationTestBase(TransactionTestCase):
new_module = os.path.basename(target_dir) + '.migrations'
with self.settings(MIGRATION_MODULES={app_label: new_module}):
yield target_migrations_dir
finally:
shutil.rmtree(temp_dir)