Index: base/trace_event/memory_dump_manager_test_utils.h |
diff --git a/base/trace_event/memory_dump_manager_test_utils.h b/base/trace_event/memory_dump_manager_test_utils.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..bb32f5c6dedb9c7d7c7bbef41da5b37430d4e3e2 |
--- /dev/null |
+++ b/base/trace_event/memory_dump_manager_test_utils.h |
@@ -0,0 +1,46 @@ |
+// Copyright 2017 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#ifndef BASE_TRACE_EVENT_MEMORY_DUMP_MANAGER_TEST_UTILS_H_ |
+#define BASE_TRACE_EVENT_MEMORY_DUMP_MANAGER_TEST_UTILS_H_ |
+ |
+#include "base/bind.h" |
+#include "base/trace_event/memory_dump_manager.h" |
+#include "base/trace_event/memory_dump_request_args.h" |
+ |
+namespace base { |
+namespace trace_event { |
+ |
+// Short-circuit the RequestGlobalDump() calls to CreateProcessDump(). |
+// Rationale: only the in-process logic is required for unittests. |
+void RequestGlobalDumpForInProcessTesting( |
+ const MemoryDumpRequestArgs& args, |
+ const GlobalMemoryDumpCallback& global_callback) { |
+ // Turns a ProcessMemoryDumpCallback into a GlobalMemoryDumpCallback. |
+ auto callback_adapter = [](const GlobalMemoryDumpCallback& global_callback, |
+ uint64_t dump_guid, bool success, |
+ const Optional<MemoryDumpCallbackResult>& result) { |
+ if (!global_callback.is_null()) |
+ global_callback.Run(dump_guid, success); |
+ }; |
+ MemoryDumpManager::GetInstance()->CreateProcessDump( |
+ args, Bind(callback_adapter, global_callback)); |
+}; |
+ |
+// Short circuits the RequestGlobalDumpFunction() to CreateProcessDump(), |
+// effectively allowing to use both in unittests with the same behavior. |
+// Unittests are in-process only and don't require all the multi-process |
+// dump handshaking (which would require bits outside of base). |
+void InitializeMemoryDumpManagerForInProcessTesting(bool is_coordinator) { |
+ MemoryDumpManager* instance = MemoryDumpManager::GetInstance(); |
+ instance->set_dumper_registrations_ignored_for_testing(true); |
+ instance->Initialize(BindRepeating(&RequestGlobalDumpForInProcessTesting), |
+ is_coordinator); |
+ instance->set_dumper_registrations_ignored_for_testing(false); |
+} |
+ |
+} // namespace trace_event |
+} // namespace base |
+ |
+#endif // BASE_TRACE_EVENT_MEMORY_DUMP_MANAGER_TEST_UTILS_H_ |