chrome/browser/services/gcm/gcm_client_mock.h

// 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.

#ifndef CHROME_BROWSER_SERVICES_GCM_GCM_CLIENT_MOCK_H_
#define CHROME_BROWSER_SERVICES_GCM_GCM_CLIENT_MOCK_H_

#include <map>

#include "base/compiler_specific.h"
#include "google_apis/gcm/gcm_client.h"

namespace gcm {

class GCMClientMock : public GCMClient {
 public:
  GCMClientMock();
  virtual ~GCMClientMock();

  // Overridden from GCMClient:
  // Called on IO thread.
  virtual void AddUser(const std::string& username,
                       Delegate* delegate) OVERRIDE;
  virtual void Register(const std::string& username,
                        const std::string& app_id,
                        const std::string& cert,
                        const std::vector<std::string>& sender_ids) OVERRIDE;
  virtual void Unregister(const std::string& username,
                          const std::string& app_id) OVERRIDE;
  virtual void Send(const std::string& username,
                    const std::string& app_id,
                    const std::string& receiver_id,
                    const OutgoingMessage& message) OVERRIDE;
  virtual bool IsLoading() const OVERRIDE;

  // Simulate receiving something from the server.
  // Called on UI thread.
  void ReceiveMessage(const std::string& username,
                      const std::string& app_id,
                      const IncomingMessage& message);
  void DeleteMessages(const std::string& username, const std::string& app_id);

  CheckInInfo GetCheckInInfoFromUsername(const std::string& username) const;
  std::string GetRegistrationIdFromSenderIds(
      const std::vector<std::string>& sender_ids) const;

 private:
  Delegate* GetDelegate(const std::string& username) const;

  // Called on IO thread.
  void AddUserFinished(std::string username, CheckInInfo checkin_info);
  void RegisterFinished(std::string username,
                        std::string app_id,
                        std::string registrion_id);
  void SendFinished(std::string username,
                    std::string app_id,
                    std::string message_id);
  void MessageReceived(std::string username,
                       std::string app_id,
                       IncomingMessage message);
  void MessagesDeleted(std::string username, std::string app_id);
  void MessageSendError(std::string username,
                        std::string app_id,
                        std::string message_id);

  std::map<std::string, Delegate*> delegates_;

  DISALLOW_COPY_AND_ASSIGN(GCMClientMock);
};

}  // namespace gcm

#endif  // CHROME_BROWSER_SERVICES_GCM_GCM_CLIENT_MOCK_H_