Ensure Origin Thread Affinity for BlameContext

components/scheduler/base/task_queue_manager_unittest.cc

 // Copyright 2014 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.
 
 #include "components/scheduler/base/task_queue_manager.h"
 
 #include <stddef.h>
 
 #include <utility>
 
@@ skipping 1876 matching lines @@
   run_loop.Run();
   buffer.Finish();
 
   return base::WrapUnique(
       trace_analyzer::TraceAnalyzer::Create(trace_output.json_output));
 }
 
 TEST_F(TaskQueueManagerTestWithTracing, BlameContextAttribution) {
   using trace_analyzer::Query;
 
-  Initialize(1u);
+  InitializeWithRealMessageLoop(1u);
   TaskQueue* queue = runners_[0].get();
 
   StartTracing();
   {
     base::trace_event::BlameContext blame_context("cat", "name", "type",
                                                   "scope", 0, nullptr);
     blame_context.Initialize();
     queue->SetBlameContext(&blame_context);
     queue->PostTask(FROM_HERE, base::Bind(&NopTask));
-    test_task_runner_->RunUntilIdle();
+    message_loop_->RunUntilIdle();
   }
   StopTracing();
   std::unique_ptr<trace_analyzer::TraceAnalyzer> analyzer =
       CreateTraceAnalyzer();
 
   trace_analyzer::TraceEventVector events;
   Query q = Query::EventPhaseIs(TRACE_EVENT_PHASE_ENTER_CONTEXT) ||
             Query::EventPhaseIs(TRACE_EVENT_PHASE_LEAVE_CONTEXT);
   analyzer->FindEvents(q, &events);
 
   EXPECT_EQ(2u, events.size());
 }
 
 }  // namespace scheduler