Index: components/cast_channel/cast_test_util.h |
diff --git a/components/cast_channel/cast_test_util.h b/components/cast_channel/cast_test_util.h |
index 48273c223efe8267c194b6d2a60c64f42fc08b34..5c5510148bc05325b946c1ab8bb17a8e49773705 100644 |
--- a/components/cast_channel/cast_test_util.h |
+++ b/components/cast_channel/cast_test_util.h |
@@ -74,21 +74,43 @@ class MockCastSocket : public CastSocket { |
} |
MOCK_METHOD1(Close, void(const net::CompletionCallback& callback)); |
- MOCK_CONST_METHOD0(ip_endpoint, const net::IPEndPoint&()); |
- MOCK_CONST_METHOD0(id, int()); |
- MOCK_METHOD1(set_id, void(int id)); |
- MOCK_CONST_METHOD0(channel_auth, ChannelAuthType()); |
MOCK_CONST_METHOD0(ready_state, ReadyState()); |
- MOCK_CONST_METHOD0(error_state, ChannelError()); |
- MOCK_CONST_METHOD0(keep_alive, bool(void)); |
- MOCK_CONST_METHOD0(audio_only, bool(void)); |
- MOCK_METHOD1(SetErrorState, void(ChannelError error_state)); |
- CastTransport* transport() const override { return mock_transport_.get(); } |
+ const net::IPEndPoint& ip_endpoint() const override { return ip_endpoint_; } |
+ void SetIPEndpoint(const net::IPEndPoint& ip_endpoint) { |
+ ip_endpoint_ = ip_endpoint; |
+ } |
+ |
+ int id() const override { return channel_id_; } |
+ void set_id(int id) override { channel_id_ = id; } |
+ ChannelAuthType channel_auth() const override { return channel_auth_; } |
+ void SetChannelAuth(ChannelAuthType channel_auth) { |
+ channel_auth_ = channel_auth; |
+ } |
+ |
+ ChannelError error_state() const override { return error_state_; } |
+ void SetErrorState(ChannelError error_state) override { |
+ error_state_ = error_state; |
+ } |
+ |
+ bool keep_alive() const override { return keep_alive_; } |
+ void SetKeepAlive(bool keep_alive) { keep_alive_ = keep_alive; } |
+ |
+ bool audio_only() const override { return audio_only_; } |
+ void SetAudioOnly(bool audio_only) { audio_only_ = audio_only; } |
+ |
+ CastTransport* transport() const override { return mock_transport_.get(); } |
MockCastTransport* mock_transport() const { return mock_transport_.get(); } |
private: |
+ net::IPEndPoint ip_endpoint_; |
+ int channel_id_; |
+ ChannelAuthType channel_auth_; |
+ ChannelError error_state_; |
+ bool keep_alive_; |
+ bool audio_only_; |
+ |
std::unique_ptr<MockCastTransport> mock_transport_; |
std::unique_ptr<CastTransport::Delegate> delegate_; |