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 TranslationUnit
7
8import unittest
9from .util import skip_if_no_fspath
10from .util import str_to_path
11
12
13class TestCodeCompletion(unittest.TestCase):
14    def check_completion_results(self, cr, expected):
15        self.assertIsNotNone(cr)
16        self.assertEqual(len(cr.diagnostics), 0)
17
18        completions = [str(c) for c in cr.results]
19
20        for c in expected:
21            self.assertIn(c, completions)
22
23    def test_code_complete(self):
24        files = [('fake.c', """
25/// Aaa.
26int test1;
27
28/// Bbb.
29void test2(void);
30
31void f() {
32
33}
34""")]
35
36        tu = TranslationUnit.from_source('fake.c', ['-std=c99'], unsaved_files=files,
37                options=TranslationUnit.PARSE_INCLUDE_BRIEF_COMMENTS_IN_CODE_COMPLETION)
38
39        cr = tu.codeComplete('fake.c', 9, 1, unsaved_files=files, include_brief_comments=True)
40
41        expected = [
42          "{'int', ResultType} | {'test1', TypedText} || Priority: 50 || Availability: Available || Brief comment: Aaa.",
43          "{'void', ResultType} | {'test2', TypedText} | {'(', LeftParen} | {')', RightParen} || Priority: 50 || Availability: Available || Brief comment: Bbb.",
44          "{'return', TypedText} | {';', SemiColon} || Priority: 40 || Availability: Available || Brief comment: None"
45        ]
46        self.check_completion_results(cr, expected)
47
48    @skip_if_no_fspath
49    def test_code_complete_pathlike(self):
50        files = [(str_to_path('fake.c'), """
51/// Aaa.
52int test1;
53
54/// Bbb.
55void test2(void);
56
57void f() {
58
59}
60""")]
61
62        tu = TranslationUnit.from_source(str_to_path('fake.c'), ['-std=c99'], unsaved_files=files,
63                options=TranslationUnit.PARSE_INCLUDE_BRIEF_COMMENTS_IN_CODE_COMPLETION)
64
65        cr = tu.codeComplete(str_to_path('fake.c'), 9, 1, unsaved_files=files, include_brief_comments=True)
66
67        expected = [
68          "{'int', ResultType} | {'test1', TypedText} || Priority: 50 || Availability: Available || Brief comment: Aaa.",
69          "{'void', ResultType} | {'test2', TypedText} | {'(', LeftParen} | {')', RightParen} || Priority: 50 || Availability: Available || Brief comment: Bbb.",
70          "{'return', TypedText} | {';', SemiColon} || Priority: 40 || Availability: Available || Brief comment: None"
71        ]
72        self.check_completion_results(cr, expected)
73
74    def test_code_complete_availability(self):
75        files = [('fake.cpp', """
76class P {
77protected:
78  int member;
79};
80
81class Q : public P {
82public:
83  using P::member;
84};
85
86void f(P x, Q y) {
87  x.; // member is inaccessible
88  y.; // member is accessible
89}
90""")]
91
92        tu = TranslationUnit.from_source('fake.cpp', ['-std=c++98'], unsaved_files=files)
93
94        cr = tu.codeComplete('fake.cpp', 12, 5, unsaved_files=files)
95
96        expected = [
97          "{'const', TypedText} || Priority: 50 || Availability: Available || Brief comment: None",
98          "{'volatile', TypedText} || Priority: 50 || Availability: Available || Brief comment: None",
99          "{'operator', TypedText} || Priority: 40 || Availability: Available || Brief comment: None",
100          "{'P', TypedText} || Priority: 50 || Availability: Available || Brief comment: None",
101          "{'Q', TypedText} || Priority: 50 || Availability: Available || Brief comment: None"
102        ]
103        self.check_completion_results(cr, expected)
104
105        cr = tu.codeComplete('fake.cpp', 13, 5, unsaved_files=files)
106        expected = [
107            "{'P', TypedText} | {'::', Text} || Priority: 75 || Availability: Available || Brief comment: None",
108            "{'P &', ResultType} | {'operator=', TypedText} | {'(', LeftParen} | {'const P &', Placeholder} | {')', RightParen} || Priority: 79 || Availability: Available || Brief comment: None",
109            "{'int', ResultType} | {'member', TypedText} || Priority: 35 || Availability: NotAccessible || Brief comment: None",
110            "{'void', ResultType} | {'~P', TypedText} | {'(', LeftParen} | {')', RightParen} || Priority: 79 || Availability: Available || Brief comment: None"
111        ]
112        self.check_completion_results(cr, expected)
113