src/google_apis/gcm/engine/fake_connection_factory.h

   1 // Copyright 2013 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 GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_FACTORY_H_
   6 #define GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_FACTORY_H_
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "google_apis/gcm/engine/connection_factory.h"
  10
  11 namespace gcm {
  12
  13 class FakeConnectionHandler;
  14
  15 // A connection factory that mocks out real connections, using a fake connection
  16 // handler instead.
  17 class FakeConnectionFactory : public ConnectionFactory {
  18  public:
  19   FakeConnectionFactory();
  20   virtual ~FakeConnectionFactory();
  21
  22   // ConnectionFactory implementation.
  23   virtual void Initialize(
  24       const BuildLoginRequestCallback& request_builder,
  25       const ConnectionHandler::ProtoReceivedCallback& read_callback,
  26       const ConnectionHandler::ProtoSentCallback& write_callback) OVERRIDE;
  27   virtual ConnectionHandler* GetConnectionHandler() const OVERRIDE;
  28   virtual void Connect() OVERRIDE;
  29   virtual bool IsEndpointReachable() const OVERRIDE;
  30   virtual base::TimeTicks NextRetryAttempt() const OVERRIDE;
  31   virtual void SignalConnectionReset(ConnectionResetReason reason) OVERRIDE;
  32
  33   // Whether a connection reset has been triggered and is yet to run.
  34   bool reconnect_pending() const { return reconnect_pending_; }
  35
  36   // Whether connection resets should be handled immediately or delayed until
  37   // release.
  38   void set_delay_reconnect(bool should_delay) {
  39     delay_reconnect_ = should_delay;
  40   }
  41
  42  private:
  43   scoped_ptr<FakeConnectionHandler> connection_handler_;
  44
  45   BuildLoginRequestCallback request_builder_;
  46
  47   // Logic for handling connection resets.
  48   bool reconnect_pending_;
  49   bool delay_reconnect_;
  50
  51   DISALLOW_COPY_AND_ASSIGN(FakeConnectionFactory);
  52 };
  53
  54 }  // namespace gcm
  55
  56 #endif  // GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_FACTORY_H_