Home | History | Annotate | Download | only in test_tools 
      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_
     34