| """Tests for scripts in the Tools directory. |
| |
| This file contains regression tests for some of the scripts found in the |
| Tools directory of a Python checkout or tarball, such as reindent.py. |
| """ |
| |
| import os |
| import sys |
| import imp |
| import unittest |
| from unittest import mock |
| import sysconfig |
| import tempfile |
| from test import support |
| from test.script_helper import assert_python_ok |
| |
| if not sysconfig.is_python_build(): |
| # XXX some installers do contain the tools, should we detect that |
| # and run the tests in that case too? |
| raise unittest.SkipTest('test irrelevant for an installed Python') |
| |
| srcdir = sysconfig.get_config_var('projectbase') |
| basepath = os.path.join(os.getcwd(), srcdir, 'Tools') |
| scriptsdir = os.path.join(basepath, 'scripts') |
| |
| |
| class ReindentTests(unittest.TestCase): |
| script = os.path.join(scriptsdir, 'reindent.py') |
| |
| def test_noargs(self): |
| assert_python_ok(self.script) |
| |
| def test_help(self): |
| rc, out, err = assert_python_ok(self.script, '-h') |
| self.assertEqual(out, b'') |
| self.assertGreater(err, b'') |
| |
| |
| class TestSundryScripts(unittest.TestCase): |
| # At least make sure the rest don't have syntax errors. When tests are |
| # added for a script it should be added to the whitelist below. |
| |
| # scripts that have independent tests. |
| whitelist = ['reindent.py', 'pdeps.py', 'gprof2html'] |
| # scripts that can't be imported without running |
| blacklist = ['make_ctype.py'] |
| # scripts that use windows-only modules |
| windows_only = ['win_add2path.py'] |
| # blacklisted for other reasons |
| other = ['analyze_dxp.py'] |
| |
| skiplist = blacklist + whitelist + windows_only + other |
| |
| def setUp(self): |
| cm = support.DirsOnSysPath(scriptsdir) |
| cm.__enter__() |
| self.addCleanup(cm.__exit__) |
| |
| def test_sundry(self): |
| for fn in os.listdir(scriptsdir): |
| if fn.endswith('.py') and fn not in self.skiplist: |
| __import__(fn[:-3]) |
| |
| @unittest.skipIf(sys.platform != "win32", "Windows-only test") |
| def test_sundry_windows(self): |
| for fn in self.windows_only: |
| __import__(fn[:-3]) |
| |
| @unittest.skipIf(not support.threading, "test requires _thread module") |
| def test_analyze_dxp_import(self): |
| if hasattr(sys, 'getdxp'): |
| import analyze_dxp |
| else: |
| with self.assertRaises(RuntimeError): |
| import analyze_dxp |
| |
| |
| class PdepsTests(unittest.TestCase): |
| |
| @classmethod |
| def setUpClass(self): |
| path = os.path.join(scriptsdir, 'pdeps.py') |
| self.pdeps = imp.load_source('pdeps', path) |
| |
| @classmethod |
| def tearDownClass(self): |
| if 'pdeps' in sys.modules: |
| del sys.modules['pdeps'] |
| |
| def test_process_errors(self): |
| # Issue #14492: m_import.match(line) can be None. |
| with tempfile.TemporaryDirectory() as tmpdir: |
| fn = os.path.join(tmpdir, 'foo') |
| with open(fn, 'w') as stream: |
| stream.write("#!/this/will/fail") |
| self.pdeps.process(fn, {}) |
| |
| def test_inverse_attribute_error(self): |
| # Issue #14492: this used to fail with an AttributeError. |
| 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)]) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |