QUIC - refactor stream creation. Guarded by --quic_reloadable_flag_quic_refactor_stream_creation.

net/tools/quic/test_tools/quic_test_server.cc

Index: net/tools/quic/test_tools/quic_test_server.cc
diff --git a/net/tools/quic/test_tools/quic_test_server.cc b/net/tools/quic/test_tools/quic_test_server.cc
index 95bf8d2c9e79d555a5c65e82fd6a26fb52731a67..a99706f96b30bd4bcd60c76ef22b1c9f76bd49f0 100644
--- a/net/tools/quic/test_tools/quic_test_server.cc
+++ b/net/tools/quic/test_tools/quic_test_server.cc
@@ -38,6 +38,7 @@ class CustomStreamSession : public QuicSimpleServerSession {
         crypto_stream_factory_(crypto_stream_factory) {}
 
   QuicSpdyStream* CreateIncomingDynamicStream(QuicStreamId id) override {
+    DCHECK(!FLAGS_quic_reloadable_flag_quic_refactor_stream_creation);
     if (!ShouldCreateIncomingDynamicStream(id)) {
       return nullptr;
     }
@@ -50,6 +51,14 @@ class CustomStreamSession : public QuicSimpleServerSession {
     return QuicSimpleServerSession::CreateIncomingDynamicStream(id);
   }
 
+  std::unique_ptr<QuicStream> CreateStream(QuicStreamId id) override {
+    if (stream_factory_) {
+      return QuicWrapUnique<QuicSpdyStream>(
+          stream_factory_->CreateStream(id, this, response_cache()));
+    }
+    return QuicSimpleServerSession::CreateStream(id);
+  }
+
   QuicCryptoServerStreamBase* CreateQuicCryptoServerStream(
       const QuicCryptoServerConfig* crypto_config,
       QuicCompressedCertsCache* compressed_certs_cache) override {