| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_ | 5 #ifndef GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_ |
| 6 #define GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_ | 6 #define GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_ |
| 7 | 7 |
| 8 #include <list> | 8 #include <list> |
| 9 | 9 |
| 10 #include "google_apis/gcm/base/mcs_message.h" | 10 #include "google_apis/gcm/base/mcs_message.h" |
| 11 #include "google_apis/gcm/engine/connection_handler.h" | 11 #include "google_apis/gcm/engine/connection_handler.h" |
| 12 | 12 |
| 13 namespace gcm { | 13 namespace gcm { |
| 14 | 14 |
| 15 // A fake implementation of a ConnectionHandler that can arbitrarily receive | 15 // A fake implementation of a ConnectionHandler that can arbitrarily receive |
| 16 // messages and verify expectations for outgoing messages. | 16 // messages and verify expectations for outgoing messages. |
| 17 class FakeConnectionHandler : public ConnectionHandler { | 17 class FakeConnectionHandler : public ConnectionHandler { |
| 18 public: | 18 public: |
| 19 FakeConnectionHandler( | 19 FakeConnectionHandler( |
| 20 const ConnectionHandler::ProtoReceivedCallback& read_callback, | 20 const ConnectionHandler::ProtoReceivedCallback& read_callback, |
| 21 const ConnectionHandler::ProtoSentCallback& write_callback); | 21 const ConnectionHandler::ProtoSentCallback& write_callback); |
| 22 virtual ~FakeConnectionHandler(); | 22 ~FakeConnectionHandler() override; |
| 23 | 23 |
| 24 // ConnectionHandler implementation. | 24 // ConnectionHandler implementation. |
| 25 virtual void Init(const mcs_proto::LoginRequest& login_request, | 25 void Init(const mcs_proto::LoginRequest& login_request, |
| 26 net::StreamSocket* socket) override; | 26 net::StreamSocket* socket) override; |
| 27 virtual void Reset() override; | 27 void Reset() override; |
| 28 virtual bool CanSendMessage() const override; | 28 bool CanSendMessage() const override; |
| 29 virtual void SendMessage(const google::protobuf::MessageLite& message) | 29 void SendMessage(const google::protobuf::MessageLite& message) override; |
| 30 override; | |
| 31 | 30 |
| 32 // EXPECT's receipt of |message| via SendMessage(..). | 31 // EXPECT's receipt of |message| via SendMessage(..). |
| 33 void ExpectOutgoingMessage(const MCSMessage& message); | 32 void ExpectOutgoingMessage(const MCSMessage& message); |
| 34 | 33 |
| 35 // Reset the expected outgoing messages. | 34 // Reset the expected outgoing messages. |
| 36 void ResetOutgoingMessageExpectations(); | 35 void ResetOutgoingMessageExpectations(); |
| 37 | 36 |
| 38 // Whether all expected outgoing messages have been received; | 37 // Whether all expected outgoing messages have been received; |
| 39 bool AllOutgoingMessagesReceived() const; | 38 bool AllOutgoingMessagesReceived() const; |
| 40 | 39 |
| (...skipping 25 matching lines...) Expand all Loading... |
| 66 | 65 |
| 67 // Whether a successful login has completed. | 66 // Whether a successful login has completed. |
| 68 bool initialized_; | 67 bool initialized_; |
| 69 | 68 |
| 70 DISALLOW_COPY_AND_ASSIGN(FakeConnectionHandler); | 69 DISALLOW_COPY_AND_ASSIGN(FakeConnectionHandler); |
| 71 }; | 70 }; |
| 72 | 71 |
| 73 } // namespace gcm | 72 } // namespace gcm |
| 74 | 73 |
| 75 #endif // GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_ | 74 #endif // GOOGLE_APIS_GCM_ENGINE_FAKE_CONNECTION_HANDLER_H_ |
| OLD | NEW |