net/quic/test_tools/quic_session_peer.h

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef NET_QUIC_TEST_TOOLS_QUIC_SESSION_PEER_H_
   6 #define NET_QUIC_TEST_TOOLS_QUIC_SESSION_PEER_H_
   7
   8 #include "net/quic/quic_protocol.h"
   9 #include "net/spdy/write_blocked_list.h"
  10
  11 namespace net {
  12
  13 class QuicDataStream;
  14 class QuicHeadersStream;
  15 class QuicSession;
  16
  17 namespace test {
  18
  19 class QuicSessionPeer {
  20  public:
  21   static void SetNextStreamId(QuicSession* session, QuicStreamId id);
  22   static void SetMaxOpenStreams(QuicSession* session, uint32 max_streams);
  23   static QuicHeadersStream* GetHeadersStream(QuicSession* session);
  24   static WriteBlockedList<QuicStreamId>* GetWriteblockedStreams(
  25       QuicSession* session);
  26   static QuicDataStream* GetIncomingDataStream(QuicSession* session,
  27                                                QuicStreamId stream_id);
  28
  29  private:
  30   DISALLOW_COPY_AND_ASSIGN(QuicSessionPeer);
  31 };
  32
  33 }  // namespace test
  34 }  // namespace net
  35
  36 #endif  // NET_QUIC_TEST_TOOLS_QUIC_SESSION_PEER_H_