Index: net/quic/test_tools/quic_connection_peer.cc |
diff --git a/net/quic/test_tools/quic_connection_peer.cc b/net/quic/test_tools/quic_connection_peer.cc |
index 5bd6370d9777f51c61981c969cc382d79048f706..15653ceabe15938b73678a1ad13f5cfd936f5527 100644 |
--- a/net/quic/test_tools/quic_connection_peer.cc |
+++ b/net/quic/test_tools/quic_connection_peer.cc |
@@ -73,6 +73,12 @@ QuicTime::Delta QuicConnectionPeer::GetNetworkTimeout( |
} |
// static |
+QuicSentEntropyManager* QuicConnectionPeer::GetSentEntropyManager( |
+ QuicConnection* connection) { |
+ return &connection->sent_entropy_manager_; |
+} |
+ |
+// static |
// TODO(ianswett): Create a GetSentEntropyHash which accepts an AckFrame. |
QuicPacketEntropyHash QuicConnectionPeer::GetSentEntropyHash( |
QuicConnection* connection, |
@@ -213,7 +219,11 @@ void QuicConnectionPeer::CloseConnection(QuicConnection* connection) { |
// static |
QuicEncryptedPacket* QuicConnectionPeer::GetConnectionClosePacket( |
QuicConnection* connection) { |
- return connection->connection_close_packet_.get(); |
+ if (connection->termination_packets_ == nullptr || |
+ connection->termination_packets_->empty()) { |
+ return nullptr; |
+ } |
+ return (*connection->termination_packets_)[0]; |
} |
// static |