| 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 66bac563872b031489810cbfbcc961208e2961a0..7e37ca45da188f8fc16d619ce8bb60d9079afac2 100644
|
| --- a/components/cast_channel/cast_test_util.h
|
| +++ b/components/cast_channel/cast_test_util.h
|
| @@ -11,6 +11,7 @@
|
| #include "base/macros.h"
|
| #include "base/threading/thread_task_runner_handle.h"
|
| #include "components/cast_channel/cast_socket.h"
|
| +#include "components/cast_channel/cast_socket_service.h"
|
| #include "components/cast_channel/cast_transport.h"
|
| #include "components/cast_channel/proto/cast_channel.pb.h"
|
| #include "net/base/ip_endpoint.h"
|
| @@ -64,6 +65,19 @@ class MockCastSocketObserver : public CastSocket::Observer {
|
| void(const CastSocket& socket, const CastMessage& message));
|
| };
|
|
|
| +class MockCastSocketService : public CastSocketService {
|
| + public:
|
| + MockCastSocketService();
|
| + ~MockCastSocketService() override;
|
| +
|
| + MOCK_METHOD4(OpenSocket,
|
| + int(const net::IPEndPoint& ip_endpoint,
|
| + net::NetLog* net_log,
|
| + const CastSocket::OnOpenCallback& open_cb,
|
| + CastSocket::Observer* observer));
|
| + MOCK_CONST_METHOD1(GetSocket, CastSocket*(int channel_id));
|
| +};
|
| +
|
| class MockCastSocket : public CastSocket {
|
| public:
|
| MockCastSocket();
|
| @@ -73,6 +87,7 @@ class MockCastSocket : public CastSocket {
|
| MOCK_METHOD1(Close, void(const net::CompletionCallback& callback));
|
| MOCK_CONST_METHOD0(ready_state, ReadyState());
|
| MOCK_METHOD1(AddObserver, void(Observer* observer));
|
| + MOCK_METHOD1(RemoveObserver, void(Observer* observer));
|
|
|
| const net::IPEndPoint& ip_endpoint() const override { return ip_endpoint_; }
|
| void SetIPEndpoint(const net::IPEndPoint& ip_endpoint) {
|
|
|