Commit 8b899dab authored by Jason R. Coombs's avatar Jason R. Coombs

Use tmpdir fixture

parent b96f9cd4
"""develop tests """develop tests
""" """
import os import os
import shutil
import tempfile
import types import types
import pytest import pytest
...@@ -14,14 +12,8 @@ from setuptools.sandbox import DirectorySandbox, SandboxViolation ...@@ -14,14 +12,8 @@ from setuptools.sandbox import DirectorySandbox, SandboxViolation
class TestSandbox: class TestSandbox:
def setup_method(self, method): def test_devnull(self, tmpdir):
self.dir = tempfile.mkdtemp() sandbox = DirectorySandbox(str(tmpdir))
def teardown_method(self, method):
shutil.rmtree(self.dir)
def test_devnull(self):
sandbox = DirectorySandbox(self.dir)
sandbox.run(self._file_writer(os.devnull)) sandbox.run(self._file_writer(os.devnull))
@staticmethod @staticmethod
...@@ -31,7 +23,7 @@ class TestSandbox: ...@@ -31,7 +23,7 @@ class TestSandbox:
f.write('xxx') f.write('xxx')
return do_write return do_write
def test_win32com(self): def test_win32com(self, tmpdir):
""" """
win32com should not be prevented from caching COM interfaces win32com should not be prevented from caching COM interfaces
in gen_py. in gen_py.
...@@ -39,7 +31,7 @@ class TestSandbox: ...@@ -39,7 +31,7 @@ class TestSandbox:
win32com = pytest.importorskip('win32com') win32com = pytest.importorskip('win32com')
gen_py = win32com.__gen_path__ gen_py = win32com.__gen_path__
target = os.path.join(gen_py, 'test_write') target = os.path.join(gen_py, 'test_write')
sandbox = DirectorySandbox(self.dir) sandbox = DirectorySandbox(str(tmpdir))
try: try:
try: try:
sandbox.run(self._file_writer(target)) sandbox.run(self._file_writer(target))
...@@ -59,8 +51,8 @@ class TestSandbox: ...@@ -59,8 +51,8 @@ class TestSandbox:
setuptools.sandbox._execfile(target, vars(namespace)) setuptools.sandbox._execfile(target, vars(namespace))
assert namespace.result == 'passed' assert namespace.result == 'passed'
def test_setup_py_with_CRLF(self): def test_setup_py_with_CRLF(self, tmpdir):
setup_py = os.path.join(self.dir, 'setup.py') setup_py = tmpdir / 'setup.py'
with open(setup_py, 'wb') as stream: with setup_py.open('wb') as stream:
stream.write(b'"degenerate script"\r\n') stream.write(b'"degenerate script"\r\n')
setuptools.sandbox._execfile(setup_py, globals()) setuptools.sandbox._execfile(str(setup_py), globals())
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