68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
|
import os
|
||
|
|
import platform
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
sys.path.insert(0, '..')
|
||
|
|
from pycparser import parse_file, c_ast
|
||
|
|
|
||
|
|
CPPPATH = 'cpp'
|
||
|
|
|
||
|
|
|
||
|
|
# Test successful parsing
|
||
|
|
#
|
||
|
|
class TestParsing(unittest.TestCase):
|
||
|
|
def _find_file(self, name):
|
||
|
|
""" Find a c file by name, taking into account the current dir can be
|
||
|
|
in a couple of typical places
|
||
|
|
"""
|
||
|
|
testdir = os.path.dirname(__file__)
|
||
|
|
name = os.path.join(testdir, 'c_files', name)
|
||
|
|
assert os.path.exists(name)
|
||
|
|
return name
|
||
|
|
|
||
|
|
def test_without_cpp(self):
|
||
|
|
ast = parse_file(self._find_file('example_c_file.c'))
|
||
|
|
self.assertIsInstance(ast, c_ast.FileAST)
|
||
|
|
|
||
|
|
@unittest.skipUnless(platform.system() == 'Linux',
|
||
|
|
'cpp only works on Linux')
|
||
|
|
def test_with_cpp(self):
|
||
|
|
memmgr_path = self._find_file('memmgr.c')
|
||
|
|
c_files_path = os.path.dirname(memmgr_path)
|
||
|
|
ast = parse_file(memmgr_path, use_cpp=True,
|
||
|
|
cpp_path=CPPPATH,
|
||
|
|
cpp_args='-I%s' % c_files_path)
|
||
|
|
self.assertIsInstance(ast, c_ast.FileAST)
|
||
|
|
|
||
|
|
fake_libc = os.path.join(c_files_path, '..', '..',
|
||
|
|
'utils', 'fake_libc_include')
|
||
|
|
ast2 = parse_file(self._find_file('year.c'), use_cpp=True,
|
||
|
|
cpp_path=CPPPATH,
|
||
|
|
cpp_args=[r'-I%s' % fake_libc])
|
||
|
|
|
||
|
|
self.assertIsInstance(ast2, c_ast.FileAST)
|
||
|
|
|
||
|
|
@unittest.skipUnless(platform.system() == 'Linux',
|
||
|
|
'cpp only works on Linux')
|
||
|
|
def test_cpp_funkydir(self):
|
||
|
|
# This test contains Windows specific path escapes
|
||
|
|
if sys.platform != 'win32':
|
||
|
|
return
|
||
|
|
|
||
|
|
c_files_path = os.path.join('tests', 'c_files')
|
||
|
|
ast = parse_file(self._find_file('simplemain.c'), use_cpp=True,
|
||
|
|
cpp_path=CPPPATH, cpp_args='-I%s' % c_files_path)
|
||
|
|
self.assertIsInstance(ast, c_ast.FileAST)
|
||
|
|
|
||
|
|
@unittest.skipUnless(platform.system() == 'Linux',
|
||
|
|
'cpp only works on Linux')
|
||
|
|
def test_no_real_content_after_cpp(self):
|
||
|
|
ast = parse_file(self._find_file('empty.h'), use_cpp=True,
|
||
|
|
cpp_path=CPPPATH)
|
||
|
|
self.assertIsInstance(ast, c_ast.FileAST)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|