def setUp(self):
super(TempdirManager, self).setUp()
self._basetempdir = tempfile.mkdtemp()
+ self._files = []
def tearDown(self):
shutil.rmtree(self._basetempdir, os.name in ('nt', 'cygwin'))
+
+ for handle, name in self._files:
+ handle.close()
+ if os.path.exists(name):
+ try:
+ os.remove(name)
+ except OSError as exc:
+ if exc.errno != errno.ENOENT:
+ raise
+
super(TempdirManager, self).tearDown()
def mktempfile(self):
"""Create a read-write temporary file and return it."""
-
- def _delete_file(filename):
- try:
- os.remove(filename)
- except OSError as exc:
- if exc.errno != errno.ENOENT:
- raise
-
fd, fn = tempfile.mkstemp(dir=self._basetempdir)
os.close(fd)
fp = open(fn, 'w+')
- self.addCleanup(fp.close)
- self.addCleanup(_delete_file, fn)
+ self._files.append((fp, fn))
return fp
def mkdtemp(self):
startupinfo=None, creationflags=0,
restore_signals=True, start_new_session=False,
pass_fds=()):
- self.cmd = args.split()[0]
+ if isinstance(args, str):
+ args = args.split()
+ self.cmd = args[0]
exes = self.test_class._exes
if self.cmd not in exes:
# we don't want to call the system, returning an empty
def communicate(self, input=None, timeout=None):
return self.stdout.read(), self.stderr.read()
+ def wait(self, timeout=None):
+ return 0
+
class UtilTestCase(support.EnvironRestorer,
support.TempdirManager,
@unittest.skipUnless(os.name in ('nt', 'posix'),
'runs only under posix or nt')
def test_spawn(self):
- # Do not patch subprocess on unix because
- # packaging.util._spawn_posix uses it
- if os.name in 'posix':
- subprocess.Popen = self.old_popen
+ # no patching of Popen here
+ subprocess.Popen = self.old_popen
tmpdir = self.mkdtemp()
# creating something executable