| import os |
| |
| from clang.cindex import Config, CursorKind, SourceLocation, SourceRange, TokenKind |
| |
| if "CLANG_LIBRARY_PATH" in os.environ: |
| Config.set_library_path(os.environ["CLANG_LIBRARY_PATH"]) |
| |
| import unittest |
| |
| from .util import get_tu |
| |
| |
| class TestTokens(unittest.TestCase): |
| def test_token_to_cursor(self): |
| """Ensure we can obtain a Cursor from a Token instance.""" |
| tu = get_tu("int i = 5;") |
| r = tu.get_extent("t.c", (0, 9)) |
| tokens = list(tu.get_tokens(extent=r)) |
| |
| self.assertEqual(len(tokens), 4) |
| self.assertEqual(tokens[1].spelling, "i") |
| self.assertEqual(tokens[1].kind, TokenKind.IDENTIFIER) |
| |
| cursor = tokens[1].cursor |
| self.assertEqual(cursor.kind, CursorKind.VAR_DECL) |
| self.assertEqual(tokens[1].cursor, tokens[2].cursor) |
| |
| def test_token_location(self): |
| """Ensure Token.location works.""" |
| |
| tu = get_tu("int foo = 10;") |
| r = tu.get_extent("t.c", (0, 11)) |
| |
| tokens = list(tu.get_tokens(extent=r)) |
| self.assertEqual(len(tokens), 4) |
| |
| loc = tokens[1].location |
| self.assertIsInstance(loc, SourceLocation) |
| self.assertEqual(loc.line, 1) |
| self.assertEqual(loc.column, 5) |
| self.assertEqual(loc.offset, 4) |
| |
| def test_token_extent(self): |
| """Ensure Token.extent works.""" |
| tu = get_tu("int foo = 10;") |
| r = tu.get_extent("t.c", (0, 11)) |
| |
| tokens = list(tu.get_tokens(extent=r)) |
| self.assertEqual(len(tokens), 4) |
| |
| extent = tokens[1].extent |
| self.assertIsInstance(extent, SourceRange) |
| |
| self.assertEqual(extent.start.offset, 4) |
| self.assertEqual(extent.end.offset, 7) |
