mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 05:31:20 +00:00 
			
		
		
		
	 c76bceae1a
			
		
	
	
		c76bceae1a
		
	
	
	
	
		
			
			svn+ssh://pythondev@svn.python.org/python/branches/py3k
................
  r75704 | tarek.ziade | 2009-10-26 00:08:47 +0100 (Mon, 26 Oct 2009) | 17 lines
  Merged revisions 75669-75671 via svnmerge from
  svn+ssh://pythondev@svn.python.org/python/trunk
  ........
    r75669 | tarek.ziade | 2009-10-24 17:10:37 +0200 (Sat, 24 Oct 2009) | 1 line
    Issue #7071: byte-compilation in Distutils now looks at sys.dont_write_bytecode
  ........
    r75670 | tarek.ziade | 2009-10-24 17:19:03 +0200 (Sat, 24 Oct 2009) | 1 line
    fixed finally state in distutils.test_util
  ........
    r75671 | tarek.ziade | 2009-10-24 17:51:30 +0200 (Sat, 24 Oct 2009) | 1 line
    fixed warning and error message
  ........
................
		
	
			
		
			
				
	
	
		
			272 lines
		
	
	
	
		
			9.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			272 lines
		
	
	
	
		
			9.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """Tests for distutils.util."""
 | |
| import os
 | |
| import sys
 | |
| import unittest
 | |
| from copy import copy
 | |
| 
 | |
| from distutils.errors import DistutilsPlatformError, DistutilsByteCompileError
 | |
| from distutils.util import (get_platform, convert_path, change_root,
 | |
|                             check_environ, split_quoted, strtobool,
 | |
|                             rfc822_escape, byte_compile)
 | |
| from distutils import util # used to patch _environ_checked
 | |
| from distutils.sysconfig import get_config_vars
 | |
| from distutils import sysconfig
 | |
| from distutils.tests import support
 | |
| 
 | |
| class UtilTestCase(support.EnvironGuard, unittest.TestCase):
 | |
| 
 | |
|     def setUp(self):
 | |
|         super(UtilTestCase, self).setUp()
 | |
|         # saving the environment
 | |
|         self.name = os.name
 | |
|         self.platform = sys.platform
 | |
|         self.version = sys.version
 | |
|         self.sep = os.sep
 | |
|         self.join = os.path.join
 | |
|         self.isabs = os.path.isabs
 | |
|         self.splitdrive = os.path.splitdrive
 | |
|         self._config_vars = copy(sysconfig._config_vars)
 | |
| 
 | |
|         # patching os.uname
 | |
|         if hasattr(os, 'uname'):
 | |
|             self.uname = os.uname
 | |
|             self._uname = os.uname()
 | |
|         else:
 | |
|             self.uname = None
 | |
|             self._uname = None
 | |
| 
 | |
|         os.uname = self._get_uname
 | |
| 
 | |
|     def tearDown(self):
 | |
|         # getting back the environment
 | |
|         os.name = self.name
 | |
|         sys.platform = self.platform
 | |
|         sys.version = self.version
 | |
|         os.sep = self.sep
 | |
|         os.path.join = self.join
 | |
|         os.path.isabs = self.isabs
 | |
|         os.path.splitdrive = self.splitdrive
 | |
|         if self.uname is not None:
 | |
|             os.uname = self.uname
 | |
|         else:
 | |
|             del os.uname
 | |
|         sysconfig._config_vars = copy(self._config_vars)
 | |
|         super(UtilTestCase, self).tearDown()
 | |
| 
 | |
|     def _set_uname(self, uname):
 | |
|         self._uname = uname
 | |
| 
 | |
|     def _get_uname(self):
 | |
|         return self._uname
 | |
| 
 | |
|     def test_get_platform(self):
 | |
| 
 | |
|         # windows XP, 32bits
 | |
|         os.name = 'nt'
 | |
|         sys.version = ('2.4.4 (#71, Oct 18 2006, 08:34:43) '
 | |
|                        '[MSC v.1310 32 bit (Intel)]')
 | |
|         sys.platform = 'win32'
 | |
|         self.assertEquals(get_platform(), 'win32')
 | |
| 
 | |
|         # windows XP, amd64
 | |
|         os.name = 'nt'
 | |
|         sys.version = ('2.4.4 (#71, Oct 18 2006, 08:34:43) '
 | |
|                        '[MSC v.1310 32 bit (Amd64)]')
 | |
|         sys.platform = 'win32'
 | |
|         self.assertEquals(get_platform(), 'win-amd64')
 | |
| 
 | |
|         # windows XP, itanium
 | |
|         os.name = 'nt'
 | |
|         sys.version = ('2.4.4 (#71, Oct 18 2006, 08:34:43) '
 | |
|                        '[MSC v.1310 32 bit (Itanium)]')
 | |
|         sys.platform = 'win32'
 | |
|         self.assertEquals(get_platform(), 'win-ia64')
 | |
| 
 | |
|         # macbook
 | |
|         os.name = 'posix'
 | |
|         sys.version = ('2.5 (r25:51918, Sep 19 2006, 08:49:13) '
 | |
|                        '\n[GCC 4.0.1 (Apple Computer, Inc. build 5341)]')
 | |
|         sys.platform = 'darwin'
 | |
|         self._set_uname(('Darwin', 'macziade', '8.11.1',
 | |
|                    ('Darwin Kernel Version 8.11.1: '
 | |
|                     'Wed Oct 10 18:23:28 PDT 2007; '
 | |
|                     'root:xnu-792.25.20~1/RELEASE_I386'), 'i386'))
 | |
|         os.environ['MACOSX_DEPLOYMENT_TARGET'] = '10.3'
 | |
| 
 | |
|         get_config_vars()['CFLAGS'] = ('-fno-strict-aliasing -DNDEBUG -g '
 | |
|                                        '-fwrapv -O3 -Wall -Wstrict-prototypes')
 | |
| 
 | |
|         self.assertEquals(get_platform(), 'macosx-10.3-i386')
 | |
| 
 | |
|         # macbook with fat binaries (fat, universal or fat64)
 | |
|         os.environ['MACOSX_DEPLOYMENT_TARGET'] = '10.4'
 | |
|         get_config_vars()['CFLAGS'] = ('-arch ppc -arch i386 -isysroot '
 | |
|                                        '/Developer/SDKs/MacOSX10.4u.sdk  '
 | |
|                                        '-fno-strict-aliasing -fno-common '
 | |
|                                        '-dynamic -DNDEBUG -g -O3')
 | |
| 
 | |
|         self.assertEquals(get_platform(), 'macosx-10.4-fat')
 | |
| 
 | |
|         get_config_vars()['CFLAGS'] = ('-arch x86_64 -arch i386 -isysroot '
 | |
|                                        '/Developer/SDKs/MacOSX10.4u.sdk  '
 | |
|                                        '-fno-strict-aliasing -fno-common '
 | |
|                                        '-dynamic -DNDEBUG -g -O3')
 | |
| 
 | |
|         self.assertEquals(get_platform(), 'macosx-10.4-intel')
 | |
| 
 | |
|         get_config_vars()['CFLAGS'] = ('-arch x86_64 -arch ppc -arch i386 -isysroot '
 | |
|                                        '/Developer/SDKs/MacOSX10.4u.sdk  '
 | |
|                                        '-fno-strict-aliasing -fno-common '
 | |
|                                        '-dynamic -DNDEBUG -g -O3')
 | |
|         self.assertEquals(get_platform(), 'macosx-10.4-fat3')
 | |
| 
 | |
|         get_config_vars()['CFLAGS'] = ('-arch ppc64 -arch x86_64 -arch ppc -arch i386 -isysroot '
 | |
|                                        '/Developer/SDKs/MacOSX10.4u.sdk  '
 | |
|                                        '-fno-strict-aliasing -fno-common '
 | |
|                                        '-dynamic -DNDEBUG -g -O3')
 | |
|         self.assertEquals(get_platform(), 'macosx-10.4-universal')
 | |
| 
 | |
|         get_config_vars()['CFLAGS'] = ('-arch x86_64 -arch ppc64 -isysroot '
 | |
|                                        '/Developer/SDKs/MacOSX10.4u.sdk  '
 | |
|                                        '-fno-strict-aliasing -fno-common '
 | |
|                                        '-dynamic -DNDEBUG -g -O3')
 | |
| 
 | |
|         self.assertEquals(get_platform(), 'macosx-10.4-fat64')
 | |
| 
 | |
|         for arch in ('ppc', 'i386', 'x86_64', 'ppc64'):
 | |
|             get_config_vars()['CFLAGS'] = ('-arch %s -isysroot '
 | |
|                                            '/Developer/SDKs/MacOSX10.4u.sdk  '
 | |
|                                            '-fno-strict-aliasing -fno-common '
 | |
|                                            '-dynamic -DNDEBUG -g -O3'%(arch,))
 | |
| 
 | |
|             self.assertEquals(get_platform(), 'macosx-10.4-%s'%(arch,))
 | |
| 
 | |
|         # linux debian sarge
 | |
|         os.name = 'posix'
 | |
|         sys.version = ('2.3.5 (#1, Jul  4 2007, 17:28:59) '
 | |
|                        '\n[GCC 4.1.2 20061115 (prerelease) (Debian 4.1.1-21)]')
 | |
|         sys.platform = 'linux2'
 | |
|         self._set_uname(('Linux', 'aglae', '2.6.21.1dedibox-r7',
 | |
|                     '#1 Mon Apr 30 17:25:38 CEST 2007', 'i686'))
 | |
| 
 | |
|         self.assertEquals(get_platform(), 'linux-i686')
 | |
| 
 | |
|         # XXX more platforms to tests here
 | |
| 
 | |
|     def test_convert_path(self):
 | |
|         # linux/mac
 | |
|         os.sep = '/'
 | |
|         def _join(path):
 | |
|             return '/'.join(path)
 | |
|         os.path.join = _join
 | |
| 
 | |
|         self.assertEquals(convert_path('/home/to/my/stuff'),
 | |
|                           '/home/to/my/stuff')
 | |
| 
 | |
|         # win
 | |
|         os.sep = '\\'
 | |
|         def _join(*path):
 | |
|             return '\\'.join(path)
 | |
|         os.path.join = _join
 | |
| 
 | |
|         self.assertRaises(ValueError, convert_path, '/home/to/my/stuff')
 | |
|         self.assertRaises(ValueError, convert_path, 'home/to/my/stuff/')
 | |
| 
 | |
|         self.assertEquals(convert_path('home/to/my/stuff'),
 | |
|                           'home\\to\\my\\stuff')
 | |
|         self.assertEquals(convert_path('.'),
 | |
|                           os.curdir)
 | |
| 
 | |
|     def test_change_root(self):
 | |
|         # linux/mac
 | |
|         os.name = 'posix'
 | |
|         def _isabs(path):
 | |
|             return path[0] == '/'
 | |
|         os.path.isabs = _isabs
 | |
|         def _join(*path):
 | |
|             return '/'.join(path)
 | |
|         os.path.join = _join
 | |
| 
 | |
|         self.assertEquals(change_root('/root', '/old/its/here'),
 | |
|                           '/root/old/its/here')
 | |
|         self.assertEquals(change_root('/root', 'its/here'),
 | |
|                           '/root/its/here')
 | |
| 
 | |
|         # windows
 | |
|         os.name = 'nt'
 | |
|         def _isabs(path):
 | |
|             return path.startswith('c:\\')
 | |
|         os.path.isabs = _isabs
 | |
|         def _splitdrive(path):
 | |
|             if path.startswith('c:'):
 | |
|                 return ('', path.replace('c:', ''))
 | |
|             return ('', path)
 | |
|         os.path.splitdrive = _splitdrive
 | |
|         def _join(*path):
 | |
|             return '\\'.join(path)
 | |
|         os.path.join = _join
 | |
| 
 | |
|         self.assertEquals(change_root('c:\\root', 'c:\\old\\its\\here'),
 | |
|                           'c:\\root\\old\\its\\here')
 | |
|         self.assertEquals(change_root('c:\\root', 'its\\here'),
 | |
|                           'c:\\root\\its\\here')
 | |
| 
 | |
|         # BugsBunny os (it's a great os)
 | |
|         os.name = 'BugsBunny'
 | |
|         self.assertRaises(DistutilsPlatformError,
 | |
|                           change_root, 'c:\\root', 'its\\here')
 | |
| 
 | |
|         # XXX platforms to be covered: os2, mac
 | |
| 
 | |
|     def test_check_environ(self):
 | |
|         util._environ_checked = 0
 | |
|         if 'HOME' in os.environ:
 | |
|             del os.environ['HOME']
 | |
| 
 | |
|         # posix without HOME
 | |
|         if os.name == 'posix':  # this test won't run on windows
 | |
|             check_environ()
 | |
|             import pwd
 | |
|             self.assertEquals(os.environ['HOME'], pwd.getpwuid(os.getuid())[5])
 | |
|         else:
 | |
|             check_environ()
 | |
| 
 | |
|         self.assertEquals(os.environ['PLAT'], get_platform())
 | |
|         self.assertEquals(util._environ_checked, 1)
 | |
| 
 | |
|     def test_split_quoted(self):
 | |
|         self.assertEquals(split_quoted('""one"" "two" \'three\' \\four'),
 | |
|                           ['one', 'two', 'three', 'four'])
 | |
| 
 | |
|     def test_strtobool(self):
 | |
|         yes = ('y', 'Y', 'yes', 'True', 't', 'true', 'True', 'On', 'on', '1')
 | |
|         no = ('n', 'no', 'f', 'false', 'off', '0', 'Off', 'No', 'N')
 | |
| 
 | |
|         for y in yes:
 | |
|             self.assertTrue(strtobool(y))
 | |
| 
 | |
|         for n in no:
 | |
|             self.assertTrue(not strtobool(n))
 | |
| 
 | |
|     def test_rfc822_escape(self):
 | |
|         header = 'I am a\npoor\nlonesome\nheader\n'
 | |
|         res = rfc822_escape(header)
 | |
|         wanted = ('I am a%(8s)spoor%(8s)slonesome%(8s)s'
 | |
|                   'header%(8s)s') % {'8s': '\n'+8*' '}
 | |
|         self.assertEquals(res, wanted)
 | |
| 
 | |
|     def test_dont_write_bytecode(self):
 | |
|         # makes sure byte_compile raise a DistutilsError
 | |
|         # if sys.dont_write_bytecode is True
 | |
|         old_dont_write_bytecode = sys.dont_write_bytecode
 | |
|         sys.dont_write_bytecode = True
 | |
|         try:
 | |
|             self.assertRaises(DistutilsByteCompileError, byte_compile, [])
 | |
|         finally:
 | |
|             sys.dont_write_bytecode = old_dont_write_bytecode
 | |
| 
 | |
| def test_suite():
 | |
|     return unittest.makeSuite(UtilTestCase)
 | |
| 
 | |
| if __name__ == "__main__":
 | |
|     unittest.main(defaultTest="test_suite")
 |