OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 COMPONENTS_CAST_CHANNEL_CAST_TEST_UTIL_H_ | 5 #ifndef COMPONENTS_CAST_CHANNEL_CAST_TEST_UTIL_H_ |
6 #define COMPONENTS_CAST_CHANNEL_CAST_TEST_UTIL_H_ | 6 #define COMPONENTS_CAST_CHANNEL_CAST_TEST_UTIL_H_ |
7 | 7 |
8 #include <string> | 8 #include <string> |
9 #include <utility> | 9 #include <utility> |
10 | 10 |
(...skipping 51 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
62 MOCK_METHOD2(OnError, void(const CastSocket& socket, ChannelError error)); | 62 MOCK_METHOD2(OnError, void(const CastSocket& socket, ChannelError error)); |
63 MOCK_METHOD2(OnMessage, | 63 MOCK_METHOD2(OnMessage, |
64 void(const CastSocket& socket, const CastMessage& message)); | 64 void(const CastSocket& socket, const CastMessage& message)); |
65 }; | 65 }; |
66 | 66 |
67 class MockCastSocket : public CastSocket { | 67 class MockCastSocket : public CastSocket { |
68 public: | 68 public: |
69 MockCastSocket(); | 69 MockCastSocket(); |
70 ~MockCastSocket() override; | 70 ~MockCastSocket() override; |
71 | 71 |
72 MOCK_METHOD1(Connect, void(base::Callback<void(ChannelError)> callback)); | 72 MOCK_METHOD1(Connect, void(const OnOpenCallback& callback)); |
73 MOCK_METHOD1(Close, void(const net::CompletionCallback& callback)); | 73 MOCK_METHOD1(Close, void(const net::CompletionCallback& callback)); |
74 MOCK_CONST_METHOD0(ip_endpoint, const net::IPEndPoint&()); | |
75 MOCK_CONST_METHOD0(id, int()); | |
76 MOCK_METHOD1(set_id, void(int id)); | |
77 MOCK_CONST_METHOD0(ready_state, ReadyState()); | 74 MOCK_CONST_METHOD0(ready_state, ReadyState()); |
78 MOCK_CONST_METHOD0(error_state, ChannelError()); | |
79 MOCK_CONST_METHOD0(keep_alive, bool(void)); | |
80 MOCK_CONST_METHOD0(audio_only, bool(void)); | |
81 MOCK_METHOD1(SetErrorState, void(ChannelError error_state)); | |
82 MOCK_METHOD1(AddObserver, void(Observer* observer)); | 75 MOCK_METHOD1(AddObserver, void(Observer* observer)); |
83 | 76 |
| 77 const net::IPEndPoint& ip_endpoint() const override { return ip_endpoint_; } |
| 78 void SetIPEndpoint(const net::IPEndPoint& ip_endpoint) { |
| 79 ip_endpoint_ = ip_endpoint; |
| 80 } |
| 81 |
| 82 int id() const override { return channel_id_; } |
| 83 void set_id(int id) override { channel_id_ = id; } |
| 84 |
| 85 ChannelError error_state() const override { return error_state_; } |
| 86 void SetErrorState(ChannelError error_state) override { |
| 87 error_state_ = error_state; |
| 88 } |
| 89 |
| 90 bool keep_alive() const override { return keep_alive_; } |
| 91 void SetKeepAlive(bool keep_alive) { keep_alive_ = keep_alive; } |
| 92 |
| 93 bool audio_only() const override { return audio_only_; } |
| 94 void SetAudioOnly(bool audio_only) { audio_only_ = audio_only; } |
| 95 |
84 CastTransport* transport() const override { return mock_transport_.get(); } | 96 CastTransport* transport() const override { return mock_transport_.get(); } |
85 | |
86 MockCastTransport* mock_transport() const { return mock_transport_.get(); } | 97 MockCastTransport* mock_transport() const { return mock_transport_.get(); } |
87 | 98 |
88 private: | 99 private: |
| 100 net::IPEndPoint ip_endpoint_; |
| 101 int channel_id_; |
| 102 ChannelError error_state_; |
| 103 bool keep_alive_; |
| 104 bool audio_only_; |
| 105 |
89 std::unique_ptr<MockCastTransport> mock_transport_; | 106 std::unique_ptr<MockCastTransport> mock_transport_; |
90 std::unique_ptr<Observer> observer_; | 107 std::unique_ptr<Observer> observer_; |
91 | 108 |
92 DISALLOW_COPY_AND_ASSIGN(MockCastSocket); | 109 DISALLOW_COPY_AND_ASSIGN(MockCastSocket); |
93 }; | 110 }; |
94 | 111 |
95 // Creates the IPEndpoint 192.168.1.1. | 112 // Creates the IPEndpoint 192.168.1.1. |
96 net::IPEndPoint CreateIPEndPointForTest(); | 113 net::IPEndPoint CreateIPEndPointForTest(); |
97 | 114 |
98 // Checks if two proto messages are the same. | 115 // Checks if two proto messages are the same. |
(...skipping 10 matching lines...) Expand all Loading... |
109 ACTION_TEMPLATE(PostCompletionCallbackTask, | 126 ACTION_TEMPLATE(PostCompletionCallbackTask, |
110 HAS_1_TEMPLATE_PARAMS(int, cb_idx), | 127 HAS_1_TEMPLATE_PARAMS(int, cb_idx), |
111 AND_1_VALUE_PARAMS(rv)) { | 128 AND_1_VALUE_PARAMS(rv)) { |
112 base::ThreadTaskRunnerHandle::Get()->PostTask( | 129 base::ThreadTaskRunnerHandle::Get()->PostTask( |
113 FROM_HERE, base::Bind(testing::get<cb_idx>(args), rv)); | 130 FROM_HERE, base::Bind(testing::get<cb_idx>(args), rv)); |
114 } | 131 } |
115 | 132 |
116 } // namespace cast_channel | 133 } // namespace cast_channel |
117 | 134 |
118 #endif // COMPONENTS_CAST_CHANNEL_CAST_TEST_UTIL_H_ | 135 #endif // COMPONENTS_CAST_CHANNEL_CAST_TEST_UTIL_H_ |
OLD | NEW |