| Index: blimp/net/test_common.h
|
| diff --git a/blimp/net/test_common.h b/blimp/net/test_common.h
|
| index 58ee2f6a5c3a0b792c0d93ca1716452befa584de..b42c3d82c8209e8093696644a2be91a2c738381d 100644
|
| --- a/blimp/net/test_common.h
|
| +++ b/blimp/net/test_common.h
|
| @@ -9,6 +9,7 @@
|
|
|
| #include "base/memory/scoped_ptr.h"
|
| #include "blimp/common/proto/blimp_message.pb.h"
|
| +#include "blimp/net/blimp_connection.h"
|
| #include "blimp/net/blimp_message_processor.h"
|
| #include "blimp/net/blimp_transport.h"
|
| #include "blimp/net/connection_error_observer.h"
|
| @@ -165,6 +166,20 @@ class MockPacketWriter : public PacketWriter {
|
| const net::CompletionCallback&));
|
| };
|
|
|
| +class MockBlimpConnection : public BlimpConnection {
|
| + public:
|
| + MockBlimpConnection();
|
| + ~MockBlimpConnection() override;
|
| +
|
| + MOCK_METHOD1(SetConnectionErrorObserver,
|
| + void(ConnectionErrorObserver* observer));
|
| +
|
| + MOCK_METHOD1(SetIncomingMessageProcessor,
|
| + void(BlimpMessageProcessor* processor));
|
| +
|
| + MOCK_CONST_METHOD0(GetOutgoingMessageProcessor, BlimpMessageProcessor*());
|
| +};
|
| +
|
| class MockConnectionErrorObserver : public ConnectionErrorObserver {
|
| public:
|
| MockConnectionErrorObserver();
|
|
|