Replace MessageLoop::current() with ThreadTaskRunnerHandle::Get() in GCM

google_apis/gcm/engine/heartbeat_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 "google_apis/gcm/engine/heartbeat_manager.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/memory/scoped_ptr.h"
-#include "base/message_loop/message_loop.h"
+#include "base/test/test_simple_task_runner.h"
+#include "base/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "base/timer/timer.h"
 #include "google_apis/gcm/protocol/mcs.pb.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace gcm {
 
 namespace {
 
 mcs_proto::HeartbeatConfig BuildHeartbeatConfig(int interval_ms) {
@@ skipping 37 matching lines @@
  private:
   // Helper functions for verifying heartbeat manager effects.
   void SendHeartbeatClosure();
   void TriggerReconnectClosure(ConnectionFactory::ConnectionResetReason reason);
 
   scoped_ptr<TestHeartbeatManager> manager_;
 
   int heartbeats_sent_;
   int reconnects_triggered_;
 
-  base::MessageLoop message_loop_;
+  scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
+  base::ThreadTaskRunnerHandle task_runner_handle_;
 };
 
 HeartbeatManagerTest::HeartbeatManagerTest()
     : manager_(new TestHeartbeatManager()),
       heartbeats_sent_(0),
-      reconnects_triggered_(0) {
+      reconnects_triggered_(0),
+      task_runner_(new base::TestSimpleTaskRunner()),
+      task_runner_handle_(task_runner_) {
 }
 
 void HeartbeatManagerTest::StartManager() {
   manager_->Start(base::Bind(&HeartbeatManagerTest::SendHeartbeatClosure,
                              base::Unretained(this)),
                   base::Bind(&HeartbeatManagerTest::TriggerReconnectClosure,
                              base::Unretained(this)));
 }
 
 void HeartbeatManagerTest::SendHeartbeatClosure() {
@@ skipping 194 matching lines @@
   manager()->OnHeartbeatAcked();
 
   till_heartbeat = manager()->GetNextHeartbeatTime() - base::TimeTicks::Now();
   EXPECT_LE(till_heartbeat, base::TimeDelta::FromMilliseconds(
       manager()->GetMaxClientHeartbeatIntervalMs()));
 }
 
 }  // namespace
 
 }  // namespace gcm