121 lines
4.2 KiB
Python
121 lines
4.2 KiB
Python
|
"""Tests for distutils.cygwinccompiler."""
|
||
|
|
import unittest
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
from io import StringIO
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
from distutils import cygwinccompiler
|
||
|
|
from distutils.cygwinccompiler import (CygwinCCompiler, check_config_h,
|
||
|
|
CONFIG_H_OK, CONFIG_H_NOTOK,
|
||
|
|
CONFIG_H_UNCERTAIN, get_versions)
|
||
|
|
from distutils.tests import support
|
||
|
|
|
||
|
|
class FakePopen(object):
|
||
|
|
test_class = None
|
||
|
|
|
||
|
|
def __init__(self, cmd, shell, stdout):
|
||
|
|
self.cmd = cmd.split()[0]
|
||
|
|
exes = self.test_class._exes
|
||
|
|
if self.cmd in exes:
|
||
|
|
self.stdout = StringIO(exes[self.cmd])
|
||
|
|
else:
|
||
|
|
self.stdout = os.popen(cmd, 'r')
|
||
|
|
|
||
|
|
|
||
|
|
class CygwinCCompilerTestCase(support.TempdirManager,
|
||
|
|
unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
super(CygwinCCompilerTestCase, self).setUp()
|
||
|
|
self.version = sys.version
|
||
|
|
self.python_h = os.path.join(self.mkdtemp(), 'python.h')
|
||
|
|
from distutils import sysconfig
|
||
|
|
self.old_get_config_h_filename = sysconfig.get_config_h_filename
|
||
|
|
sysconfig.get_config_h_filename = self._get_config_h_filename
|
||
|
|
self.old_find_executable = cygwinccompiler.find_executable
|
||
|
|
cygwinccompiler.find_executable = self._find_executable
|
||
|
|
self._exes = {}
|
||
|
|
self.old_popen = cygwinccompiler.Popen
|
||
|
|
FakePopen.test_class = self
|
||
|
|
cygwinccompiler.Popen = FakePopen
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
sys.version = self.version
|
||
|
|
from distutils import sysconfig
|
||
|
|
sysconfig.get_config_h_filename = self.old_get_config_h_filename
|
||
|
|
cygwinccompiler.find_executable = self.old_find_executable
|
||
|
|
cygwinccompiler.Popen = self.old_popen
|
||
|
|
super(CygwinCCompilerTestCase, self).tearDown()
|
||
|
|
|
||
|
|
def _get_config_h_filename(self):
|
||
|
|
return self.python_h
|
||
|
|
|
||
|
|
def _find_executable(self, name):
|
||
|
|
if name in self._exes:
|
||
|
|
return name
|
||
|
|
return None
|
||
|
|
|
||
|
|
def test_check_config_h(self):
|
||
|
|
|
||
|
|
# check_config_h looks for "GCC" in sys.version first
|
||
|
|
# returns CONFIG_H_OK if found
|
||
|
|
sys.version = ('2.6.1 (r261:67515, Dec 6 2008, 16:42:21) \n[GCC '
|
||
|
|
'4.0.1 (Apple Computer, Inc. build 5370)]')
|
||
|
|
|
||
|
|
self.assertEquals(check_config_h()[0], CONFIG_H_OK)
|
||
|
|
|
||
|
|
# then it tries to see if it can find "__GNUC__" in pyconfig.h
|
||
|
|
sys.version = 'something without the *CC word'
|
||
|
|
|
||
|
|
# if the file doesn't exist it returns CONFIG_H_UNCERTAIN
|
||
|
|
self.assertEquals(check_config_h()[0], CONFIG_H_UNCERTAIN)
|
||
|
|
|
||
|
|
# if it exists but does not contain __GNUC__, it returns CONFIG_H_NOTOK
|
||
|
|
self.write_file(self.python_h, 'xxx')
|
||
|
|
self.assertEquals(check_config_h()[0], CONFIG_H_NOTOK)
|
||
|
|
|
||
|
|
# and CONFIG_H_OK if __GNUC__ is found
|
||
|
|
self.write_file(self.python_h, 'xxx __GNUC__ xxx')
|
||
|
|
self.assertEquals(check_config_h()[0], CONFIG_H_OK)
|
||
|
|
|
||
|
|
def test_get_versions(self):
|
||
|
|
|
||
|
|
# get_versions calls distutils.spawn.find_executable on
|
||
|
|
# 'gcc', 'ld' and 'dllwrap'
|
||
|
|
self.assertEquals(get_versions(), (None, None, None))
|
||
|
|
|
||
|
|
# Let's fake we have 'gcc' and it returns '3.4.5'
|
||
|
|
self._exes['gcc'] = 'gcc (GCC) 3.4.5 (mingw special)\nFSF'
|
||
|
|
res = get_versions()
|
||
|
|
self.assertEquals(str(res[0]), '3.4.5')
|
||
|
|
|
||
|
|
# and let's see what happens when the version
|
||
|
|
# doesn't match the regular expression
|
||
|
|
# (\d+\.\d+(\.\d+)*)
|
||
|
|
self._exes['gcc'] = 'very strange output'
|
||
|
|
res = get_versions()
|
||
|
|
self.assertEquals(res[0], None)
|
||
|
|
|
||
|
|
# same thing for ld
|
||
|
|
self._exes['ld'] = 'GNU ld version 2.17.50 20060824'
|
||
|
|
res = get_versions()
|
||
|
|
self.assertEquals(str(res[1]), '2.17.50')
|
||
|
|
self._exes['ld'] = '@(#)PROGRAM:ld PROJECT:ld64-77'
|
||
|
|
res = get_versions()
|
||
|
|
self.assertEquals(res[1], None)
|
||
|
|
|
||
|
|
# and dllwrap
|
||
|
|
self._exes['dllwrap'] = 'GNU dllwrap 2.17.50 20060824\nFSF'
|
||
|
|
res = get_versions()
|
||
|
|
self.assertEquals(str(res[2]), '2.17.50')
|
||
|
|
self._exes['dllwrap'] = 'Cheese Wrap'
|
||
|
|
res = get_versions()
|
||
|
|
self.assertEquals(res[2], None)
|
||
|
|
|
||
|
|
def test_suite():
|
||
|
|
return unittest.makeSuite(CygwinCCompilerTestCase)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
test_support.run_unittest(test_suite())
|