Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(711)

Unified Diff: components/tracing/child_trace_message_filter_browsertest.cc

Issue 1427963002: [tracing] Move memory-infra dumps to dedicated thread (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@memory-infra-names
Patch Set: remove extra reset() Created 5 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « base/trace_event/memory_dump_manager_unittest.cc ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: components/tracing/child_trace_message_filter_browsertest.cc
diff --git a/components/tracing/child_trace_message_filter_browsertest.cc b/components/tracing/child_trace_message_filter_browsertest.cc
index a87e976cf235d641898f84513d6989c5c0e17453..72eb8a1868554c07edc13214862be3404f8f5714 100644
--- a/components/tracing/child_trace_message_filter_browsertest.cc
+++ b/components/tracing/child_trace_message_filter_browsertest.cc
@@ -33,7 +33,7 @@ class MockDumpProvider : public base::trace_event::MemoryDumpProvider {
base::trace_event::ProcessMemoryDump* pmd));
};
-class ChildTracingTest : public content::RenderViewTest {
+class ChildTracingTest : public content::RenderViewTest, public IPC::Listener {
public:
// Used as callback argument for MemoryDumpManager::RequestGlobalDump():
void OnMemoryDumpCallback(uint64 dump_guid, bool status) {
@@ -59,18 +59,24 @@ class ChildTracingTest : public content::RenderViewTest {
callback_call_count_ = 0;
last_callback_dump_guid_ = 0;
last_callback_status_ = false;
+ wait_for_ipc_message_type_ = 0;
callback_ = base::Bind(&ChildTracingTest::OnMemoryDumpCallback,
base::Unretained(this));
task_runner_ = base::ThreadTaskRunnerHandle::Get();
ctmf_ = make_scoped_refptr(new ChildTraceMessageFilter(task_runner_.get()));
render_thread_->AddFilter(ctmf_.get());
+ // Add a filter to the TestSink which allows to WaitForIPCMessage() by
+ // posting a nested RunLoop closure when a given IPC Message is seen.
+ render_thread_->sink().AddFilter(this);
+
// Getting an instance of |ChildMemoryDumpManagerDelegateImpl| calls
// |MemoryDumpManager::Initialize| with the correct delegate.
ChildMemoryDumpManagerDelegateImpl::GetInstance();
}
void TearDown() override {
+ render_thread_->sink().RemoveFilter(this);
RenderViewTest::TearDown();
MemoryDumpManager::GetInstance()->UnregisterDumpProvider(
mock_dump_provider_.get());
@@ -79,6 +85,26 @@ class ChildTracingTest : public content::RenderViewTest {
task_runner_ = nullptr;
}
+ // IPC::Filter implementation.
+ bool OnMessageReceived(const IPC::Message& message) override {
+ if (message.type() == wait_for_ipc_message_type_) {
+ DCHECK(!wait_for_ipc_closure_.is_null());
+ task_runner_->PostTask(FROM_HERE, wait_for_ipc_closure_);
+ }
+ // Always propagate messages to the sink, never consume them here.
+ return false;
+ }
+
+ const IPC::Message* WaitForIPCMessage(uint32_t message_type) {
+ base::RunLoop run_loop;
+ wait_for_ipc_message_type_ = message_type;
+ wait_for_ipc_closure_ = run_loop.QuitClosure();
+ run_loop.Run();
+ wait_for_ipc_message_type_ = 0;
+ wait_for_ipc_closure_.Reset();
+ return render_thread_->sink().GetUniqueMessageMatching(message_type);
+ }
+
// Simulates a synthetic browser -> child (this process) IPC message.
void SimulateSyntheticMessageFromBrowser(const IPC::Message& msg) {
task_runner_->PostTask(
@@ -107,8 +133,8 @@ class ChildTracingTest : public content::RenderViewTest {
{dump_guid, MemoryDumpType::EXPLICITLY_TRIGGERED}));
// Check that a child -> browser response to the local dump request is sent.
- const IPC::Message* msg = render_thread_->sink().GetUniqueMessageMatching(
- TracingHostMsg_ProcessMemoryDumpResponse::ID);
+ const IPC::Message* msg =
+ WaitForIPCMessage(TracingHostMsg_ProcessMemoryDumpResponse::ID);
EXPECT_NE(nullptr, msg);
// Check that the |dump_guid| and the |sucess| fields are properly set.
@@ -138,6 +164,8 @@ class ChildTracingTest : public content::RenderViewTest {
scoped_ptr<MockDumpProvider> mock_dump_provider_;
scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
base::trace_event::MemoryDumpCallback callback_;
+ uint32_t wait_for_ipc_message_type_;
+ base::Closure wait_for_ipc_closure_;
uint32 callback_call_count_;
uint64 last_callback_dump_guid_;
bool last_callback_status_;
« no previous file with comments | « base/trace_event/memory_dump_manager_unittest.cc ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698