| Index: blimp/net/test_common.h
|
| diff --git a/blimp/net/test_common.h b/blimp/net/test_common.h
|
| index c726feba5a8c2645cf72d5f2d4e59f95fba64786..54d1b330a77dc6ab4c4c104c2db1fd20618cda74 100644
|
| --- a/blimp/net/test_common.h
|
| +++ b/blimp/net/test_common.h
|
| @@ -17,6 +17,7 @@
|
| #include "blimp/net/blimp_transport.h"
|
| #include "blimp/net/connection_error_observer.h"
|
| #include "blimp/net/connection_handler.h"
|
| +#include "blimp/net/message_port.h"
|
| #include "blimp/net/packet_reader.h"
|
| #include "blimp/net/packet_writer.h"
|
| #include "net/socket/stream_socket.h"
|
| @@ -139,9 +140,9 @@ class MockTransport : public BlimpTransport {
|
| ~MockTransport() override;
|
|
|
| MOCK_METHOD1(Connect, void(const net::CompletionCallback& callback));
|
| - MOCK_METHOD0(TakeConnectionPtr, BlimpConnection*());
|
| + MOCK_METHOD0(TakeMessagePortPtr, MessagePort*());
|
|
|
| - std::unique_ptr<BlimpConnection> TakeConnection() override;
|
| + std::unique_ptr<MessagePort> TakeMessagePort() override;
|
| const char* GetName() const override;
|
| };
|
|
|
|
|