bindings/python/tests/cindex/test_cursor.py

   1 from clang.cindex import Index, CursorKind, TypeKind
   2
   3 kInput = """\
   4 // FIXME: Find nicer way to drop builtins and other cruft.
   5 int start_decl;
   6
   7 struct s0 {
   8   int a;
   9   int b;
  10 };
  11
  12 struct s1;
  13
  14 void f0(int a0, int a1) {
  15   int l0, l1;
  16
  17   if (a0)
  18     return;
  19
  20   for (;;) {
  21     break;
  22   }
  23 }
  24 """
  25
  26 def test_get_children():
  27     index = Index.create()
  28     tu = index.parse('t.c', unsaved_files = [('t.c',kInput)])
  29
  30     # Skip until past start_decl.
  31     it = tu.cursor.get_children()
  32     while it.next().spelling != 'start_decl':
  33         pass
  34
  35     tu_nodes = list(it)
  36
  37     assert len(tu_nodes) == 3
  38
  39     assert tu_nodes[0].kind == CursorKind.STRUCT_DECL
  40     assert tu_nodes[0].spelling == 's0'
  41     assert tu_nodes[0].is_definition() == True
  42     assert tu_nodes[0].location.file.name == 't.c'
  43     assert tu_nodes[0].location.line == 4
  44     assert tu_nodes[0].location.column == 8
  45
  46     s0_nodes = list(tu_nodes[0].get_children())
  47     assert len(s0_nodes) == 2
  48     assert s0_nodes[0].kind == CursorKind.FIELD_DECL
  49     assert s0_nodes[0].spelling == 'a'
  50     assert s0_nodes[0].type.kind == TypeKind.INT
  51     assert s0_nodes[1].kind == CursorKind.FIELD_DECL
  52     assert s0_nodes[1].spelling == 'b'
  53     assert s0_nodes[1].type.kind == TypeKind.INT
  54
  55     assert tu_nodes[1].kind == CursorKind.STRUCT_DECL
  56     assert tu_nodes[1].spelling == 's1'
  57     assert tu_nodes[1].displayname == 's1'
  58     assert tu_nodes[1].is_definition() == False
  59
  60     assert tu_nodes[2].kind == CursorKind.FUNCTION_DECL
  61     assert tu_nodes[2].spelling == 'f0'
  62     assert tu_nodes[2].displayname == 'f0(int, int)'
  63     assert tu_nodes[2].is_definition() == True