Issue #18401: Fix test_pdb on Windows

* Use "with Popen" to cleanup properly the process
* Use support.temp_cwd() to properly change the working directory
* Use environ.pop() to cleanup the code
This commit is contained in:
Victor Stinner 2016-09-09 23:22:09 -07:00
parent 54de2b1edd
commit bc6262675c
1 changed files with 7 additions and 14 deletions

View File

@ -4,7 +4,6 @@ import doctest
import os
import pdb
import sys
import tempfile
import types
import unittest
import subprocess
@ -1057,19 +1056,15 @@ class PdbTestCase(unittest.TestCase):
def test_readrc_kwarg(self):
save_home = os.environ.get('HOME', None)
save_dir = os.getcwd()
script = textwrap.dedent("""
import pdb; pdb.Pdb(readrc=False).set_trace()
print('hello')
""")
try:
if save_home is not None:
del os.environ['HOME']
with tempfile.TemporaryDirectory() as dirname:
os.chdir(dirname)
save_home = os.environ.pop('HOME', None)
try:
with support.temp_cwd():
with open('.pdbrc', 'w') as f:
f.write("invalid\n")
@ -1083,16 +1078,14 @@ class PdbTestCase(unittest.TestCase):
stdin=subprocess.PIPE,
stderr=subprocess.PIPE,
)
self.addCleanup(proc.stdout.close)
self.addCleanup(proc.stderr.close)
stdout, stderr = proc.communicate(b'q\n')
self.assertNotIn("NameError: name 'invalid' is not defined",
stdout.decode())
with proc:
stdout, stderr = proc.communicate(b'q\n')
self.assertNotIn("NameError: name 'invalid' is not defined",
stdout.decode())
finally:
if save_home is not None:
os.environ['HOME'] = save_home
os.chdir(save_dir)
def tearDown(self):
support.unlink(support.TESTFN)