084e43bfbf
Rewrite the tests from using plain 'assert' mixed with some nosetests methods to the standard unittest module layout. Improve the code to use the most canonical assertion methods whenever possible. This has a few major advantages: - the code uses standard methods now, resulting in a reduced number of WTFs whenever someone with basic Python knowledge gets to read it, - completely unnecessary dependency on nosetests is removed since the standard library supplies all that is necessary for the tests to run, - the tests can be run via any test runner, including the one built-in in Python, - the failure output for most of the tests is improved from 'assertion x == y failed' to actually telling the values. Differential Revision: https://reviews.llvm.org/D39763 llvm-svn: 317897
101 lines
3.6 KiB
Python
101 lines
3.6 KiB
Python
from clang.cindex import Cursor
|
|
from clang.cindex import File
|
|
from clang.cindex import SourceLocation
|
|
from clang.cindex import SourceRange
|
|
from .util import get_cursor
|
|
from .util import get_tu
|
|
|
|
import unittest
|
|
|
|
|
|
baseInput="int one;\nint two;\n"
|
|
|
|
|
|
class TestLocation(unittest.TestCase):
|
|
def assert_location(self, loc, line, column, offset):
|
|
self.assertEqual(loc.line, line)
|
|
self.assertEqual(loc.column, column)
|
|
self.assertEqual(loc.offset, offset)
|
|
|
|
def test_location(self):
|
|
tu = get_tu(baseInput)
|
|
one = get_cursor(tu, 'one')
|
|
two = get_cursor(tu, 'two')
|
|
|
|
self.assertIsNotNone(one)
|
|
self.assertIsNotNone(two)
|
|
|
|
self.assert_location(one.location,line=1,column=5,offset=4)
|
|
self.assert_location(two.location,line=2,column=5,offset=13)
|
|
|
|
# adding a linebreak at top should keep columns same
|
|
tu = get_tu('\n' + baseInput)
|
|
one = get_cursor(tu, 'one')
|
|
two = get_cursor(tu, 'two')
|
|
|
|
self.assertIsNotNone(one)
|
|
self.assertIsNotNone(two)
|
|
|
|
self.assert_location(one.location,line=2,column=5,offset=5)
|
|
self.assert_location(two.location,line=3,column=5,offset=14)
|
|
|
|
# adding a space should affect column on first line only
|
|
tu = get_tu(' ' + baseInput)
|
|
one = get_cursor(tu, 'one')
|
|
two = get_cursor(tu, 'two')
|
|
|
|
self.assert_location(one.location,line=1,column=6,offset=5)
|
|
self.assert_location(two.location,line=2,column=5,offset=14)
|
|
|
|
# define the expected location ourselves and see if it matches
|
|
# the returned location
|
|
tu = get_tu(baseInput)
|
|
|
|
file = File.from_name(tu, 't.c')
|
|
location = SourceLocation.from_position(tu, file, 1, 5)
|
|
cursor = Cursor.from_location(tu, location)
|
|
|
|
one = get_cursor(tu, 'one')
|
|
self.assertIsNotNone(one)
|
|
self.assertEqual(one, cursor)
|
|
|
|
# Ensure locations referring to the same entity are equivalent.
|
|
location2 = SourceLocation.from_position(tu, file, 1, 5)
|
|
self.assertEqual(location, location2)
|
|
location3 = SourceLocation.from_position(tu, file, 1, 4)
|
|
self.assertNotEqual(location2, location3)
|
|
|
|
offset_location = SourceLocation.from_offset(tu, file, 5)
|
|
cursor = Cursor.from_location(tu, offset_location)
|
|
verified = False
|
|
for n in [n for n in tu.cursor.get_children() if n.spelling == 'one']:
|
|
self.assertEqual(n, cursor)
|
|
verified = True
|
|
|
|
self.assertTrue(verified)
|
|
|
|
def test_extent(self):
|
|
tu = get_tu(baseInput)
|
|
one = get_cursor(tu, 'one')
|
|
two = get_cursor(tu, 'two')
|
|
|
|
self.assert_location(one.extent.start,line=1,column=1,offset=0)
|
|
self.assert_location(one.extent.end,line=1,column=8,offset=7)
|
|
self.assertEqual(baseInput[one.extent.start.offset:one.extent.end.offset], "int one")
|
|
|
|
self.assert_location(two.extent.start,line=2,column=1,offset=9)
|
|
self.assert_location(two.extent.end,line=2,column=8,offset=16)
|
|
self.assertEqual(baseInput[two.extent.start.offset:two.extent.end.offset], "int two")
|
|
|
|
file = File.from_name(tu, 't.c')
|
|
location1 = SourceLocation.from_position(tu, file, 1, 1)
|
|
location2 = SourceLocation.from_position(tu, file, 1, 8)
|
|
|
|
range1 = SourceRange.from_locations(location1, location2)
|
|
range2 = SourceRange.from_locations(location1, location2)
|
|
self.assertEqual(range1, range2)
|
|
|
|
location3 = SourceLocation.from_position(tu, file, 1, 6)
|
|
range3 = SourceRange.from_locations(location1, location3)
|
|
self.assertNotEqual(range1, range3)
|