1 files changed, 74 insertions, 0 deletions
diff --git a/chromium/google_apis/gcm/engine/fake_connection_handler.h b/chromium/google_apis/gcm/engine/fake_connection_handler.h
new file mode 100644
index 00000000000..5356b771086
--- /dev/null
+++ b/chromium/google_apis/gcm/engine/fake_connection_handler.h
@@ -0,0 +1,74 @@
+// 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_HANDLER_H_
+#define GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_
+
+#include <list>
+
+#include "google_apis/gcm/base/mcs_message.h"
+#include "google_apis/gcm/engine/connection_handler.h"
+
+namespace gcm {
+
+// A fake implementation of a ConnectionHandler that can arbitrarily receive
+// messages and verify expectations for outgoing messages.
+class FakeConnectionHandler : public ConnectionHandler {
+ public:
+  FakeConnectionHandler(
+      const ConnectionHandler::ProtoReceivedCallback& read_callback,
+      const ConnectionHandler::ProtoSentCallback& write_callback);
+  virtual ~FakeConnectionHandler();
+
+  // ConnectionHandler implementation.
+  virtual void Init(const mcs_proto::LoginRequest& login_request,
+                    scoped_ptr<net::StreamSocket> socket) OVERRIDE;
+  virtual bool CanSendMessage() const OVERRIDE;
+  virtual void SendMessage(const google::protobuf::MessageLite& message)
+      OVERRIDE;
+
+  // EXPECT's receipt of |message| via SendMessage(..).
+  void ExpectOutgoingMessage(const MCSMessage& message);
+
+  // Reset the expected outgoing messages.
+  void ResetOutgoingMessageExpectations();
+
+  // Whether all expected outgoing messages have been received;
+  bool AllOutgoingMessagesReceived() const;
+
+  // Passes on |message| to |write_callback_|.
+  void ReceiveMessage(const MCSMessage& message);
+
+  // Whether to return an error with the next login response.
+  void set_fail_login(bool fail_login) {
+    fail_login_ = fail_login;
+  }
+
+  // Whether to invoke the write callback on the next send attempt or fake a
+  // connection error instead.
+  void set_fail_send(bool fail_send) {
+    fail_send_ = fail_send;
+  }
+
+ private:
+  ConnectionHandler::ProtoReceivedCallback read_callback_;
+  ConnectionHandler::ProtoSentCallback write_callback_;
+
+  std::list<MCSMessage> expected_outgoing_messages_;
+
+  // Whether to fail the login or not.
+  bool fail_login_;
+
+  // Whether to fail a SendMessage call or not.
+  bool fail_send_;
+
+  // Whether a successful login has completed.
+  bool initialized_;
+
+  DISALLOW_COPY_AND_ASSIGN(FakeConnectionHandler);
+};
+
+}  // namespace gcm
+
+#endif  // GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_