| Index: net/tools/quic/test_tools/mock_quic_server_session_visitor.h
|
| diff --git a/net/tools/quic/test_tools/mock_quic_server_session_visitor.h b/net/tools/quic/test_tools/mock_quic_server_session_visitor.h
|
| index 2d0059f616f52097be2087ceaeb3aebcd63d9a5a..d89f49cb8a00f0e0fbf61067d7be1ee522af6989 100644
|
| --- a/net/tools/quic/test_tools/mock_quic_server_session_visitor.h
|
| +++ b/net/tools/quic/test_tools/mock_quic_server_session_visitor.h
|
| @@ -6,6 +6,7 @@
|
| #define NET_TOOLS_QUIC_TEST_TOOLS_MOCK_QUIC_SERVER_SESSION_VISITOR_H_
|
|
|
| #include "base/macros.h"
|
| +#include "net/quic/core/quic_crypto_server_stream.h"
|
| #include "net/quic/core/quic_server_session_base.h"
|
| #include "testing/gmock/include/gmock/gmock.h"
|
|
|
| @@ -24,15 +25,17 @@ class MockQuicServerSessionVisitor : public QuicServerSessionBase::Visitor {
|
| void(QuicBlockedWriterInterface* blocked_writer));
|
| MOCK_METHOD1(OnConnectionAddedToTimeWaitList,
|
| void(QuicConnectionId connection_id));
|
| + MOCK_METHOD1(OnPacketBeingDispatchedToSession,
|
| + void(QuicServerSessionBase* session));
|
|
|
| private:
|
| DISALLOW_COPY_AND_ASSIGN(MockQuicServerSessionVisitor);
|
| };
|
|
|
| -class MockQuicServerSessionHelper : public QuicServerSessionBase::Helper {
|
| +class MockQuicCryptoServerStreamHelper : public QuicCryptoServerStream::Helper {
|
| public:
|
| - MockQuicServerSessionHelper();
|
| - ~MockQuicServerSessionHelper() override;
|
| + MockQuicCryptoServerStreamHelper();
|
| + ~MockQuicCryptoServerStreamHelper() override;
|
| MOCK_CONST_METHOD1(GenerateConnectionIdForReject,
|
| QuicConnectionId(QuicConnectionId connection_id));
|
| MOCK_CONST_METHOD3(CanAcceptClientHello,
|
| @@ -41,7 +44,7 @@ class MockQuicServerSessionHelper : public QuicServerSessionBase::Helper {
|
| std::string* error_details));
|
|
|
| private:
|
| - DISALLOW_COPY_AND_ASSIGN(MockQuicServerSessionHelper);
|
| + DISALLOW_COPY_AND_ASSIGN(MockQuicCryptoServerStreamHelper);
|
| };
|
|
|
| } // namespace test
|
|
|