Ensure Origin Thread Affinity for BlameContext

components/scheduler/base/task_queue_manager_unittest.cc

Index: components/scheduler/base/task_queue_manager_unittest.cc
diff --git a/components/scheduler/base/task_queue_manager_unittest.cc b/components/scheduler/base/task_queue_manager_unittest.cc
index 90b8fd8390edf33d521e64ea437f3a3867c3b0b9..588fd1e365010d91cf58bb38dbb7982bc14907c6 100644
--- a/components/scheduler/base/task_queue_manager_unittest.cc
+++ b/components/scheduler/base/task_queue_manager_unittest.cc
@@ -1894,7 +1894,7 @@ TaskQueueManagerTestWithTracing::CreateTraceAnalyzer() {
 TEST_F(TaskQueueManagerTestWithTracing, BlameContextAttribution) {
   using trace_analyzer::Query;
 
-  Initialize(1u);
+  InitializeWithRealMessageLoop(1u);
   TaskQueue* queue = runners_[0].get();
 
   StartTracing();
@@ -1904,7 +1904,7 @@ TEST_F(TaskQueueManagerTestWithTracing, BlameContextAttribution) {
     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 =