Tarek Ziadé | 9df8ce3 | 2008-12-30 23:09:20 +0000 | [diff] [blame] | 1 | """Tests for distutils.msvc9compiler.""" |
| 2 | import sys |
| 3 | import unittest |
Tarek Ziadé | c60ea32 | 2009-12-21 23:12:41 +0000 | [diff] [blame] | 4 | import os |
Tarek Ziadé | 9df8ce3 | 2008-12-30 23:09:20 +0000 | [diff] [blame] | 5 | |
| 6 | from distutils.errors import DistutilsPlatformError |
Tarek Ziadé | c60ea32 | 2009-12-21 23:12:41 +0000 | [diff] [blame] | 7 | from distutils.tests import support |
Éric Araujo | 70ec44a | 2010-11-06 02:44:43 +0000 | [diff] [blame] | 8 | from test.support import run_unittest |
Tarek Ziadé | 9df8ce3 | 2008-12-30 23:09:20 +0000 | [diff] [blame] | 9 | |
Tarek Ziadé | c60ea32 | 2009-12-21 23:12:41 +0000 | [diff] [blame] | 10 | _MANIFEST = """\ |
| 11 | <?xml version="1.0" encoding="UTF-8" standalone="yes"?> |
| 12 | <assembly xmlns="urn:schemas-microsoft-com:asm.v1" |
| 13 | manifestVersion="1.0"> |
| 14 | <trustInfo xmlns="urn:schemas-microsoft-com:asm.v3"> |
| 15 | <security> |
| 16 | <requestedPrivileges> |
| 17 | <requestedExecutionLevel level="asInvoker" uiAccess="false"> |
| 18 | </requestedExecutionLevel> |
| 19 | </requestedPrivileges> |
| 20 | </security> |
| 21 | </trustInfo> |
| 22 | <dependency> |
| 23 | <dependentAssembly> |
| 24 | <assemblyIdentity type="win32" name="Microsoft.VC90.CRT" |
| 25 | version="9.0.21022.8" processorArchitecture="x86" |
| 26 | publicKeyToken="XXXX"> |
| 27 | </assemblyIdentity> |
| 28 | </dependentAssembly> |
| 29 | </dependency> |
| 30 | <dependency> |
| 31 | <dependentAssembly> |
| 32 | <assemblyIdentity type="win32" name="Microsoft.VC90.MFC" |
| 33 | version="9.0.21022.8" processorArchitecture="x86" |
| 34 | publicKeyToken="XXXX"></assemblyIdentity> |
| 35 | </dependentAssembly> |
| 36 | </dependency> |
| 37 | </assembly> |
| 38 | """ |
| 39 | |
| 40 | _CLEANED_MANIFEST = """\ |
| 41 | <?xml version="1.0" encoding="UTF-8" standalone="yes"?> |
| 42 | <assembly xmlns="urn:schemas-microsoft-com:asm.v1" |
| 43 | manifestVersion="1.0"> |
| 44 | <trustInfo xmlns="urn:schemas-microsoft-com:asm.v3"> |
| 45 | <security> |
| 46 | <requestedPrivileges> |
| 47 | <requestedExecutionLevel level="asInvoker" uiAccess="false"> |
| 48 | </requestedExecutionLevel> |
| 49 | </requestedPrivileges> |
| 50 | </security> |
| 51 | </trustInfo> |
| 52 | <dependency> |
| 53 | |
| 54 | </dependency> |
| 55 | <dependency> |
| 56 | <dependentAssembly> |
| 57 | <assemblyIdentity type="win32" name="Microsoft.VC90.MFC" |
| 58 | version="9.0.21022.8" processorArchitecture="x86" |
| 59 | publicKeyToken="XXXX"></assemblyIdentity> |
| 60 | </dependentAssembly> |
| 61 | </dependency> |
| 62 | </assembly>""" |
| 63 | |
Hirokazu Yamamoto | 68201bd | 2010-09-13 05:36:21 +0000 | [diff] [blame] | 64 | if sys.platform=="win32": |
| 65 | from distutils.msvccompiler import get_build_version |
Hirokazu Yamamoto | bea8ae7 | 2010-09-13 07:18:30 +0000 | [diff] [blame] | 66 | if get_build_version()>=8.0: |
| 67 | SKIP_MESSAGE = None |
| 68 | else: |
| 69 | SKIP_MESSAGE = "These tests are only for MSVC8.0 or above" |
| 70 | else: |
| 71 | SKIP_MESSAGE = "These tests are only for win32" |
Hirokazu Yamamoto | 68201bd | 2010-09-13 05:36:21 +0000 | [diff] [blame] | 72 | |
Hirokazu Yamamoto | bea8ae7 | 2010-09-13 07:18:30 +0000 | [diff] [blame] | 73 | @unittest.skipUnless(SKIP_MESSAGE is None, SKIP_MESSAGE) |
Tarek Ziadé | c60ea32 | 2009-12-21 23:12:41 +0000 | [diff] [blame] | 74 | class msvc9compilerTestCase(support.TempdirManager, |
| 75 | unittest.TestCase): |
Tarek Ziadé | 9df8ce3 | 2008-12-30 23:09:20 +0000 | [diff] [blame] | 76 | |
| 77 | def test_no_compiler(self): |
| 78 | # makes sure query_vcvarsall throws |
| 79 | # a DistutilsPlatformError if the compiler |
| 80 | # is not found |
Tarek Ziadé | 9df8ce3 | 2008-12-30 23:09:20 +0000 | [diff] [blame] | 81 | from distutils.msvc9compiler import query_vcvarsall |
| 82 | def _find_vcvarsall(version): |
| 83 | return None |
| 84 | |
| 85 | from distutils import msvc9compiler |
| 86 | old_find_vcvarsall = msvc9compiler.find_vcvarsall |
| 87 | msvc9compiler.find_vcvarsall = _find_vcvarsall |
| 88 | try: |
| 89 | self.assertRaises(DistutilsPlatformError, query_vcvarsall, |
| 90 | 'wont find this version') |
| 91 | finally: |
| 92 | msvc9compiler.find_vcvarsall = old_find_vcvarsall |
| 93 | |
Tarek Ziadé | 63b64c0 | 2009-03-07 00:51:53 +0000 | [diff] [blame] | 94 | def test_reg_class(self): |
Tarek Ziadé | 63b64c0 | 2009-03-07 00:51:53 +0000 | [diff] [blame] | 95 | from distutils.msvc9compiler import Reg |
| 96 | self.assertRaises(KeyError, Reg.get_value, 'xxx', 'xxx') |
| 97 | |
| 98 | # looking for values that should exist on all |
| 99 | # windows registeries versions. |
Tarek Ziadé | 11c9d25 | 2009-11-18 09:32:34 +0000 | [diff] [blame] | 100 | path = r'Control Panel\Desktop' |
| 101 | v = Reg.get_value(path, 'dragfullwindows') |
Tarek Ziadé | 26afd48 | 2009-11-19 05:39:00 +0000 | [diff] [blame] | 102 | self.assertTrue(v in ('0', '1', '2')) |
Tarek Ziadé | 63b64c0 | 2009-03-07 00:51:53 +0000 | [diff] [blame] | 103 | |
Georg Brandl | 4ed72ac | 2009-04-01 04:28:33 +0000 | [diff] [blame] | 104 | import winreg |
| 105 | HKCU = winreg.HKEY_CURRENT_USER |
Tarek Ziadé | 63b64c0 | 2009-03-07 00:51:53 +0000 | [diff] [blame] | 106 | keys = Reg.read_keys(HKCU, 'xxxx') |
| 107 | self.assertEquals(keys, None) |
| 108 | |
Tarek Ziadé | 11c9d25 | 2009-11-18 09:32:34 +0000 | [diff] [blame] | 109 | keys = Reg.read_keys(HKCU, r'Control Panel') |
| 110 | self.assertTrue('Desktop' in keys) |
Tarek Ziadé | 63b64c0 | 2009-03-07 00:51:53 +0000 | [diff] [blame] | 111 | |
Tarek Ziadé | c60ea32 | 2009-12-21 23:12:41 +0000 | [diff] [blame] | 112 | def test_remove_visual_c_ref(self): |
| 113 | from distutils.msvc9compiler import MSVCCompiler |
| 114 | tempdir = self.mkdtemp() |
| 115 | manifest = os.path.join(tempdir, 'manifest') |
| 116 | f = open(manifest, 'w') |
Éric Araujo | bee5cef | 2010-11-05 23:51:56 +0000 | [diff] [blame] | 117 | try: |
| 118 | f.write(_MANIFEST) |
| 119 | finally: |
| 120 | f.close() |
Tarek Ziadé | c60ea32 | 2009-12-21 23:12:41 +0000 | [diff] [blame] | 121 | |
| 122 | compiler = MSVCCompiler() |
| 123 | compiler._remove_visual_c_ref(manifest) |
| 124 | |
| 125 | # see what we got |
| 126 | f = open(manifest) |
Éric Araujo | bee5cef | 2010-11-05 23:51:56 +0000 | [diff] [blame] | 127 | try: |
| 128 | # removing trailing spaces |
| 129 | content = '\n'.join([line.rstrip() for line in f.readlines()]) |
| 130 | finally: |
| 131 | f.close() |
Tarek Ziadé | c60ea32 | 2009-12-21 23:12:41 +0000 | [diff] [blame] | 132 | |
| 133 | # makes sure the manifest was properly cleaned |
| 134 | self.assertEquals(content, _CLEANED_MANIFEST) |
| 135 | |
| 136 | |
Tarek Ziadé | 9df8ce3 | 2008-12-30 23:09:20 +0000 | [diff] [blame] | 137 | def test_suite(): |
| 138 | return unittest.makeSuite(msvc9compilerTestCase) |
| 139 | |
| 140 | if __name__ == "__main__": |
Éric Araujo | 70ec44a | 2010-11-06 02:44:43 +0000 | [diff] [blame] | 141 | run_unittest(test_suite()) |