src/net/quic/test_tools/quic_test_writer.h

   1 // Copyright 2013 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_TEST_WRITER_H_
   6 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "net/quic/quic_packet_writer.h"
  10
  11 namespace net {
  12 namespace test {
  13
  14 // Allows setting a writer for a QuicConnection, to allow
  15 // fine-grained control of writes.
  16 class QuicTestWriter : public QuicPacketWriter {
  17  public:
  18   QuicTestWriter();
  19   virtual ~QuicTestWriter();
  20   // Takes ownership of |writer|.
  21   void set_writer(QuicPacketWriter* writer);
  22
  23  protected:
  24   QuicPacketWriter* writer();
  25
  26  private:
  27   scoped_ptr<QuicPacketWriter> writer_;
  28 };
  29
  30 }  // namespace test
  31 }  // namespace net
  32
  33 #endif  // NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_