Lib/distutils/tests/test_unixccompiler.py

   1 """Tests for distutils.unixccompiler."""
   2 import sys
   3 import unittest
   4
   5 from distutils import sysconfig
   6 from distutils.unixccompiler import UnixCCompiler
   7
   8 class UnixCCompilerTestCase(unittest.TestCase):
   9
  10     def setUp(self):
  11         self._backup_platform = sys.platform
  12         self._backup_get_config_var = sysconfig.get_config_var
  13         class CompilerWrapper(UnixCCompiler):
  14             def rpath_foo(self):
  15                 return self.runtime_library_dir_option('/foo')
  16         self.cc = CompilerWrapper()
  17
  18     def tearDown(self):
  19         sys.platform = self._backup_platform
  20         sysconfig.get_config_var = self._backup_get_config_var
  21
  22     def test_runtime_libdir_option(self):
  23
  24         # not tested under windows
  25         if sys.platform == 'win32':
  26             return
  27
  28         # Issue#5900
  29         #
  30         # Ensure RUNPATH is added to extension modules with RPATH if
  31         # GNU ld is used
  32
  33         # darwin
  34         sys.platform = 'darwin'
  35         self.assertEqual(self.cc.rpath_foo(), '-L/foo')
  36
  37         # hp-ux
  38         sys.platform = 'hp-ux'
  39         old_gcv = sysconfig.get_config_var
  40         def gcv(v):
  41             return 'xxx'
  42         sysconfig.get_config_var = gcv
  43         self.assertEqual(self.cc.rpath_foo(), ['+s', '-L/foo'])
  44
  45         def gcv(v):
  46             return 'gcc'
  47         sysconfig.get_config_var = gcv
  48         self.assertEqual(self.cc.rpath_foo(), ['-Wl,+s', '-L/foo'])
  49
  50         def gcv(v):
  51             return 'g++'
  52         sysconfig.get_config_var = gcv
  53         self.assertEqual(self.cc.rpath_foo(), ['-Wl,+s', '-L/foo'])
  54
  55         sysconfig.get_config_var = old_gcv
  56
  57         # irix646
  58         sys.platform = 'irix646'
  59         self.assertEqual(self.cc.rpath_foo(), ['-rpath', '/foo'])
  60
  61         # osf1V5
  62         sys.platform = 'osf1V5'
  63         self.assertEqual(self.cc.rpath_foo(), ['-rpath', '/foo'])
  64
  65         # GCC GNULD
  66         sys.platform = 'bar'
  67         def gcv(v):
  68             if v == 'CC':
  69                 return 'gcc'
  70             elif v == 'GNULD':
  71                 return 'yes'
  72         sysconfig.get_config_var = gcv
  73         self.assertEqual(self.cc.rpath_foo(), '-Wl,--enable-new-dtags,-R/foo')
  74
  75         # GCC non-GNULD
  76         sys.platform = 'bar'
  77         def gcv(v):
  78             if v == 'CC':
  79                 return 'gcc'
  80             elif v == 'GNULD':
  81                 return 'no'
  82         sysconfig.get_config_var = gcv
  83         self.assertEqual(self.cc.rpath_foo(), '-Wl,-R/foo')
  84
  85         # non-GCC GNULD
  86         sys.platform = 'bar'
  87         def gcv(v):
  88             if v == 'CC':
  89                 return 'cc'
  90             elif v == 'GNULD':
  91                 return 'yes'
  92         sysconfig.get_config_var = gcv
  93         self.assertEqual(self.cc.rpath_foo(), '-R/foo')
  94
  95         # non-GCC non-GNULD
  96         sys.platform = 'bar'
  97         def gcv(v):
  98             if v == 'CC':
  99                 return 'cc'
 100             elif v == 'GNULD':
 101                 return 'no'
 102         sysconfig.get_config_var = gcv
 103         self.assertEqual(self.cc.rpath_foo(), '-R/foo')
 104
 105         # AIX C/C++ linker
 106         sys.platform = 'aix'
 107         def gcv(v):
 108             return 'xxx'
 109         sysconfig.get_config_var = gcv
 110         self.assertEqual(self.cc.rpath_foo(), '-blibpath:/foo')
 111
 112
 113 def test_suite():
 114     return unittest.makeSuite(UnixCCompilerTestCase)
 115
 116 if __name__ == "__main__":
 117     unittest.main(defaultTest="test_suite")