Index: chrome/browser/extensions/extension_gcm_app_handler_unittest.cc |
diff --git a/chrome/browser/extensions/extension_gcm_app_handler_unittest.cc b/chrome/browser/extensions/extension_gcm_app_handler_unittest.cc |
index 786f1d9bbf0b0da3e5eb051c07cc69210a6711af..cbfb2949072dac9fe03bad5ab062fbff4f667158 100644 |
--- a/chrome/browser/extensions/extension_gcm_app_handler_unittest.cc |
+++ b/chrome/browser/extensions/extension_gcm_app_handler_unittest.cc |
@@ -23,6 +23,7 @@ |
#include "chrome/browser/services/gcm/fake_gcm_client.h" |
#include "chrome/browser/services/gcm/fake_gcm_client_factory.h" |
#include "chrome/browser/services/gcm/fake_signin_manager.h" |
+#include "chrome/browser/services/gcm/gcm_driver.h" |
#include "chrome/browser/services/gcm/gcm_profile_service.h" |
#include "chrome/browser/services/gcm/gcm_profile_service_factory.h" |
#include "chrome/browser/signin/signin_manager_factory.h" |
@@ -288,7 +289,7 @@ class ExtensionGCMAppHandlerTest : public testing::Test { |
void Register(const std::string& app_id, |
const std::vector<std::string>& sender_ids) { |
- GetGCMProfileService()->Register( |
+ GetGCMDriver()->Register( |
app_id, |
sender_ids, |
base::Bind(&ExtensionGCMAppHandlerTest::RegisterCompleted, |
@@ -301,12 +302,12 @@ class ExtensionGCMAppHandlerTest : public testing::Test { |
waiter_.SignalCompleted(); |
} |
- gcm::GCMProfileService* GetGCMProfileService() const { |
- return gcm::GCMProfileServiceFactory::GetForProfile(profile()); |
+ gcm::GCMDriver* GetGCMDriver() const { |
+ return gcm::GCMProfileServiceFactory::GetForProfile(profile())->driver(); |
} |
bool HasAppHandlers(const std::string& app_id) const { |
- return GetGCMProfileService()->app_handlers().count(app_id); |
+ return GetGCMDriver()->app_handlers().count(app_id); |
} |
Profile* profile() const { return profile_.get(); } |