[GCM] Track connection failures properly

google_apis/gcm/engine/fake_connection_factory.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 GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_FACTORY_H_
 #define GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_FACTORY_H_
 
 #include "base/memory/scoped_ptr.h"
 #include "google_apis/gcm/engine/connection_factory.h"
 
@@ skipping 10 matching lines @@
 
   // ConnectionFactory implementation.
   virtual void Initialize(
       const BuildLoginRequestCallback& request_builder,
       const ConnectionHandler::ProtoReceivedCallback& read_callback,
       const ConnectionHandler::ProtoSentCallback& write_callback) OVERRIDE;
   virtual ConnectionHandler* GetConnectionHandler() const OVERRIDE;
   virtual void Connect() OVERRIDE;
   virtual bool IsEndpointReachable() const OVERRIDE;
   virtual base::TimeTicks NextRetryAttempt() const OVERRIDE;
-  virtual void SignalConnectionReset() OVERRIDE;
+  virtual void SignalConnectionReset(ConnectionResetReason reason) OVERRIDE;
+
+  // Whether a connection reset has been triggered and is yet to run.
+  bool reconnect_pending() const { return reconnect_pending_; }
+
+  // Whether connection resets should be handled immediately or delayed until
+  // release.
+  void set_delay_reconnect(bool should_delay) {
+    delay_reconnect_ = should_delay;
+  }
 
  private:
   scoped_ptr<FakeConnectionHandler> connection_handler_;
 
   BuildLoginRequestCallback request_builder_;
 
+  // Logic for handling connection resets.
+  bool reconnect_pending_;
+  bool delay_reconnect_;
+
   DISALLOW_COPY_AND_ASSIGN(FakeConnectionFactory);
 };
 
 }  // namespace gcm
 
 #endif  // GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_FACTORY_H_