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
|
#include "utils.hpp"
#include <pdb/all.hpp>
#include <boost/filesystem.hpp>
#include <boost/test/unit_test.hpp>
#include <algorithm>
#include <iterator>
#include <string>
#include <unordered_set>
#include <utility>
namespace {
class DbgHelp {
public:
DbgHelp() : dbghelp{pdb::DbgHelp::post_mortem()} { BOOST_TEST_MESSAGE("Initializing DbgHelp"); }
~DbgHelp() { BOOST_TEST_MESSAGE("Cleaning up DbgHelp"); }
const pdb::DbgHelp dbghelp;
private:
DbgHelp(const DbgHelp&) = delete;
DbgHelp& operator=(const DbgHelp&) = delete;
};
template <typename T>
using Set = std::unordered_set<T>;
template <typename T>
Set<T> join(Set<T>&& xs, Set<T>&& ys) {
xs.insert(std::make_move_iterator(ys.begin()), std::make_move_iterator(ys.end()));
return std::move(xs);
}
class DbgHelpWithSymbols : public DbgHelp {
public:
DbgHelpWithSymbols() { load_test_lib_pdb(); }
static const std::string& get_namespace() {
static const std::string name{"test"};
return name;
}
typedef Set<std::string> SymbolList;
static SymbolList expected_functions() { return make_qualified({"foo", "bar", "baz"}); }
static SymbolList expected_variables() { return make_qualified({"var"}); }
static SymbolList expected_symbols() {
return join(expected_functions(), expected_variables());
}
private:
static SymbolList make_qualified(SymbolList&& plain) {
SymbolList qualified;
for (auto&& name : plain) {
qualified.emplace(get_namespace() + "::" + std::move(name));
}
return qualified;
}
void load_test_lib_pdb() {
const auto pdb_path = get_test_lib_pdb_path().string();
BOOST_TEST_MESSAGE("Loading PDB: " << pdb_path);
dbghelp.load_pdb(pdb_path);
}
static boost::filesystem::path get_test_lib_pdb_path() {
return Paths::get().exe_dir / "test_lib.pdb";
}
};
} // namespace
BOOST_FIXTURE_TEST_SUITE(dbghelp_tests, DbgHelpWithSymbols)
BOOST_AUTO_TEST_CASE(enum_symbols) {
// Symbols can be enumerated, and all the expected symbols are there.
// First, enumerate all the symbols:
std::vector<std::string> all_symbols;
{
const auto callback = [&all_symbols](const pdb::SymbolInfo& symbol) {
all_symbols.emplace_back(symbol.get_name());
};
dbghelp.enum_symbols(callback);
}
// Next, check that all the expected symbols are there:
{
const auto expected = expected_symbols();
const auto check = [&all_symbols](const std::string& name) {
return std::find(all_symbols.cbegin(), all_symbols.cend(), name) != all_symbols.cend();
};
for (const auto& name : expected) {
BOOST_TEST(check(name), "Symbol wasn't enumerated: " << name);
}
}
}
BOOST_AUTO_TEST_SUITE_END()
|
