components/gcm_driver/fake_gcm_client.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef COMPONENTS_GCM_DRIVER_FAKE_GCM_CLIENT_H_
   6 #define COMPONENTS_GCM_DRIVER_FAKE_GCM_CLIENT_H_
   7
   8 #include "base/compiler_specific.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "components/gcm_driver/gcm_client.h"
  11
  12 namespace base {
  13 class SequencedTaskRunner;
  14 }
  15
  16 namespace gcm {
  17
  18 class FakeGCMClient : public GCMClient {
  19  public:
  20   enum Status {
  21     UNINITIALIZED,
  22     STARTED,
  23     STOPPED,
  24     CHECKED_OUT
  25   };
  26
  27   enum StartMode {
  28     NO_DELAY_START,
  29     DELAY_START,
  30   };
  31
  32   FakeGCMClient(StartMode start_mode,
  33                 const scoped_refptr<base::SequencedTaskRunner>& ui_thread,
  34                 const scoped_refptr<base::SequencedTaskRunner>& io_thread);
  35   ~FakeGCMClient() override;
  36
  37   // Overridden from GCMClient:
  38   // Called on IO thread.
  39   void Initialize(
  40       const ChromeBuildInfo& chrome_build_info,
  41       const base::FilePath& store_path,
  42       const scoped_refptr<base::SequencedTaskRunner>& blocking_task_runner,
  43       const scoped_refptr<net::URLRequestContextGetter>&
  44           url_request_context_getter,
  45       scoped_ptr<Encryptor> encryptor,
  46       Delegate* delegate) override;
  47   void Start() override;
  48   void Stop() override;
  49   void CheckOut() override;
  50   void Register(const std::string& app_id,
  51                 const std::vector<std::string>& sender_ids) override;
  52   void Unregister(const std::string& app_id) override;
  53   void Send(const std::string& app_id,
  54             const std::string& receiver_id,
  55             const OutgoingMessage& message) override;
  56   void SetRecording(bool recording) override;
  57   void ClearActivityLogs() override;
  58   GCMStatistics GetStatistics() const override;
  59   void SetAccountTokens(
  60       const std::vector<AccountTokenInfo>& account_tokens) override;
  61   void UpdateAccountMapping(const AccountMapping& account_mapping) override;
  62   void RemoveAccountMapping(const std::string& account_id) override;
  63   void SetLastTokenFetchTime(const base::Time& time) override;
  64   void UpdateHeartbeatTimer(scoped_ptr<base::Timer> timer) override;
  65
  66   // Initiate the loading that has been delayed.
  67   // Called on UI thread.
  68   void PerformDelayedLoading();
  69
  70   // Simulate receiving something from the server.
  71   // Called on UI thread.
  72   void ReceiveMessage(const std::string& app_id,
  73                       const IncomingMessage& message);
  74   void DeleteMessages(const std::string& app_id);
  75
  76   std::string GetRegistrationIdFromSenderIds(
  77       const std::vector<std::string>& sender_ids) const;
  78
  79   Status status() const { return status_; }
  80
  81  private:
  82   // Called on IO thread.
  83   void DoLoading();
  84   void CheckinFinished();
  85   void RegisterFinished(const std::string& app_id,
  86                         const std::string& registrion_id);
  87   void UnregisterFinished(const std::string& app_id);
  88   void SendFinished(const std::string& app_id, const OutgoingMessage& message);
  89   void MessageReceived(const std::string& app_id,
  90                        const IncomingMessage& message);
  91   void MessagesDeleted(const std::string& app_id);
  92   void MessageSendError(const std::string& app_id,
  93                         const SendErrorDetails& send_error_details);
  94   void SendAcknowledgement(const std::string& app_id,
  95                            const std::string& message_id);
  96
  97   Delegate* delegate_;
  98   // Increased at checkout in order to produce a different registration ID
  99   // after checkout and re-checkin.
 100   int sequence_id_;
 101   Status status_;
 102   StartMode start_mode_;
 103   scoped_refptr<base::SequencedTaskRunner> ui_thread_;
 104   scoped_refptr<base::SequencedTaskRunner> io_thread_;
 105   base::WeakPtrFactory<FakeGCMClient> weak_ptr_factory_;
 106
 107   DISALLOW_COPY_AND_ASSIGN(FakeGCMClient);
 108 };
 109
 110 }  // namespace gcm
 111
 112 #endif  // COMPONENTS_GCM_DRIVER_FAKE_GCM_CLIENT_H_