Index: net/tools/quic/test_tools/quic_test_utils.cc |
diff --git a/net/tools/quic/test_tools/quic_test_utils.cc b/net/tools/quic/test_tools/quic_test_utils.cc |
index 9aed3afd7b31400f61bcfff4d1b59e3c589aa731..a5c22ea6d3e0efc442ad25365195aae0a51a2123 100644 |
--- a/net/tools/quic/test_tools/quic_test_utils.cc |
+++ b/net/tools/quic/test_tools/quic_test_utils.cc |
@@ -27,7 +27,8 @@ MockConnection::MockConnection(QuicGuid guid, |
MockConnection::MockConnection(QuicGuid guid, |
IPEndPoint address, |
bool is_server) |
- : QuicConnection(guid, address, new MockHelper(), is_server), |
+ : QuicConnection(guid, address, new testing::NiceMock<MockHelper>(), |
+ is_server), |
has_mock_helper_(true) { |
} |
@@ -53,6 +54,21 @@ bool TestDecompressorVisitor::OnDecompressedData(StringPiece data) { |
return true; |
} |
+TestSession::TestSession(QuicConnection* connection, bool is_server) |
+ : QuicSession(connection, is_server), |
+ crypto_stream_(NULL) { |
+} |
+ |
+TestSession::~TestSession() {} |
+ |
+void TestSession::SetCryptoStream(QuicCryptoStream* stream) { |
+ crypto_stream_ = stream; |
+} |
+ |
+QuicCryptoStream* TestSession::GetCryptoStream() { |
+ return crypto_stream_; |
+} |
+ |
} // namespace test |
} // namespace tools |
} // namespace net |