xref: /llvm-project/lldb/test/API/commands/expression/import-std-module/vector/TestVectorFromStdModule.py (revision 2238dcc39358353cac21df75c3c3286ab20b8f53)
1"""
2Test basic std::vector functionality.
3"""
4
5from lldbsuite.test.decorators import *
6from lldbsuite.test.lldbtest import *
7from lldbsuite.test import lldbutil
8
9
10class TestBasicVector(TestBase):
11    @add_test_categories(["libc++"])
12    @skipIf(compiler=no_match("clang"))
13    @skipIf(bugnumber="rdar://100741983")
14    def test(self):
15        self.build()
16
17        lldbutil.run_to_source_breakpoint(
18            self, "// Set break point at this line.", lldb.SBFileSpec("main.cpp")
19        )
20
21        self.runCmd("settings set target.import-std-module true")
22
23        vector_type = "std::vector<int>"
24        size_type = "size_type"
25        value_type = "value_type"
26        iterator = "iterator"
27        # LLDB's formatter provides us with a artificial 'item' member.
28        iterator_children = [ValueCheck(name="item")]
29        riterator = "reverse_iterator"
30        riterator_children = [
31            ValueCheck(),  # Deprecated __t_ member; no need to check
32            ValueCheck(name="current"),
33        ]
34
35        self.expect_expr(
36            "a",
37            result_type=vector_type,
38            result_children=[
39                ValueCheck(value="3"),
40                ValueCheck(value="1"),
41                ValueCheck(value="2"),
42            ],
43        )
44        self.expect_expr("a.size()", result_type=size_type, result_value="3")
45        front = self.expect_expr("a.front()", result_type=value_type, result_value="3")
46        self.expect_expr("a[1]", result_type=value_type, result_value="1")
47        self.expect_expr("a.back()", result_type=value_type, result_value="2")
48
49        self.expect("expr std::sort(a.begin(), a.end())")
50        self.expect_expr("a.front()", result_type=value_type, result_value="1")
51        self.expect_expr("a[1]", result_type=value_type, result_value="2")
52        self.expect_expr("a.back()", result_type=value_type, result_value="3")
53
54        self.expect("expr std::reverse(a.begin(), a.end())")
55        self.expect_expr("a.front()", result_type=value_type, result_value="3")
56        self.expect_expr("a[1]", result_type=value_type, result_value="2")
57        self.expect_expr("a.back()", result_type=value_type, result_value="1")
58
59        self.expect_expr(
60            "a.begin()", result_type=iterator, result_children=iterator_children
61        )
62        self.expect_expr(
63            "a.rbegin()", result_type=riterator, result_children=riterator_children
64        )
65
66        self.expect_expr("*a.begin()", result_type="int", result_value="3")
67        self.expect_expr("*a.rbegin()", result_type="int", result_value="1")
68
69        self.expect("expr a.pop_back()")
70        self.expect_expr("a.back()", result_type=value_type, result_value="2")
71        self.expect_expr("a.size()", result_type=size_type, result_value="2")
72
73        self.expect_expr("a.at(0)", result_type=value_type, result_value="3")
74
75        self.expect("expr a.push_back(4)")
76        self.expect_expr("a.back()", result_type=value_type, result_value="4")
77        self.expect_expr("a.size()", result_type=size_type, result_value="3")
78
79        self.expect("expr a.emplace_back(5)")
80        self.expect_expr("a.back()", result_type=value_type, result_value="5")
81        self.expect_expr("a.size()", result_type=size_type, result_value="4")
82
83        self.expect_expr(
84            "a",
85            result_children=[
86                ValueCheck(value="3"),
87                ValueCheck(value="2"),
88                ValueCheck(value="4"),
89                ValueCheck(value="5"),
90            ],
91        )
92
93        # Test that the typedef'd vector type can be substituted.
94        self.expect("expr b.emplace_back(6)")
95        self.expect_expr(
96            "b",
97            result_type="vector_long",
98            result_children=[
99                ValueCheck(value="3"),
100                ValueCheck(value="1"),
101                ValueCheck(value="2"),
102                ValueCheck(value="6"),
103            ],
104        )
105