unit_tests: more call stack tests

1 files changed, 79 insertions, 0 deletions

diff --git a/test/unit_tests/call_stack.cpp b/test/unit_tests/call_stack.cpp
new file mode 100644
index 0000000..481ab6e
--- /dev/null
+++ b/test/unit_tests/call_stack.cpp
@@ -0,0 +1,79 @@
+#include "fixtures.hpp"
+
+#include <pdb/all.hpp>
+#include <test_lib.hpp>
+
+#include <boost/test/unit_test.hpp>
+
+#include <algorithm>
+#include <sstream>
+#include <string>
+#include <vector>
+
+BOOST_FIXTURE_TEST_SUITE(call_stack_tests, CurrentProcess)
+
+BOOST_AUTO_TEST_CASE(call_stack) {
+    try {
+        test_ns::throw_call_stack();
+    } catch (const pdb::CallStack& call_stack) {
+        // First, check that the call stack have been caught.
+        BOOST_TEST(true, "Caught the call stack");
+
+        // Debug output:
+        std::vector<std::string> pretty;
+        pretty.reserve(call_stack.length);
+
+        BOOST_TEST_MESSAGE("Call stack:");
+        for (const auto& addr : call_stack) {
+            pretty.emplace_back(call_stack.pretty_print_address(dbghelp, addr));
+            BOOST_TEST_MESSAGE('\t' << pdb::format_address(addr) << ' ' << pretty.back());
+        }
+
+        // Second, resolve the symbols:
+        std::vector<pdb::SymbolInfo> symbols;
+        symbols.reserve(call_stack.length);
+
+        BOOST_TEST_MESSAGE("Resolved symbols:");
+        for (const auto& addr : call_stack) {
+            symbols.emplace_back(dbghelp.resolve_symbol(addr));
+            const auto& symbol = symbols.back();
+            BOOST_TEST_MESSAGE('\t' << pdb::format_address(symbol.get_offline_address()) << ' '
+                                    << symbol.get_name());
+        }
+
+        {
+            // Third, check that the expected function frames are in the call stack.
+            const auto expected = expected_function_addresses();
+
+            const auto check = [&](pdb::Address addr) {
+                for (const auto& symbol : symbols)
+                    if (symbol.get_offline_address() == addr)
+                        return true;
+                return false;
+            };
+            for (const auto& addr : expected) {
+                BOOST_TEST(check(addr), "Function frame captured: " << pdb::format_address(addr));
+            }
+        }
+
+        {
+            // Fourth, check that the expected function names are in the call stack.
+            const auto expected = expected_functions_full();
+
+            const auto check = [&](const std::string& name) {
+                for (const auto& actual : pretty)
+                    if (actual.find(name) != std::string::npos)
+                        return true;
+                return false;
+            };
+            for (const auto& name : expected) {
+                BOOST_TEST(check(name), "Function frame captured: " << name);
+            }
+        }
+        return;
+    }
+    // We must catch the call stack.
+    BOOST_TEST(false, "Didn't catch the call stack");
+}
+
+BOOST_AUTO_TEST_SUITE_END()