1import os
2from clang.cindex import Config
3if 'CLANG_LIBRARY_PATH' in os.environ:
4    Config.set_library_path(os.environ['CLANG_LIBRARY_PATH'])
5
6from clang.cindex import AccessSpecifier
7from clang.cindex import Cursor
8from clang.cindex import TranslationUnit
9
10from .util import get_cursor
11from .util import get_tu
12
13import unittest
14
15
16class TestAccessSpecifiers(unittest.TestCase):
17    def test_access_specifiers(self):
18        """Ensure that C++ access specifiers are available on cursors"""
19
20        tu = get_tu("""
21class test_class {
22public:
23  void public_member_function();
24protected:
25  void protected_member_function();
26private:
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