forked from standardese/cppast
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathvisitor.cpp
More file actions
106 lines (95 loc) · 3.34 KB
/
Copy pathvisitor.cpp
File metadata and controls
106 lines (95 loc) · 3.34 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
#include <cppast/cpp_entity.hpp>
using namespace cppast;
#include "test_parser.hpp"
#include <iostream>
TEST_CASE("visitor_filtered")
{
auto code = R"(
namespace the_ns {
class foo {
enum inner_enum {};
class bar {};
};
class one {}; class two {}; class three {};
enum quaz {};
}
enum outer {};
)";
cpp_entity_index idx;
auto file = parse(idx, "cpp_class.cpp", code);
unsigned filtered_count = 0;
auto visitor_callback = [&](const cpp_entity&, cppast::visitor_info info) {
if (info.event != cppast::visitor_info::container_entity_exit)
++filtered_count;
return true;
};
constexpr auto all_node_count = 10, enum_count = 3, class_count = 5;
SECTION("all nodes are visited")
{
filtered_count = 0;
cppast::visit(*file, [](const cpp_entity&) { return true; }, visitor_callback);
REQUIRE(filtered_count == all_node_count);
}
SECTION("filtered callback on both enter and exit")
{
filtered_count = 0;
cppast::visit(*file, [](const cpp_entity&) { return true; },
[&](const cpp_entity&, cppast::visitor_info info) {
(void)info;
filtered_count++;
return true;
});
REQUIRE(filtered_count == all_node_count * 2);
}
SECTION("whitelist")
{
SECTION("only one kind whitelisted")
{
filtered_count = 0;
cppast::visit(*file, whitelist<cpp_entity_kind::enum_t>(), visitor_callback);
REQUIRE(filtered_count == enum_count);
}
SECTION("many kinds whitelisted")
{
filtered_count = 0;
cppast::visit(*file, whitelist<cpp_entity_kind::enum_t, cpp_entity_kind::class_t>(),
visitor_callback);
REQUIRE(filtered_count == enum_count + class_count);
}
}
SECTION("blacklist")
{
SECTION("only one kind blacklisted")
{
filtered_count = 0;
cppast::visit(*file, blacklist<cpp_entity_kind::file_t>(), visitor_callback);
REQUIRE(filtered_count == all_node_count - 1);
}
SECTION("many kinds blacklisted")
{
filtered_count = 0;
cppast::visit(*file, blacklist<cpp_entity_kind::enum_t, cpp_entity_kind::class_t>(),
visitor_callback);
REQUIRE(filtered_count == all_node_count - enum_count - class_count);
}
}
SECTION("blacklist_and_children")
{
SECTION("only one kind blacklisted")
{
filtered_count = 0;
cppast::visit(*file, blacklist_and_children<cpp_entity_kind::file_t>(),
visitor_callback);
REQUIRE(filtered_count == 0);
}
SECTION("many kinds blacklisted")
{
filtered_count = 0;
cppast::visit(*file,
blacklist_and_children<cpp_entity_kind::enum_t,
cpp_entity_kind::class_t>(),
visitor_callback);
REQUIRE(filtered_count == all_node_count - enum_count - class_count);
}
}
}