| 1 | import os |
|---|
| 2 | from clang.cindex import Config |
|---|
| 3 | if 'CLANG_LIBRARY_PATH' in os.environ: |
|---|
| 4 | Config.set_library_path(os.environ['CLANG_LIBRARY_PATH']) |
|---|
| 5 | |
|---|
| 6 | from clang.cindex import AccessSpecifier |
|---|
| 7 | from clang.cindex import Cursor |
|---|
| 8 | from clang.cindex import TranslationUnit |
|---|
| 9 | |
|---|
| 10 | from .util import get_cursor |
|---|
| 11 | from .util import get_tu |
|---|
| 12 | |
|---|
| 13 | import unittest |
|---|
| 14 | |
|---|
| 15 | |
|---|
| 16 | class TestAccessSpecifiers(unittest.TestCase): |
|---|
| 17 | def test_access_specifiers(self): |
|---|
| 18 | """Ensure that C++ access specifiers are available on cursors""" |
|---|
| 19 | |
|---|
| 20 | tu = get_tu(""" |
|---|
| 21 | class test_class { |
|---|
| 22 | public: |
|---|
| 23 | void public_member_function(); |
|---|
| 24 | protected: |
|---|
| 25 | void protected_member_function(); |
|---|
| 26 | private: |
|---|
| 27 | void private_member_function(); |
|---|
| 28 | }; |
|---|
| 29 | """, lang = 'cpp') |
|---|
| 30 | |
|---|
| 31 | test_class = get_cursor(tu, "test_class") |
|---|
| 32 | self.assertEqual(test_class.access_specifier, AccessSpecifier.INVALID) |
|---|
| 33 | |
|---|
| 34 | public = get_cursor(tu.cursor, "public_member_function") |
|---|
| 35 | self.assertEqual(public.access_specifier, AccessSpecifier.PUBLIC) |
|---|
| 36 | |
|---|
| 37 | protected = get_cursor(tu.cursor, "protected_member_function") |
|---|
| 38 | self.assertEqual(protected.access_specifier, AccessSpecifier.PROTECTED) |
|---|
| 39 | |
|---|
| 40 | private = get_cursor(tu.cursor, "private_member_function") |
|---|
| 41 | self.assertEqual(private.access_specifier, AccessSpecifier.PRIVATE) |
|---|
| 42 | |
|---|