diff options
| author | Jason R. Coombs <jaraco@jaraco.com> | 2015-01-02 13:09:46 -0500 |
|---|---|---|
| committer | Jason R. Coombs <jaraco@jaraco.com> | 2015-01-02 13:09:46 -0500 |
| commit | a87442aab8bd8779b562043540c467ffffb4d581 (patch) | |
| tree | 61bd04b676f679db2d7e59b3fc1601fe0cc900ad /setuptools/tests/test_upload_docs.py | |
| parent | b72261b9a0c9e45c730eff86fae594428f6f932a (diff) | |
| download | external_python_setuptools-a87442aab8bd8779b562043540c467ffffb4d581.tar.gz external_python_setuptools-a87442aab8bd8779b562043540c467ffffb4d581.tar.bz2 external_python_setuptools-a87442aab8bd8779b562043540c467ffffb4d581.zip | |
Use tempdir context
Diffstat (limited to 'setuptools/tests/test_upload_docs.py')
| -rw-r--r-- | setuptools/tests/test_upload_docs.py | 11 |
1 files changed, 3 insertions, 8 deletions
diff --git a/setuptools/tests/test_upload_docs.py b/setuptools/tests/test_upload_docs.py index 6fe9e051..66d06dc0 100644 --- a/setuptools/tests/test_upload_docs.py +++ b/setuptools/tests/test_upload_docs.py @@ -1,6 +1,4 @@ import os -import shutil -import tempfile import zipfile import pytest @@ -9,6 +7,7 @@ from setuptools.command.upload_docs import upload_docs from setuptools.dist import Distribution from .textwrap import DALS +from . import contexts SETUP_PY = DALS( @@ -49,9 +48,8 @@ class TestUploadDocsTest: cmd = upload_docs(dist) cmd.target_dir = cmd.upload_dir = 'build' - tmp_dir = tempfile.mkdtemp() - tmp_file = os.path.join(tmp_dir, 'foo.zip') - try: + with contexts.tempdir() as tmp_dir: + tmp_file = os.path.join(tmp_dir, 'foo.zip') zip_file = cmd.create_zipfile(tmp_file) assert zipfile.is_zipfile(tmp_file) @@ -61,6 +59,3 @@ class TestUploadDocsTest: assert zip_file.namelist() == ['index.html'] zip_file.close() - finally: - shutil.rmtree(tmp_dir) - |