| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 REMOTING_PROTOCOL_MOCK_OBJECTS_H_ | 5 #ifndef REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ |
| 6 #define REMOTING_PROTOCOL_MOCK_OBJECTS_H_ | 6 #define REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ |
| 7 | 7 |
| 8 #include "remoting/proto/internal.pb.h" | 8 #include "remoting/proto/internal.pb.h" |
| 9 #include "remoting/protocol/client_stub.h" | 9 #include "remoting/protocol/client_stub.h" |
| 10 #include "remoting/protocol/connection_to_client.h" | 10 #include "remoting/protocol/connection_to_client.h" |
| 11 #include "remoting/protocol/host_stub.h" | 11 #include "remoting/protocol/host_stub.h" |
| 12 #include "remoting/protocol/input_stub.h" | 12 #include "remoting/protocol/input_stub.h" |
| 13 #include "remoting/protocol/session.h" | 13 #include "remoting/protocol/session.h" |
| 14 #include "remoting/protocol/video_stub.h" | 14 #include "remoting/protocol/video_stub.h" |
| 15 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 16 | 16 |
| 17 namespace remoting { | 17 namespace remoting { |
| 18 namespace protocol { | 18 namespace protocol { |
| 19 | 19 |
| 20 class ChromotocolConnection; | 20 class ChromotocolConnection; |
| 21 | 21 |
| 22 class MockConnectionToClient : public ConnectionToClient { | 22 class MockConnectionToClient : public ConnectionToClient { |
| 23 public: | 23 public: |
| 24 MockConnectionToClient() {} | 24 MockConnectionToClient(); |
| 25 virtual ~MockConnectionToClient(); |
| 25 | 26 |
| 26 MOCK_METHOD1(Init, void(Session* session)); | 27 MOCK_METHOD1(Init, void(Session* session)); |
| 27 MOCK_METHOD0(video_stub, VideoStub*()); | 28 MOCK_METHOD0(video_stub, VideoStub*()); |
| 28 MOCK_METHOD0(client_stub, ClientStub*()); | 29 MOCK_METHOD0(client_stub, ClientStub*()); |
| 29 MOCK_METHOD0(session, Session*()); | 30 MOCK_METHOD0(session, Session*()); |
| 30 MOCK_METHOD0(Disconnect, void()); | 31 MOCK_METHOD0(Disconnect, void()); |
| 31 | 32 |
| 32 private: | 33 private: |
| 33 DISALLOW_COPY_AND_ASSIGN(MockConnectionToClient); | 34 DISALLOW_COPY_AND_ASSIGN(MockConnectionToClient); |
| 34 }; | 35 }; |
| 35 | 36 |
| 36 class MockConnectionToClientEventHandler : | 37 class MockConnectionToClientEventHandler : |
| 37 public ConnectionToClient::EventHandler { | 38 public ConnectionToClient::EventHandler { |
| 38 public: | 39 public: |
| 39 MockConnectionToClientEventHandler() {} | 40 MockConnectionToClientEventHandler(); |
| 41 virtual ~MockConnectionToClientEventHandler(); |
| 40 | 42 |
| 41 MOCK_METHOD1(OnConnectionOpened, void(ConnectionToClient* connection)); | 43 MOCK_METHOD1(OnConnectionOpened, void(ConnectionToClient* connection)); |
| 42 MOCK_METHOD1(OnConnectionClosed, void(ConnectionToClient* connection)); | 44 MOCK_METHOD1(OnConnectionClosed, void(ConnectionToClient* connection)); |
| 43 MOCK_METHOD1(OnConnectionFailed, void(ConnectionToClient* connection)); | 45 MOCK_METHOD1(OnConnectionFailed, void(ConnectionToClient* connection)); |
| 44 | 46 |
| 45 private: | 47 private: |
| 46 DISALLOW_COPY_AND_ASSIGN(MockConnectionToClientEventHandler); | 48 DISALLOW_COPY_AND_ASSIGN(MockConnectionToClientEventHandler); |
| 47 }; | 49 }; |
| 48 | 50 |
| 49 class MockInputStub : public InputStub { | 51 class MockInputStub : public InputStub { |
| 50 public: | 52 public: |
| 51 MockInputStub() {} | 53 MockInputStub(); |
| 54 virtual ~MockInputStub(); |
| 52 | 55 |
| 53 MOCK_METHOD2(InjectKeyEvent, void(const KeyEvent* event, Task* done)); | 56 MOCK_METHOD2(InjectKeyEvent, void(const KeyEvent* event, Task* done)); |
| 54 MOCK_METHOD2(InjectMouseEvent, void(const MouseEvent* event, Task* done)); | 57 MOCK_METHOD2(InjectMouseEvent, void(const MouseEvent* event, Task* done)); |
| 55 | 58 |
| 56 private: | 59 private: |
| 57 DISALLOW_COPY_AND_ASSIGN(MockInputStub); | 60 DISALLOW_COPY_AND_ASSIGN(MockInputStub); |
| 58 }; | 61 }; |
| 59 | 62 |
| 60 class MockHostStub : public HostStub { | 63 class MockHostStub : public HostStub { |
| 61 public: | 64 public: |
| 62 MockHostStub() {} | 65 MockHostStub(); |
| 66 ~MockHostStub(); |
| 63 | 67 |
| 64 MOCK_METHOD2(SuggestResolution, void(const SuggestResolutionRequest* msg, | 68 MOCK_METHOD2(SuggestResolution, void(const SuggestResolutionRequest* msg, |
| 65 Task* done)); | 69 Task* done)); |
| 66 MOCK_METHOD2(BeginSessionRequest, | 70 MOCK_METHOD2(BeginSessionRequest, |
| 67 void(const LocalLoginCredentials* credentials, Task* done)); | 71 void(const LocalLoginCredentials* credentials, Task* done)); |
| 68 | 72 |
| 69 private: | 73 private: |
| 70 DISALLOW_COPY_AND_ASSIGN(MockHostStub); | 74 DISALLOW_COPY_AND_ASSIGN(MockHostStub); |
| 71 }; | 75 }; |
| 72 | 76 |
| 73 class MockClientStub : public ClientStub { | 77 class MockClientStub : public ClientStub { |
| 74 public: | 78 public: |
| 75 MockClientStub() {} | 79 MockClientStub(); |
| 80 virtual ~MockClientStub(); |
| 76 | 81 |
| 77 MOCK_METHOD2(NotifyResolution, void(const NotifyResolutionRequest* msg, | 82 MOCK_METHOD2(NotifyResolution, void(const NotifyResolutionRequest* msg, |
| 78 Task* done)); | 83 Task* done)); |
| 79 MOCK_METHOD2(BeginSessionResponse, void(const LocalLoginStatus* msg, | 84 MOCK_METHOD2(BeginSessionResponse, void(const LocalLoginStatus* msg, |
| 80 Task* done)); | 85 Task* done)); |
| 81 | 86 |
| 82 private: | 87 private: |
| 83 DISALLOW_COPY_AND_ASSIGN(MockClientStub); | 88 DISALLOW_COPY_AND_ASSIGN(MockClientStub); |
| 84 }; | 89 }; |
| 85 | 90 |
| 86 class MockVideoStub : public VideoStub { | 91 class MockVideoStub : public VideoStub { |
| 87 public: | 92 public: |
| 88 MockVideoStub() {} | 93 MockVideoStub(); |
| 94 virtual ~MockVideoStub(); |
| 89 | 95 |
| 90 MOCK_METHOD2(ProcessVideoPacket, void(const VideoPacket* video_packet, | 96 MOCK_METHOD2(ProcessVideoPacket, void(const VideoPacket* video_packet, |
| 91 Task* done)); | 97 Task* done)); |
| 92 MOCK_METHOD0(GetPendingPackets, int()); | 98 MOCK_METHOD0(GetPendingPackets, int()); |
| 93 | 99 |
| 94 private: | 100 private: |
| 95 DISALLOW_COPY_AND_ASSIGN(MockVideoStub); | 101 DISALLOW_COPY_AND_ASSIGN(MockVideoStub); |
| 96 }; | 102 }; |
| 97 | 103 |
| 98 class MockSession : public Session { | 104 class MockSession : public Session { |
| 99 public: | 105 public: |
| 100 MockSession() {} | 106 MockSession(); |
| 107 virtual ~MockSession(); |
| 101 | 108 |
| 102 MOCK_METHOD1(SetStateChangeCallback, void(StateChangeCallback* callback)); | 109 MOCK_METHOD1(SetStateChangeCallback, void(StateChangeCallback* callback)); |
| 103 MOCK_METHOD0(control_channel, net::Socket*()); | 110 MOCK_METHOD0(control_channel, net::Socket*()); |
| 104 MOCK_METHOD0(event_channel, net::Socket*()); | 111 MOCK_METHOD0(event_channel, net::Socket*()); |
| 105 MOCK_METHOD0(video_channel, net::Socket*()); | 112 MOCK_METHOD0(video_channel, net::Socket*()); |
| 106 MOCK_METHOD0(video_rtp_channel, net::Socket*()); | 113 MOCK_METHOD0(video_rtp_channel, net::Socket*()); |
| 107 MOCK_METHOD0(video_rtcp_channel, net::Socket*()); | 114 MOCK_METHOD0(video_rtcp_channel, net::Socket*()); |
| 108 MOCK_METHOD0(jid, const std::string&()); | 115 MOCK_METHOD0(jid, const std::string&()); |
| 109 MOCK_METHOD0(message_loop, MessageLoop*()); | 116 MOCK_METHOD0(message_loop, MessageLoop*()); |
| 110 MOCK_METHOD0(candidate_config, const CandidateSessionConfig*()); | 117 MOCK_METHOD0(candidate_config, const CandidateSessionConfig*()); |
| 111 MOCK_METHOD0(config, const SessionConfig*()); | 118 MOCK_METHOD0(config, const SessionConfig*()); |
| 112 MOCK_METHOD1(set_config, void(const SessionConfig* config)); | 119 MOCK_METHOD1(set_config, void(const SessionConfig* config)); |
| 113 MOCK_METHOD0(initiator_token, const std::string&()); | 120 MOCK_METHOD0(initiator_token, const std::string&()); |
| 114 MOCK_METHOD1(set_initiator_token, void(const std::string& initiator_token)); | 121 MOCK_METHOD1(set_initiator_token, void(const std::string& initiator_token)); |
| 115 MOCK_METHOD0(receiver_token, const std::string&()); | 122 MOCK_METHOD0(receiver_token, const std::string&()); |
| 116 MOCK_METHOD1(set_receiver_token, void(const std::string& receiver_token)); | 123 MOCK_METHOD1(set_receiver_token, void(const std::string& receiver_token)); |
| 117 MOCK_METHOD1(Close, void(Task* closed_task)); | 124 MOCK_METHOD1(Close, void(Task* closed_task)); |
| 118 | 125 |
| 119 private: | 126 private: |
| 120 DISALLOW_COPY_AND_ASSIGN(MockSession); | 127 DISALLOW_COPY_AND_ASSIGN(MockSession); |
| 121 }; | 128 }; |
| 122 | 129 |
| 123 } // namespace protocol | 130 } // namespace protocol |
| 124 } // namespace remoting | 131 } // namespace remoting |
| 125 | 132 |
| 126 #endif // REMOTING_PROTOCOL_MOCK_OBJECTS_H_ | 133 #endif // REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ |
| OLD | NEW |