diff options
| author | Jon Dufresne <jon.dufresne@gmail.com> | 2020-07-20 02:08:23 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-07-20 11:08:23 +0200 |
| commit | e1e4fd707f3f906a1c23f52c987f8b04b0dfab63 (patch) | |
| tree | 8e0df455ba8e50c15c2c256ed691d955511d33f1 | |
| parent | 83f55aafdd635189c010cff403f66b54d695921a (diff) | |
Used context manager version of tempfile.TemporaryDirectory() in utils_tests.test_autoreload.
| -rw-r--r-- | tests/utils_tests/test_autoreload.py | 34 |
1 files changed, 16 insertions, 18 deletions
diff --git a/tests/utils_tests/test_autoreload.py b/tests/utils_tests/test_autoreload.py index 72ae20ac12..5b1a733b47 100644 --- a/tests/utils_tests/test_autoreload.py +++ b/tests/utils_tests/test_autoreload.py @@ -173,27 +173,25 @@ class TestChildArguments(SimpleTestCase): @mock.patch('sys.warnoptions', []) def test_exe_fallback(self): - tmpdir = tempfile.TemporaryDirectory() - self.addCleanup(tmpdir.cleanup) - exe_path = Path(tmpdir.name) / 'django-admin.exe' - exe_path.touch() - with mock.patch('sys.argv', [exe_path.with_suffix(''), 'runserver']): - self.assertEqual( - autoreload.get_child_arguments(), - [exe_path, 'runserver'] - ) + with tempfile.TemporaryDirectory() as tmpdir: + exe_path = Path(tmpdir) / 'django-admin.exe' + exe_path.touch() + with mock.patch('sys.argv', [exe_path.with_suffix(''), 'runserver']): + self.assertEqual( + autoreload.get_child_arguments(), + [exe_path, 'runserver'] + ) @mock.patch('sys.warnoptions', []) def test_entrypoint_fallback(self): - tmpdir = tempfile.TemporaryDirectory() - self.addCleanup(tmpdir.cleanup) - script_path = Path(tmpdir.name) / 'django-admin-script.py' - script_path.touch() - with mock.patch('sys.argv', [script_path.with_name('django-admin'), 'runserver']): - self.assertEqual( - autoreload.get_child_arguments(), - [sys.executable, script_path, 'runserver'] - ) + with tempfile.TemporaryDirectory() as tmpdir: + script_path = Path(tmpdir) / 'django-admin-script.py' + script_path.touch() + with mock.patch('sys.argv', [script_path.with_name('django-admin'), 'runserver']): + self.assertEqual( + autoreload.get_child_arguments(), + [sys.executable, script_path, 'runserver'] + ) @mock.patch('sys.argv', ['does-not-exist', 'runserver']) @mock.patch('sys.warnoptions', []) |