PushMessagingBrowserTest for registration success and failure

chrome/browser/services/gcm/fake_gcm_profile_service.cc

 // Copyright 2013 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/fake_gcm_profile_service.h"
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "chrome/browser/profiles/profile.h"
@@ skipping 94 matching lines @@
 
 // static
 KeyedService* FakeGCMProfileService::Build(content::BrowserContext* context) {
   Profile* profile = static_cast<Profile*>(context);
   FakeGCMProfileService* service = new FakeGCMProfileService(profile);
   service->SetDriverForTesting(new CustomFakeGCMDriver(service));
   return service;
 }
 
 FakeGCMProfileService::FakeGCMProfileService(Profile* profile)
-    : collect_(false) {}
+    : collect_(false) {
+  static_cast<PushMessagingServiceImpl*>(push_messaging_service())
+      ->SetProfileForTesting(profile);
+}
 
 FakeGCMProfileService::~FakeGCMProfileService() {}
 
 void FakeGCMProfileService::RegisterFinished(
     const std::string& app_id,
     const std::vector<std::string>& sender_ids) {
   if (collect_) {
     last_registered_app_id_ = app_id;
     last_registered_sender_ids_ = sender_ids;
   }
@@ skipping 30 matching lines @@
       static_cast<CustomFakeGCMDriver*>(driver());
   custom_driver->OnSendFinished(app_id, message.id, GCMClient::SUCCESS);
 }
 
 void FakeGCMProfileService::AddExpectedUnregisterResponse(
     GCMClient::Result result) {
   unregister_responses_.push_back(result);
 }
 
 }  // namespace gcm