Initial implementation of RequestPriority-based HTTP/2 dependencies.

net/spdy/spdy_test_util_common.h

 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef NET_SPDY_SPDY_TEST_UTIL_COMMON_H_
 #define NET_SPDY_SPDY_TEST_UTIL_COMMON_H_
 
 #include <string>
 #include <vector>
 
@@ skipping 280 matching lines @@
   void SetStreamInitialRecvWindowSize(size_t window);
 
  private:
   SpdySessionPool* const pool_;
 
   DISALLOW_COPY_AND_ASSIGN(SpdySessionPoolPeer);
 };
 
 class SpdyTestUtil {
  public:
-  explicit SpdyTestUtil(NextProto protocol);
+  explicit SpdyTestUtil(NextProto protocol, bool dependency_priorities);
+  ~SpdyTestUtil();
 
   // Add the appropriate headers to put |url| into |block|.
   void AddUrlToHeaderBlock(base::StringPiece url,
                            SpdyHeaderBlock* headers) const;
 
   scoped_ptr<SpdyHeaderBlock> ConstructGetHeaderBlock(
       base::StringPiece url) const;
   scoped_ptr<SpdyHeaderBlock> ConstructGetHeaderBlockForProxy(
       base::StringPiece url) const;
   scoped_ptr<SpdyHeaderBlock> ConstructHeadHeaderBlock(
@@ skipping 94 matching lines @@
                                     SpdyRstStreamStatus status) const;
 
   // Constructs a standard SPDY GET SYN frame, optionally compressed
   // for |url|.
   // |extra_headers| are the extra header-value pairs, which typically
   // will vary the most between calls.
   // Returns a SpdyFrame.
   SpdyFrame* ConstructSpdyGet(const char* const url,
                               bool compressed,
                               SpdyStreamId stream_id,
-                              RequestPriority request_priority) const;
+                              RequestPriority request_priority);
 
   SpdyFrame* ConstructSpdyGetForProxy(const char* const url,
                                       bool compressed,
                                       SpdyStreamId stream_id,
                                       RequestPriority request_priority) const;
 
   // Constructs a standard SPDY GET SYN frame, optionally compressed.
   // |extra_headers| are the extra header-value pairs, which typically
   // will vary the most between calls.  If |direct| is false, the
   // the full url will be used instead of simply the path.
   // Returns a SpdyFrame.
   SpdyFrame* ConstructSpdyGet(const char* const extra_headers[],
                               int extra_header_count,
                               bool compressed,
                               int stream_id,
                               RequestPriority request_priority,
-                              bool direct) const;
+                              bool direct);
 
   // Constructs a standard SPDY SYN_STREAM frame for a CONNECT request.
   SpdyFrame* ConstructSpdyConnect(const char* const extra_headers[],
                                   int extra_header_count,
                                   int stream_id,
                                   RequestPriority priority,
-                                  const HostPortPair& host_port_pair) const;
+                                  const HostPortPair& host_port_pair);
 
   // Constructs a standard SPDY push SYN frame.
   // |extra_headers| are the extra header-value pairs, which typically
   // will vary the most between calls.
   // Returns a SpdyFrame.
   SpdyFrame* ConstructSpdyPush(const char* const extra_headers[],
                                int extra_header_count,
                                int stream_id,
                                int associated_stream_id,
                                const char* url);
@@ skipping 16 matching lines @@
   SpdyFrame* ConstructSpdyHeaderFrame(int stream_id,
                                       const char* const headers[],
                                       int header_count);
 
   // Construct a SPDY syn (HEADERS or SYN_STREAM, depending on protocol
   // version) carrying exactly the given headers and priority.
   SpdyFrame* ConstructSpdySyn(int stream_id,
                               const SpdyHeaderBlock& headers,
                               RequestPriority priority,
                               bool compressed,
-                              bool fin) const;
+                              bool fin);
 
   // Construct a SPDY reply (HEADERS or SYN_REPLY, depending on protocol
   // version) carrying exactly the given headers, and the default priority
   // (or no priority, depending on protocl version).
   // The |headers| parameter variant is preferred.
   SpdyFrame* ConstructSpdyReply(int stream_id, const SpdyHeaderBlock& headers);
 
   // Constructs a standard SPDY SYN_REPLY frame to match the SPDY GET.
   // |extra_headers| are the extra header-value pairs, which typically
   // will vary the most between calls.
@@ skipping 57 matching lines @@
   SpdyFrame* ConstructSpdyBodyFrame(int stream_id,
                                     const char* data,
                                     uint32 len,
                                     bool fin,
                                     int padding_length);
 
   // Wraps |frame| in the payload of a data frame in stream |stream_id|.
   SpdyFrame* ConstructWrappedSpdyFrame(const scoped_ptr<SpdyFrame>& frame,
                                        int stream_id);
 
+  // Called when necessary (when it will affect stream dependency specification
+  // when setting dependencies based on priorioties) to notify the utility
+  // class of stream destruction.
+  void UpdateWithStreamDestruction(int stream_id);
+
   const SpdyHeaderInfo MakeSpdyHeader(SpdyFrameType type);
 
   // For versions below SPDY4, adds the version HTTP/1.1 header.
   void MaybeAddVersionHeader(SpdyFrameWithHeaderBlockIR* frame_ir) const;
   void MaybeAddVersionHeader(SpdyHeaderBlock* block) const;
 
   // Maps |priority| to SPDY version priority, and sets it on |frame_ir|.
   void SetPriority(RequestPriority priority, SpdySynStreamIR* frame_ir) const;
 
   NextProto protocol() const { return protocol_; }
@@ skipping 15 matching lines @@
   // |content_length| may be NULL, in which case the content-length
   // header will be omitted.
   scoped_ptr<SpdyHeaderBlock> ConstructHeaderBlock(
       base::StringPiece method,
       base::StringPiece url,
       int64* content_length) const;
 
   const NextProto protocol_;
   const SpdyMajorVersion spdy_version_;
   GURL default_url_;
+  bool dependency_priorities_;
+
+  // Track a FIFO list of the stream_id of all created requests by priority.
+  std::map<int, std::vector<int>> priority_to_stream_id_list_;
 };
 
 }  // namespace net
 
 #endif  // NET_SPDY_SPDY_TEST_UTIL_COMMON_H_