diff options
Diffstat (limited to 'Lib/test/test_tools.py')
| -rw-r--r-- | Lib/test/test_tools.py | 25 |
1 files changed, 24 insertions, 1 deletions
diff --git a/Lib/test/test_tools.py b/Lib/test/test_tools.py index cfe13ac..8ccabfb 100644 --- a/Lib/test/test_tools.py +++ b/Lib/test/test_tools.py @@ -8,6 +8,7 @@ import os import sys import imp import unittest +from unittest import mock import sysconfig import tempfile from test import support @@ -40,7 +41,7 @@ class TestSundryScripts(unittest.TestCase): # added for a script it should be added to the whitelist below. # scripts that have independent tests. - whitelist = ['reindent.py'] + whitelist = ['reindent.py', 'pdeps.py', 'gprof2html'] # scripts that can't be imported without running blacklist = ['make_ctype.py'] # scripts that use windows-only modules @@ -99,6 +100,28 @@ class PdepsTests(unittest.TestCase): self.pdeps.inverse({'a': []}) +class Gprof2htmlTests(unittest.TestCase): + + def setUp(self): + path = os.path.join(scriptsdir, 'gprof2html.py') + self.gprof = imp.load_source('gprof2html', path) + oldargv = sys.argv + def fixup(): + sys.argv = oldargv + self.addCleanup(fixup) + sys.argv = [] + + def test_gprof(self): + # Issue #14508: this used to fail with an NameError. + with mock.patch.object(self.gprof, 'webbrowser') as wmock, \ + tempfile.TemporaryDirectory() as tmpdir: + fn = os.path.join(tmpdir, 'abc') + open(fn, 'w').close() + sys.argv = ['gprof2html', fn] + self.gprof.main() + self.assertTrue(wmock.open.called) + + def test_main(): support.run_unittest(*[obj for obj in globals().values() if isinstance(obj, type)]) |