OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_PROTOCOL_MOCK_OBJECTS_H_ | 5 #ifndef REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ |
6 #define REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ | 6 #define REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ |
7 | 7 |
8 #include <map> | 8 #include <map> |
9 #include <string> | 9 #include <string> |
10 | 10 |
11 #include "base/location.h" | 11 #include "base/location.h" |
12 #include "base/single_thread_task_runner.h" | 12 #include "base/single_thread_task_runner.h" |
13 #include "base/values.h" | 13 #include "base/values.h" |
14 #include "net/base/ip_endpoint.h" | 14 #include "net/base/ip_endpoint.h" |
15 #include "remoting/proto/internal.pb.h" | 15 #include "remoting/proto/internal.pb.h" |
16 #include "remoting/proto/video.pb.h" | 16 #include "remoting/proto/video.pb.h" |
17 #include "remoting/protocol/authenticator.h" | 17 #include "remoting/protocol/authenticator.h" |
18 #include "remoting/protocol/client_stub.h" | 18 #include "remoting/protocol/client_stub.h" |
19 #include "remoting/protocol/clipboard_stub.h" | 19 #include "remoting/protocol/clipboard_stub.h" |
20 #include "remoting/protocol/connection_to_client.h" | 20 #include "remoting/protocol/connection_to_client.h" |
21 #include "remoting/protocol/host_stub.h" | 21 #include "remoting/protocol/host_stub.h" |
22 #include "remoting/protocol/input_stub.h" | 22 #include "remoting/protocol/input_stub.h" |
23 #include "remoting/protocol/pairing_registry.h" | 23 #include "remoting/protocol/pairing_registry.h" |
24 #include "remoting/protocol/session.h" | 24 #include "remoting/protocol/session.h" |
25 #include "remoting/protocol/session_manager.h" | 25 #include "remoting/protocol/session_manager.h" |
26 #include "remoting/protocol/transport.h" | 26 #include "remoting/protocol/transport.h" |
27 #include "remoting/protocol/video_stub.h" | 27 #include "remoting/protocol/video_sender.h" |
28 #include "testing/gmock/include/gmock/gmock.h" | 28 #include "testing/gmock/include/gmock/gmock.h" |
29 | 29 |
30 namespace remoting { | 30 namespace remoting { |
31 namespace protocol { | 31 namespace protocol { |
32 | 32 |
33 class MockConnectionToClient : public ConnectionToClient { | 33 class MockConnectionToClient : public ConnectionToClient { |
34 public: | 34 public: |
35 MockConnectionToClient(Session* session, | 35 MockConnectionToClient(Session* session, |
36 HostStub* host_stub); | 36 HostStub* host_stub); |
37 virtual ~MockConnectionToClient(); | 37 virtual ~MockConnectionToClient(); |
38 | 38 |
39 MOCK_METHOD1(Init, void(Session* session)); | 39 MOCK_METHOD1(Init, void(Session* session)); |
40 MOCK_METHOD0(video_stub, VideoStub*()); | 40 MOCK_METHOD0(video_sender, VideoSender*()); |
41 MOCK_METHOD0(client_stub, ClientStub*()); | 41 MOCK_METHOD0(client_stub, ClientStub*()); |
42 MOCK_METHOD0(session, Session*()); | 42 MOCK_METHOD0(session, Session*()); |
43 MOCK_METHOD0(Disconnect, void()); | 43 MOCK_METHOD0(Disconnect, void()); |
44 | 44 |
45 private: | 45 private: |
46 DISALLOW_COPY_AND_ASSIGN(MockConnectionToClient); | 46 DISALLOW_COPY_AND_ASSIGN(MockConnectionToClient); |
47 }; | 47 }; |
48 | 48 |
49 class MockConnectionToClientEventHandler : | 49 class MockConnectionToClientEventHandler : |
50 public ConnectionToClient::EventHandler { | 50 public ConnectionToClient::EventHandler { |
(...skipping 104 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
155 DISALLOW_COPY_AND_ASSIGN(MockCursorShapeStub); | 155 DISALLOW_COPY_AND_ASSIGN(MockCursorShapeStub); |
156 }; | 156 }; |
157 | 157 |
158 class MockVideoStub : public VideoStub { | 158 class MockVideoStub : public VideoStub { |
159 public: | 159 public: |
160 MockVideoStub(); | 160 MockVideoStub(); |
161 virtual ~MockVideoStub(); | 161 virtual ~MockVideoStub(); |
162 | 162 |
163 MOCK_METHOD2(ProcessVideoPacketPtr, void(const VideoPacket* video_packet, | 163 MOCK_METHOD2(ProcessVideoPacketPtr, void(const VideoPacket* video_packet, |
164 const base::Closure& done)); | 164 const base::Closure& done)); |
165 virtual void ProcessVideoPacket(scoped_ptr<VideoPacket> video_packet, | 165 void ProcessVideoPacket(scoped_ptr<VideoPacket> video_packet, |
166 const base::Closure& done) { | 166 const base::Closure& done) override { |
167 ProcessVideoPacketPtr(video_packet.get(), done); | 167 ProcessVideoPacketPtr(video_packet.get(), done); |
168 } | 168 } |
169 | 169 |
170 private: | 170 private: |
171 DISALLOW_COPY_AND_ASSIGN(MockVideoStub); | 171 DISALLOW_COPY_AND_ASSIGN(MockVideoStub); |
172 }; | 172 }; |
173 | 173 |
| 174 class MockVideoSender : public VideoSender { |
| 175 public: |
| 176 MockVideoSender(); |
| 177 virtual ~MockVideoSender(); |
| 178 |
| 179 MOCK_METHOD2(ProcessVideoPacketPtr, void(const VideoPacket* video_packet, |
| 180 const base::Closure& done)); |
| 181 void ProcessVideoPacket(scoped_ptr<VideoPacket> video_packet, |
| 182 const base::Closure& done) override { |
| 183 ProcessVideoPacketPtr(video_packet.get(), done); |
| 184 } |
| 185 bool SupportsAcks() override { return true; } |
| 186 |
| 187 private: |
| 188 DISALLOW_COPY_AND_ASSIGN(MockVideoSender); |
| 189 }; |
| 190 |
174 class MockSession : public Session { | 191 class MockSession : public Session { |
175 public: | 192 public: |
176 MockSession(); | 193 MockSession(); |
177 virtual ~MockSession(); | 194 virtual ~MockSession(); |
178 | 195 |
179 MOCK_METHOD1(SetEventHandler, void(Session::EventHandler* event_handler)); | 196 MOCK_METHOD1(SetEventHandler, void(Session::EventHandler* event_handler)); |
180 MOCK_METHOD0(error, ErrorCode()); | 197 MOCK_METHOD0(error, ErrorCode()); |
181 MOCK_METHOD0(GetTransportChannelFactory, StreamChannelFactory*()); | 198 MOCK_METHOD0(GetTransportChannelFactory, StreamChannelFactory*()); |
182 MOCK_METHOD0(GetMultiplexedChannelFactory, StreamChannelFactory*()); | 199 MOCK_METHOD0(GetMultiplexedChannelFactory, StreamChannelFactory*()); |
183 MOCK_METHOD0(jid, const std::string&()); | 200 MOCK_METHOD0(jid, const std::string&()); |
(...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
254 // Runs tasks synchronously instead of posting them to |task_runner|. | 271 // Runs tasks synchronously instead of posting them to |task_runner|. |
255 void PostTask(const scoped_refptr<base::SingleThreadTaskRunner>& task_runner, | 272 void PostTask(const scoped_refptr<base::SingleThreadTaskRunner>& task_runner, |
256 const tracked_objects::Location& from_here, | 273 const tracked_objects::Location& from_here, |
257 const base::Closure& task) override; | 274 const base::Closure& task) override; |
258 }; | 275 }; |
259 | 276 |
260 } // namespace protocol | 277 } // namespace protocol |
261 } // namespace remoting | 278 } // namespace remoting |
262 | 279 |
263 #endif // REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ | 280 #endif // REMOTING_PROTOCOL_PROTOCOL_MOCK_OBJECTS_H_ |
OLD | NEW |