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

google_apis/gcm/engine/unregistration_request_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 <map>
 #include <string>
 #include <vector>
 
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/string_tokenizer.h"
+#include "base/test/test_simple_task_runner.h"
+#include "base/thread_task_runner_handle.h"
 #include "google_apis/gcm/engine/gcm_unregistration_request_handler.h"
 #include "google_apis/gcm/engine/instance_id_delete_token_request_handler.h"
 #include "google_apis/gcm/monitoring/fake_gcm_stats_recorder.h"
 #include "net/base/load_flags.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace gcm {
 
@@ skipping 55 matching lines @@
   int max_retry_count() const { return max_retry_count_; }
   void set_max_retry_count(int max_retry_count) {
     max_retry_count_ = max_retry_count;
   }
 
  protected:
   int max_retry_count_;
   bool callback_called_;
   UnregistrationRequest::Status status_;
   scoped_ptr<UnregistrationRequest> request_;
-  base::MessageLoop message_loop_;
+  scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
+  base::ThreadTaskRunnerHandle task_runner_handle_;
   net::TestURLFetcherFactory url_fetcher_factory_;
   scoped_refptr<net::TestURLRequestContextGetter> url_request_context_getter_;
   FakeGCMStatsRecorder recorder_;
 };
 
 UnregistrationRequestTest::UnregistrationRequestTest()
     : max_retry_count_(kMaxRetries),
       callback_called_(false),
       status_(UnregistrationRequest::UNREGISTRATION_STATUS_COUNT),
+      task_runner_(new base::TestSimpleTaskRunner()),
+      task_runner_handle_(task_runner_),
       url_request_context_getter_(new net::TestURLRequestContextGetter(
-          message_loop_.task_runner())) {}
+          task_runner_)) {}
 
 UnregistrationRequestTest::~UnregistrationRequestTest() {}
 
 void UnregistrationRequestTest::UnregistrationCallback(
     UnregistrationRequest::Status status) {
   callback_called_ = true;
   status_ = status;
 }
 
 void UnregistrationRequestTest::SetResponseStatusAndString(
@@ skipping 394 matching lines @@
   request_->Start();
 
   SetResponseStatusAndString(net::HTTP_OK, "Error=XXX");
   CompleteFetch();
 
   EXPECT_TRUE(callback_called_);
   EXPECT_EQ(UnregistrationRequest::UNKNOWN_ERROR, status_);
 }
 
 }  // namespace gcm