Commit a87442aa authored by Jason R. Coombs's avatar Jason R. Coombs

Use tempdir context

parent b72261b9
import os import os
import shutil
import tempfile
import zipfile import zipfile
import pytest import pytest
...@@ -9,6 +7,7 @@ from setuptools.command.upload_docs import upload_docs ...@@ -9,6 +7,7 @@ from setuptools.command.upload_docs import upload_docs
from setuptools.dist import Distribution from setuptools.dist import Distribution
from .textwrap import DALS from .textwrap import DALS
from . import contexts
SETUP_PY = DALS( SETUP_PY = DALS(
...@@ -49,9 +48,8 @@ class TestUploadDocsTest: ...@@ -49,9 +48,8 @@ class TestUploadDocsTest:
cmd = upload_docs(dist) cmd = upload_docs(dist)
cmd.target_dir = cmd.upload_dir = 'build' cmd.target_dir = cmd.upload_dir = 'build'
tmp_dir = tempfile.mkdtemp() with contexts.tempdir() as tmp_dir:
tmp_file = os.path.join(tmp_dir, 'foo.zip') tmp_file = os.path.join(tmp_dir, 'foo.zip')
try:
zip_file = cmd.create_zipfile(tmp_file) zip_file = cmd.create_zipfile(tmp_file)
assert zipfile.is_zipfile(tmp_file) assert zipfile.is_zipfile(tmp_file)
...@@ -61,6 +59,3 @@ class TestUploadDocsTest: ...@@ -61,6 +59,3 @@ class TestUploadDocsTest:
assert zip_file.namelist() == ['index.html'] assert zip_file.namelist() == ['index.html']
zip_file.close() zip_file.close()
finally:
shutil.rmtree(tmp_dir)
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment