Componentize Unit Test for gcm_profile_service

components/gcm_driver/gcm_profile_service_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/gcm_driver/gcm_profile_service.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/callback.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/message_loop/message_loop.h"
+#include "base/prefs/testing_pref_service.h"
 #include "base/run_loop.h"
-#include "chrome/browser/services/gcm/gcm_profile_service_factory.h"
-#include "chrome/browser/signin/profile_oauth2_token_service_factory.h"
-#include "chrome/browser/signin/signin_manager_factory.h"
-#include "chrome/browser/ui/webui/signin/login_ui_service_factory.h"
-#include "chrome/common/channel_info.h"
-#include "chrome/test/base/testing_profile.h"
+#include "base/single_thread_task_runner.h"
+#include "base/test/test_simple_task_runner.h"
 #if defined(OS_CHROMEOS)
 #include "chromeos/dbus/dbus_thread_manager.h"
 #endif
 #include "components/gcm_driver/fake_gcm_app_handler.h"
 #include "components/gcm_driver/fake_gcm_client.h"
 #include "components/gcm_driver/fake_gcm_client_factory.h"
+#include "components/gcm_driver/gcm_channel_status_syncer.h"
 #include "components/gcm_driver/gcm_client.h"
 #include "components/gcm_driver/gcm_client_factory.h"
 #include "components/gcm_driver/gcm_driver.h"
 #include "components/pref_registry/pref_registry_syncable.h"
+#include "components/signin/core/browser/fake_profile_oauth2_token_service.h"
+#include "components/signin/core/browser/fake_signin_manager.h"
 #include "components/signin/core/browser/signin_manager.h"
-#include "content/public/browser/browser_context.h"
-#include "content/public/browser/browser_thread.h"
-#include "content/public/test/test_browser_thread_bundle.h"
+#include "components/signin/core/browser/test_signin_client.h"
+#include "components/signin/core/common/signin_pref_names.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace gcm {
 
 namespace {
 
 const char kTestAppID[] = "TestApp";
 const char kUserID[] = "user";
 
-scoped_ptr<KeyedService> BuildGCMProfileService(
-    content::BrowserContext* context) {
-  Profile* profile = Profile::FromBrowserContext(context);
-  base::SequencedWorkerPool* worker_pool =
-      content::BrowserThread::GetBlockingPool();
-  scoped_refptr<base::SequencedTaskRunner> blocking_task_runner(
-      worker_pool->GetSequencedTaskRunnerWithShutdownBehavior(
-          worker_pool->GetSequenceToken(),
-          base::SequencedWorkerPool::SKIP_ON_SHUTDOWN));
-  return make_scoped_ptr(new gcm::GCMProfileService(
-      profile->GetPrefs(), profile->GetPath(), profile->GetRequestContext(),
-      chrome::GetChannel(),
-      scoped_ptr<ProfileIdentityProvider>(new ProfileIdentityProvider(
-          SigninManagerFactory::GetForProfile(profile),
-          ProfileOAuth2TokenServiceFactory::GetForProfile(profile),
-          LoginUIServiceFactory::GetShowLoginPopupCallbackForProfile(profile))),
-      scoped_ptr<gcm::GCMClientFactory>(new gcm::FakeGCMClientFactory(
-          content::BrowserThread::GetMessageLoopProxyForThread(
-              content::BrowserThread::UI),
-          content::BrowserThread::GetMessageLoopProxyForThread(
-              content::BrowserThread::IO))),
-      content::BrowserThread::GetMessageLoopProxyForThread(
-          content::BrowserThread::UI),
-      content::BrowserThread::GetMessageLoopProxyForThread(
-          content::BrowserThread::IO),
-      blocking_task_runner));
-}
-
 }  // namespace
 
 class GCMProfileServiceTest : public testing::Test {
  protected:
   GCMProfileServiceTest();
   ~GCMProfileServiceTest() override;
 
   // testing::Test:
   void SetUp() override;
   void TearDown() override;
@@ skipping 18 matching lines @@
   GCMDriver* driver() const { return gcm_profile_service_->driver(); }
   std::string registration_id() const { return registration_id_; }
   GCMClient::Result registration_result() const { return registration_result_; }
   GCMClient::Result unregistration_result() const {
     return unregistration_result_;
   }
   std::string send_message_id() const { return send_message_id_; }
   GCMClient::Result send_result() const { return send_result_; }
 
  private:
-  content::TestBrowserThreadBundle thread_bundle_;
-  scoped_ptr<TestingProfile> profile_;
+  base::ScopedTempDir temp_dir_;
+  base::MessageLoop message_loop_;
+
+  TestingPrefServiceSimple prefs_;
+  scoped_ptr<FakeSigninManagerBase> signin_manager_;
+  scoped_ptr<ProfileOAuth2TokenService> fake_token_service_;
   GCMProfileService* gcm_profile_service_;
   scoped_ptr<FakeGCMAppHandler> gcm_app_handler_;
 
   std::string registration_id_;
   GCMClient::Result registration_result_;
   GCMClient::Result unregistration_result_;
   std::string send_message_id_;
   GCMClient::Result send_result_;
 
   DISALLOW_COPY_AND_ASSIGN(GCMProfileServiceTest);
 };
 
 GCMProfileServiceTest::GCMProfileServiceTest()
     : gcm_profile_service_(NULL),
       gcm_app_handler_(new FakeGCMAppHandler),
       registration_result_(GCMClient::UNKNOWN_ERROR),
-      send_result_(GCMClient::UNKNOWN_ERROR) {
-}
+      send_result_(GCMClient::UNKNOWN_ERROR) {}
 
-GCMProfileServiceTest::~GCMProfileServiceTest() {
-}
+GCMProfileServiceTest::~GCMProfileServiceTest() {}
 
 FakeGCMClient* GCMProfileServiceTest::GetGCMClient() const {
   return static_cast<FakeGCMClient*>(
       gcm_profile_service_->driver()->GetGCMClientForTesting());
 }
 
 void GCMProfileServiceTest::SetUp() {
+  GCMChannelStatusSyncer::RegisterPrefs(prefs_.registry());
+
+  prefs_.registry()->RegisterListPref(AccountTrackerService::kAccountInfoPref);
+  prefs_.registry()->RegisterIntegerPref(
+      ::prefs::kAccountIdMigrationState,
+      AccountTrackerService::MIGRATION_NOT_STARTED);

[droger, 2015/12/17 13:23:22]

These prefs may no longer be necessary, please check.

[Jitu( very slow this week), 2015/12/17 14:02:01]

Done.

+
+  ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+  signin_manager_.reset(new FakeSigninManagerBase(nullptr, nullptr));
+  fake_token_service_.reset(new FakeProfileOAuth2TokenService());
+
 #if defined(OS_CHROMEOS)
   // Create a DBus thread manager setter for its side effect.
   // Ignore the return value.
   chromeos::DBusThreadManager::GetSetterForTesting();
 #endif
-  TestingProfile::Builder builder;
-  profile_ = builder.Build();
 }
 
 void GCMProfileServiceTest::TearDown() {
   gcm_profile_service_->driver()->RemoveAppHandler(kTestAppID);
 }
 
 void GCMProfileServiceTest::CreateGCMProfileService() {
-  gcm_profile_service_ = static_cast<GCMProfileService*>(
-      GCMProfileServiceFactory::GetInstance()->SetTestingFactoryAndUse(
-          profile_.get(),
-          &BuildGCMProfileService));
-  gcm_profile_service_->driver()->AddAppHandler(
-      kTestAppID, gcm_app_handler_.get());
+  scoped_refptr<base::SingleThreadTaskRunner> task_runner =
+    message_loop_.task_runner();
+  net::URLRequestContextGetter* request_context =

[droger, 2015/12/17 13:23:22]

Nit:
it would be more correct to use a scoped_refptr here, to ensure that it does not
leak.

scoped_refptr<net::URLRequestContextGetter> request_context =

[Jitu( very slow this week), 2015/12/17 14:02:01]

Done.

+      new net::TestURLRequestContextGetter(task_runner);
+  gcm_profile_service_ = new gcm::GCMProfileService(
+      &prefs_, temp_dir_.path(), request_context,
+      version_info::Channel::UNKNOWN,
+      scoped_ptr<ProfileIdentityProvider>(
+          new ProfileIdentityProvider(signin_manager_.get(),
+                                      fake_token_service_.get(),
+                                      base::Closure()))
+          .Pass(),
+      scoped_ptr<gcm::GCMClientFactory>(
+          new gcm::FakeGCMClientFactory(base::ThreadTaskRunnerHandle::Get(),
+                                        task_runner))
+          .Pass(),
+      base::ThreadTaskRunnerHandle::Get(), task_runner,
+      task_runner);
+  gcm_profile_service_->driver()->AddAppHandler(kTestAppID,
+                                                gcm_app_handler_.get());
 }
 
 void GCMProfileServiceTest::RegisterAndWaitForCompletion(
     const std::vector<std::string>& sender_ids) {
   base::RunLoop run_loop;
   gcm_profile_service_->driver()->Register(
       kTestAppID,
       sender_ids,
       base::Bind(&GCMProfileServiceTest::RegisterCompleted,
                  base::Unretained(this),
@@ skipping 71 matching lines @@
   OutgoingMessage message;
   message.id = "1";
   message.data["key1"] = "value1";
   SendAndWaitForCompletion(message);
 
   EXPECT_EQ(message.id, send_message_id());
   EXPECT_EQ(GCMClient::SUCCESS, send_result());
 }
 
 }  // namespace gcm