Make ReleaseStream() take a Delegate

net/quic/chromium/quic_chromium_client_session.cc

Index: net/quic/chromium/quic_chromium_client_session.cc
diff --git a/net/quic/chromium/quic_chromium_client_session.cc b/net/quic/chromium/quic_chromium_client_session.cc
index 042fb5ae132bdf52749fa68cb92f726e8b9d0d48..6d46a328b9b570d5c18509b00c948e12264835ec 100644
--- a/net/quic/chromium/quic_chromium_client_session.cc
+++ b/net/quic/chromium/quic_chromium_client_session.cc
@@ -301,10 +301,11 @@ int QuicChromiumClientSession::Handle::RequestStream(
   return stream_request_->StartRequest(callback);
 }
 
-QuicChromiumClientStream* QuicChromiumClientSession::Handle::ReleaseStream() {
+QuicChromiumClientStream* QuicChromiumClientSession::Handle::ReleaseStream(
+    QuicChromiumClientStream::Delegate* delegate) {
   DCHECK(stream_request_);
 
-  auto* stream = stream_request_->ReleaseStream();
+  auto* stream = stream_request_->ReleaseStream(delegate);
   stream_request_.reset();
   return stream;
 }
@@ -376,10 +377,12 @@ int QuicChromiumClientSession::StreamRequest::StartRequest(
 }
 
 QuicChromiumClientStream*
-QuicChromiumClientSession::StreamRequest::ReleaseStream() {
+QuicChromiumClientSession::StreamRequest::ReleaseStream(
+    QuicChromiumClientStream::Delegate* delegate) {
   DCHECK(stream_);
   QuicChromiumClientStream* stream = stream_;
   stream_ = nullptr;
+  stream->SetDelegate(delegate);
   return stream;
 }