Support getting and deleting token for Instance ID.

components/gcm_driver/fake_gcm_client.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/fake_gcm_client.h"
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/message_loop/message_loop.h"
 #include "base/sequenced_task_runner.h"
@@ skipping 56 matching lines @@
       base::Bind(&FakeGCMClient::Started,
                  weak_ptr_factory_.GetWeakPtr()));
 }
 
 void FakeGCMClient::Stop() {
   DCHECK(io_thread_->RunsTasksOnCurrentThread());
   started_ = false;
   delegate_->OnDisconnected();
 }
 
-void FakeGCMClient::Register(const std::string& app_id,
-                             const std::vector<std::string>& sender_ids) {
+void FakeGCMClient::Register(
+    const linked_ptr<RegistrationInfo>& registration_info) {
   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 
-  std::string registration_id = GetRegistrationIdFromSenderIds(sender_ids);
+  GCMRegistrationInfo* gcm_registration_info =
+      GCMRegistrationInfo::FromRegistrationInfo(registration_info.get());
+  DCHECK(gcm_registration_info);
+
+  std::string registration_id = GetRegistrationIdFromSenderIds(
+      gcm_registration_info->sender_ids);
   base::MessageLoop::current()->PostTask(
       FROM_HERE,
       base::Bind(&FakeGCMClient::RegisterFinished,
                  weak_ptr_factory_.GetWeakPtr(),
-                 app_id,
+                 registration_info,
                  registration_id));
 }
 
-void FakeGCMClient::Unregister(const std::string& app_id) {
+void FakeGCMClient::Unregister(
+    const linked_ptr<RegistrationInfo>& registration_info) {
   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 
   base::MessageLoop::current()->PostTask(
       FROM_HERE,
       base::Bind(&FakeGCMClient::UnregisterFinished,
                  weak_ptr_factory_.GetWeakPtr(),
-                 app_id));
+                 registration_info));
 }
 
 void FakeGCMClient::Send(const std::string& app_id,
                          const std::string& receiver_id,
                          const OutgoingMessage& message) {
   DCHECK(io_thread_->RunsTasksOnCurrentThread());
 
   base::MessageLoop::current()->PostTask(
       FROM_HERE,
       base::Bind(&FakeGCMClient::SendFinished,
@@ skipping 23 matching lines @@
 void FakeGCMClient::RemoveAccountMapping(const std::string& account_id) {
 }
 
 void FakeGCMClient::SetLastTokenFetchTime(const base::Time& time) {
 }
 
 void FakeGCMClient::UpdateHeartbeatTimer(scoped_ptr<base::Timer> timer) {
 }
 
 void FakeGCMClient::AddInstanceIDData(const std::string& app_id,
-                                      const std::string& instance_id_data) {
+                                      const std::string& instance_id,
+                                      const std::string& extra_data) {
 }
 
 void FakeGCMClient::RemoveInstanceIDData(const std::string& app_id) {
 }
 
-std::string FakeGCMClient::GetInstanceIDData(const std::string& app_id) {
-  return std::string();
+void FakeGCMClient::GetInstanceIDData(const std::string& app_id,
+                                      std::string* instance_id,
+                                      std::string* extra_data) {
 }
 
 void FakeGCMClient::PerformDelayedStart() {
   DCHECK(ui_thread_->RunsTasksOnCurrentThread());
 
   io_thread_->PostTask(
       FROM_HERE,
       base::Bind(&FakeGCMClient::DoStart, weak_ptr_factory_.GetWeakPtr()));
 }
 
@@ skipping 38 matching lines @@
     }
   }
   return registration_id;
 }
 
 void FakeGCMClient::Started() {
   delegate_->OnGCMReady(std::vector<AccountMapping>(), base::Time());
   delegate_->OnConnected(net::IPEndPoint());
 }
 
-void FakeGCMClient::RegisterFinished(const std::string& app_id,
-                                     const std::string& registrion_id) {
+void FakeGCMClient::RegisterFinished(
+    const linked_ptr<RegistrationInfo>& registration_info,
+    const std::string& registrion_id) {
   delegate_->OnRegisterFinished(
-      app_id, registrion_id, registrion_id.empty() ? SERVER_ERROR : SUCCESS);
+      registration_info,
+      registrion_id,
+      registrion_id.empty() ? SERVER_ERROR : SUCCESS);
 }
 
-void FakeGCMClient::UnregisterFinished(const std::string& app_id) {
-  delegate_->OnUnregisterFinished(app_id, GCMClient::SUCCESS);
+void FakeGCMClient::UnregisterFinished(
+    const linked_ptr<RegistrationInfo>& registration_info) {
+  delegate_->OnUnregisterFinished(registration_info, GCMClient::SUCCESS);
 }
 
 void FakeGCMClient::SendFinished(const std::string& app_id,
                                  const OutgoingMessage& message) {
   delegate_->OnSendFinished(app_id, message.id, SUCCESS);
 
   // Simulate send error if message id contains a hint.
   if (message.id.find("error") != std::string::npos) {
     SendErrorDetails send_error_details;
     send_error_details.message_id = message.id;
@@ skipping 36 matching lines @@
     delegate_->OnMessageSendError(app_id, send_error_details);
 }
 
 void FakeGCMClient::SendAcknowledgement(const std::string& app_id,
                                         const std::string& message_id) {
   if (delegate_)
     delegate_->OnSendAcknowledged(app_id, message_id);
 }
 
 }  // namespace gcm