Tarek Ziadé | 275958a | 2009-02-13 22:22:03 +0000 | [diff] [blame] | 1 | """Tests for distutils.util.""" |
Tarek Ziadé | 275958a | 2009-02-13 22:22:03 +0000 | [diff] [blame] | 2 | import sys |
| 3 | import unittest |
Éric Araujo | 54274ad | 2011-02-03 00:12:18 +0000 | [diff] [blame] | 4 | from test.test_support import run_unittest |
Tarek Ziadé | 275958a | 2009-02-13 22:22:03 +0000 | [diff] [blame] | 5 | |
Éric Araujo | 3d1134e | 2014-03-12 03:14:48 -0400 | [diff] [blame] | 6 | from distutils.errors import DistutilsByteCompileError |
Victor Stinner | 15f8d0d | 2017-05-03 17:28:10 +0200 | [diff] [blame] | 7 | from distutils.tests import support |
Éric Araujo | 3d1134e | 2014-03-12 03:14:48 -0400 | [diff] [blame] | 8 | from distutils.util import byte_compile, grok_environment_error |
| 9 | |
Tarek Ziadé | a99dedf | 2009-07-16 15:35:45 +0000 | [diff] [blame] | 10 | |
Victor Stinner | 15f8d0d | 2017-05-03 17:28:10 +0200 | [diff] [blame] | 11 | class UtilTestCase(support.EnvironGuard, unittest.TestCase): |
Tarek Ziadé | a99dedf | 2009-07-16 15:35:45 +0000 | [diff] [blame] | 12 | |
Tarek Ziadé | b9c1cfc | 2009-10-24 15:10:37 +0000 | [diff] [blame] | 13 | def test_dont_write_bytecode(self): |
| 14 | # makes sure byte_compile raise a DistutilsError |
| 15 | # if sys.dont_write_bytecode is True |
| 16 | old_dont_write_bytecode = sys.dont_write_bytecode |
| 17 | sys.dont_write_bytecode = True |
| 18 | try: |
| 19 | self.assertRaises(DistutilsByteCompileError, byte_compile, []) |
| 20 | finally: |
Tarek Ziadé | 468f700 | 2009-10-24 15:19:03 +0000 | [diff] [blame] | 21 | sys.dont_write_bytecode = old_dont_write_bytecode |
Tarek Ziadé | b9c1cfc | 2009-10-24 15:10:37 +0000 | [diff] [blame] | 22 | |
Éric Araujo | 3d1134e | 2014-03-12 03:14:48 -0400 | [diff] [blame] | 23 | def test_grok_environment_error(self): |
| 24 | # test obsolete function to ensure backward compat (#4931) |
| 25 | exc = IOError("Unable to find batch file") |
| 26 | msg = grok_environment_error(exc) |
| 27 | self.assertEqual(msg, "error: Unable to find batch file") |
| 28 | |
| 29 | |
Tarek Ziadé | 275958a | 2009-02-13 22:22:03 +0000 | [diff] [blame] | 30 | def test_suite(): |
Tarek Ziadé | 450ca11 | 2009-05-10 12:17:30 +0000 | [diff] [blame] | 31 | return unittest.makeSuite(UtilTestCase) |
Tarek Ziadé | 275958a | 2009-02-13 22:22:03 +0000 | [diff] [blame] | 32 | |
| 33 | if __name__ == "__main__": |
Éric Araujo | 54274ad | 2011-02-03 00:12:18 +0000 | [diff] [blame] | 34 | run_unittest(test_suite()) |