commit fbeb5bf958fece5c9b5e377f013b0119057a3fa8
author renjietang <renjietang@google.com> Fri Apr 19 15:06:20 2019 -0700
committer Copybara-Service <copybara-worker@google.com> Mon Apr 22 08:41:09 2019 -0700
tree ca6d002685fbcfc59117e62a3857661455078573
parent a745e39b671f22a92dff3fc918617fbfaf60ed1e

gfe-relnote: Move headers streams out of static stream map. Protected by
gfe_reloadable_quic_eliminate_static_stream_map.

Following up this CL, crypto stream will be moved out of static_stream_map too.

PiperOrigin-RevId: 244429772
Change-Id: I5186bc7ab4e6ee9c9f546a3c9456be489a4ddc26

quic/test_tools/quic_spdy_session_peer.cc

diff --git a/quic/test_tools/quic_spdy_session_peer.cc b/quic/test_tools/quic_spdy_session_peer.cc
index dcba12c..956cd88 100644
--- a/quic/test_tools/quic_spdy_session_peer.cc
+++ b/quic/test_tools/quic_spdy_session_peer.cc
@@ -13,7 +13,7 @@
 // static
 QuicHeadersStream* QuicSpdySessionPeer::GetHeadersStream(
     QuicSpdySession* session) {
-  return session->headers_stream_.get();
+  return session->headers_stream();
 }
 
 // static
@@ -25,6 +25,20 @@
   }
 }
 
+void QuicSpdySessionPeer::SetUnownedHeadersStream(
+    QuicSpdySession* session,
+    QuicHeadersStream* headers_stream) {
+  for (auto& it : session->dynamic_streams()) {
+    if (it.first == QuicUtils::GetHeadersStreamId(
+                        session->connection()->transport_version())) {
+      it.second.reset(headers_stream);
+      session->unowned_headers_stream_ =
+          static_cast<QuicHeadersStream*>(it.second.get());
+      break;
+    }
+  }
+}
+
 // static
 const spdy::SpdyFramer& QuicSpdySessionPeer::GetSpdyFramer(
     QuicSpdySession* session) {