| // 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 "chrome/browser/services/gcm/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/run_loop.h" |
| #include "chrome/browser/services/gcm/fake_signin_manager.h" |
| #include "chrome/browser/services/gcm/gcm_profile_service_factory.h" |
| #include "chrome/browser/signin/signin_manager_factory.h" |
| #include "chrome/test/base/testing_profile.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_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 "content/public/browser/browser_context.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/test/test_browser_thread_bundle.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace gcm { |
| |
| namespace { |
| |
| const char kTestAccountID[] = "user@example.com"; |
| const char kTestAppID[] = "TestApp"; |
| const char kUserID[] = "user"; |
| |
| KeyedService* BuildGCMProfileService(content::BrowserContext* context) { |
| return new GCMProfileService( |
| Profile::FromBrowserContext(context), |
| scoped_ptr<GCMClientFactory>(new FakeGCMClientFactory( |
| FakeGCMClient::NO_DELAY_START, |
| content::BrowserThread::GetMessageLoopProxyForThread( |
| content::BrowserThread::UI), |
| content::BrowserThread::GetMessageLoopProxyForThread( |
| content::BrowserThread::IO)))); |
| } |
| |
| } // namespace |
| |
| class GCMProfileServiceTest : public testing::Test { |
| protected: |
| GCMProfileServiceTest(); |
| virtual ~GCMProfileServiceTest(); |
| |
| // testing::Test: |
| virtual void SetUp() override; |
| virtual void TearDown() override; |
| |
| FakeGCMClient* GetGCMClient() const; |
| |
| void CreateGCMProfileService(); |
| void SignIn(); |
| void SignOut(); |
| |
| void RegisterAndWaitForCompletion(const std::vector<std::string>& sender_ids); |
| void UnregisterAndWaitForCompletion(); |
| void SendAndWaitForCompletion(const GCMClient::OutgoingMessage& message); |
| |
| void RegisterCompleted(const base::Closure& callback, |
| const std::string& registration_id, |
| GCMClient::Result result); |
| void UnregisterCompleted(const base::Closure& callback, |
| GCMClient::Result result); |
| void SendCompleted(const base::Closure& callback, |
| const std::string& message_id, |
| GCMClient::Result result); |
| |
| 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_; |
| 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) { |
| } |
| |
| GCMProfileServiceTest::~GCMProfileServiceTest() { |
| } |
| |
| FakeGCMClient* GCMProfileServiceTest::GetGCMClient() const { |
| return static_cast<FakeGCMClient*>( |
| gcm_profile_service_->driver()->GetGCMClientForTesting()); |
| } |
| |
| void GCMProfileServiceTest::SetUp() { |
| #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; |
| builder.AddTestingFactory(SigninManagerFactory::GetInstance(), |
| FakeSigninManager::Build); |
| 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()); |
| } |
| |
| void GCMProfileServiceTest::SignIn() { |
| FakeSigninManager* signin_manager = static_cast<FakeSigninManager*>( |
| SigninManagerFactory::GetInstance()->GetForProfile(profile_.get())); |
| signin_manager->SignIn(kTestAccountID); |
| base::RunLoop().RunUntilIdle(); |
| } |
| |
| void GCMProfileServiceTest::SignOut() { |
| FakeSigninManager* signin_manager = static_cast<FakeSigninManager*>( |
| SigninManagerFactory::GetInstance()->GetForProfile(profile_.get())); |
| signin_manager->SignOut(signin_metrics::SIGNOUT_TEST); |
| base::RunLoop().RunUntilIdle(); |
| } |
| |
| 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), |
| run_loop.QuitClosure())); |
| run_loop.Run(); |
| } |
| |
| void GCMProfileServiceTest::UnregisterAndWaitForCompletion() { |
| base::RunLoop run_loop; |
| gcm_profile_service_->driver()->Unregister( |
| kTestAppID, |
| base::Bind(&GCMProfileServiceTest::UnregisterCompleted, |
| base::Unretained(this), |
| run_loop.QuitClosure())); |
| run_loop.Run(); |
| } |
| |
| void GCMProfileServiceTest::SendAndWaitForCompletion( |
| const GCMClient::OutgoingMessage& message) { |
| base::RunLoop run_loop; |
| gcm_profile_service_->driver()->Send( |
| kTestAppID, |
| kUserID, |
| message, |
| base::Bind(&GCMProfileServiceTest::SendCompleted, |
| base::Unretained(this), |
| run_loop.QuitClosure())); |
| run_loop.Run(); |
| } |
| |
| void GCMProfileServiceTest::RegisterCompleted( |
| const base::Closure& callback, |
| const std::string& registration_id, |
| GCMClient::Result result) { |
| registration_id_ = registration_id; |
| registration_result_ = result; |
| callback.Run(); |
| } |
| |
| void GCMProfileServiceTest::UnregisterCompleted( |
| const base::Closure& callback, |
| GCMClient::Result result) { |
| unregistration_result_ = result; |
| callback.Run(); |
| } |
| |
| void GCMProfileServiceTest::SendCompleted( |
| const base::Closure& callback, |
| const std::string& message_id, |
| GCMClient::Result result) { |
| send_message_id_ = message_id; |
| send_result_ = result; |
| callback.Run(); |
| } |
| |
| TEST_F(GCMProfileServiceTest, CreateGCMProfileServiceBeforeSignIn) { |
| CreateGCMProfileService(); |
| EXPECT_FALSE(driver()->IsStarted()); |
| |
| SignIn(); |
| EXPECT_TRUE(driver()->IsStarted()); |
| } |
| |
| TEST_F(GCMProfileServiceTest, CreateGCMProfileServiceAfterSignIn) { |
| SignIn(); |
| // Note that we can't check if GCM is started or not since the |
| // GCMProfileService that hosts the GCMDriver is not created yet. |
| |
| CreateGCMProfileService(); |
| EXPECT_TRUE(driver()->IsStarted()); |
| } |
| |
| TEST_F(GCMProfileServiceTest, SignInAgain) { |
| CreateGCMProfileService(); |
| SignIn(); |
| EXPECT_TRUE(driver()->IsStarted()); |
| |
| SignOut(); |
| EXPECT_FALSE(driver()->IsStarted()); |
| |
| SignIn(); |
| EXPECT_TRUE(driver()->IsStarted()); |
| } |
| |
| TEST_F(GCMProfileServiceTest, RegisterAndUnregister) { |
| CreateGCMProfileService(); |
| SignIn(); |
| |
| std::vector<std::string> sender_ids; |
| sender_ids.push_back("sender"); |
| RegisterAndWaitForCompletion(sender_ids); |
| |
| std::string expected_registration_id = |
| GetGCMClient()->GetRegistrationIdFromSenderIds(sender_ids); |
| EXPECT_EQ(expected_registration_id, registration_id()); |
| EXPECT_EQ(GCMClient::SUCCESS, registration_result()); |
| |
| UnregisterAndWaitForCompletion(); |
| EXPECT_EQ(GCMClient::SUCCESS, unregistration_result()); |
| } |
| |
| TEST_F(GCMProfileServiceTest, Send) { |
| CreateGCMProfileService(); |
| SignIn(); |
| |
| GCMClient::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 |