TestLibCython.py 4.13 KB
Newer Older
Mark Florisson's avatar
Mark Florisson committed
1 2 3 4 5 6 7 8 9 10 11
import os
import re
import sys
import uuid
import shutil
import textwrap
import unittest
import tempfile
import subprocess
import distutils.core
from distutils import sysconfig
Mark Florisson's avatar
Mark Florisson committed
12
from distutils import ccompiler
Mark Florisson's avatar
Mark Florisson committed
13 14 15 16

import Cython.Distutils.extension
from Cython.Debugger import Cygdb as cygdb

Mark Florisson's avatar
Mark Florisson committed
17 18 19 20 21
root = os.path.dirname(os.path.abspath(__file__))
codefile = os.path.join(root, 'codefile')
cfuncs_file = os.path.join(root, 'cfuncs.c')
with open(codefile) as f:
    source_to_lineno = dict((line.strip(), i + 1) for i, line in enumerate(f))
Mark Florisson's avatar
Mark Florisson committed
22 23 24 25 26 27 28 29 30 31 32 33 34

class DebuggerTestCase(unittest.TestCase):
    
    def setUp(self):
        """
        Run gdb and have cygdb import the debug information from the code
        defined in TestParseTreeTransforms's setUp method
        """
        self.tempdir = tempfile.mkdtemp()
        self.destfile = os.path.join(self.tempdir, 'codefile.pyx')
        self.debug_dest = os.path.join(self.tempdir, 
                                      'cython_debug', 
                                      'cython_debug_info_codefile')
Mark Florisson's avatar
Mark Florisson committed
35
        self.cfuncs_destfile = os.path.join(self.tempdir, 'cfuncs')
Mark Florisson's avatar
Mark Florisson committed
36 37 38 39
        
        self.cwd = os.getcwd()
        os.chdir(self.tempdir)
        
Mark Florisson's avatar
Mark Florisson committed
40 41 42 43 44
        shutil.copy(codefile, self.destfile)
        shutil.copy(cfuncs_file, self.cfuncs_destfile + '.c')
        
        compiler = ccompiler.new_compiler()
        compiler.compile(['cfuncs.c'], debug=True)
Mark Florisson's avatar
Mark Florisson committed
45 46 47 48
        
        ext = Cython.Distutils.extension.Extension(
            'codefile',
            ['codefile.pyx'], 
Mark Florisson's avatar
Mark Florisson committed
49 50
            pyrex_debug=True,
            extra_objects=['cfuncs.o'])
Mark Florisson's avatar
Mark Florisson committed
51 52 53
            
        distutils.core.setup(
            script_args=['build_ext', '--inplace'],
Mark Florisson's avatar
Mark Florisson committed
54
            ext_modules=[ext],
Mark Florisson's avatar
Mark Florisson committed
55 56 57 58 59 60 61 62 63 64 65 66
            cmdclass=dict(build_ext=Cython.Distutils.build_ext)
        )
    
    def tearDown(self):
        os.chdir(self.cwd)
        shutil.rmtree(self.tempdir)


class GdbDebuggerTestCase(DebuggerTestCase):
    def setUp(self):
        super(GdbDebuggerTestCase, self).setUp()
        
Mark Florisson's avatar
Mark Florisson committed
67 68
        prefix_code = textwrap.dedent('''\
            python
Mark Florisson's avatar
Mark Florisson committed
69
                
Mark Florisson's avatar
Mark Florisson committed
70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
            import os
            import sys
            import traceback
            
            def excepthook(type, value, tb):
                traceback.print_exception(type, value, tb)
                os._exit(1)
            
            sys.excepthook = excepthook
            
            # Have tracebacks end up on sys.stderr (gdb replaces sys.stderr
            # with an object that calls gdb.write())
            sys.stderr = sys.__stderr__
            
            end
            ''')
        
        code = textwrap.dedent('''\
            python
            
            from Cython.Debugger.Tests import test_libcython_in_gdb
            test_libcython_in_gdb.main()
            
            end
            ''')
        
        self.gdb_command_file = cygdb.make_command_file(self.tempdir, 
                                                        prefix_code)
        open(self.gdb_command_file, 'a').write(code)
        
Mark Florisson's avatar
Mark Florisson committed
100 101 102 103 104 105 106 107 108 109
        args = ['gdb', '-batch', '-x', self.gdb_command_file, '-n', '--args',
                sys.executable, '-c', 'import codefile']
        
        paths = []
        path = os.environ.get('PYTHONPATH')
        if path:
            paths.append(path)
        paths.append(os.path.dirname(os.path.dirname(
            os.path.abspath(Cython.__file__))))
        env = dict(os.environ, PYTHONPATH=os.pathsep.join(paths))
Mark Florisson's avatar
Mark Florisson committed
110

Mark Florisson's avatar
Mark Florisson committed
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
        self.p = subprocess.Popen(
            args,
            stdout=open(os.devnull, 'w'),
            stderr=subprocess.PIPE,
            env=env)
        
    def tearDown(self):
        super(GdbDebuggerTestCase, self).tearDown()
        self.p.stderr.close()
        self.p.wait()
        os.remove(self.gdb_command_file)
        
   
class TestAll(GdbDebuggerTestCase):
    
    def test_all(self):
        out, err = self.p.communicate()
        border = '*' * 30
        start = '%s   v INSIDE GDB v   %s' % (border, border)
        end   = '%s   ^ INSIDE GDB ^   %s' % (border, border)
        errmsg = '\n%s\n%s%s' % (start, err, end)
        self.assertEquals(0, self.p.wait(), errmsg)
        sys.stderr.write(err)

if __name__ == '__main__':
    unittest.main()