test_cursor.py revision 51c8bac81e103f334f0dafb1bf2d4c335ac3a28b 
1from clang.cindex import CursorKind
2from clang.cindex import TypeKind
3from .util import get_cursor
4from .util import get_tu
5
6kInput = """\
7// FIXME: Find nicer way to drop builtins and other cruft.
8int start_decl;
9
10struct s0 {
11  int a;
12  int b;
13};
14
15struct s1;
16
17void f0(int a0, int a1) {
18  int l0, l1;
19
20  if (a0)
21    return;
22
23  for (;;) {
24    break;
25  }
26}
27"""
28
29def test_get_children():
30    tu = get_tu(kInput)
31
32    # Skip until past start_decl.
33    it = tu.cursor.get_children()
34    while it.next().spelling != 'start_decl':
35        pass
36
37    tu_nodes = list(it)
38
39    assert len(tu_nodes) == 3
40
41    assert tu_nodes[0] != tu_nodes[1]
42    assert tu_nodes[0].kind == CursorKind.STRUCT_DECL
43    assert tu_nodes[0].spelling == 's0'
44    assert tu_nodes[0].is_definition() == True
45    assert tu_nodes[0].location.file.name == 't.c'
46    assert tu_nodes[0].location.line == 4
47    assert tu_nodes[0].location.column == 8
48    assert tu_nodes[0].hash > 0
49
50    s0_nodes = list(tu_nodes[0].get_children())
51    assert len(s0_nodes) == 2
52    assert s0_nodes[0].kind == CursorKind.FIELD_DECL
53    assert s0_nodes[0].spelling == 'a'
54    assert s0_nodes[0].type.kind == TypeKind.INT
55    assert s0_nodes[1].kind == CursorKind.FIELD_DECL
56    assert s0_nodes[1].spelling == 'b'
57    assert s0_nodes[1].type.kind == TypeKind.INT
58
59    assert tu_nodes[1].kind == CursorKind.STRUCT_DECL
60    assert tu_nodes[1].spelling == 's1'
61    assert tu_nodes[1].displayname == 's1'
62    assert tu_nodes[1].is_definition() == False
63
64    assert tu_nodes[2].kind == CursorKind.FUNCTION_DECL
65    assert tu_nodes[2].spelling == 'f0'
66    assert tu_nodes[2].displayname == 'f0(int, int)'
67    assert tu_nodes[2].is_definition() == True
68
69def test_underlying_type():
70    tu = get_tu('typedef int foo;')
71    typedef = get_cursor(tu, 'foo')
72    assert typedef is not None
73
74    assert typedef.kind.is_declaration()
75    underlying = typedef.underlying_typedef_type
76    assert underlying.kind == TypeKind.INT
77
78def test_enum_type():
79    tu = get_tu('enum TEST { FOO=1, BAR=2 };')
80    enum = get_cursor(tu, 'TEST')
81    assert enum is not None
82
83    assert enum.kind == CursorKind.ENUM_DECL
84    enum_type = enum.enum_type
85    assert enum_type.kind == TypeKind.UINT
86
87def test_enum_type_cpp():
88    tu = get_tu('enum TEST : long long { FOO=1, BAR=2 };', lang="cpp")
89    enum = get_cursor(tu, 'TEST')
90    assert enum is not None
91
92    assert enum.kind == CursorKind.ENUM_DECL
93    assert enum.enum_type.kind == TypeKind.LONGLONG
94
95def test_objc_type_encoding():
96    tu = get_tu('int i;', lang='objc')
97    i = get_cursor(tu, 'i')
98
99    assert i is not None
100    assert i.objc_type_encoding == 'i'
101