| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_ | 5 #ifndef NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_ |
| 6 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_ | 6 #define NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_ |
| 7 | 7 |
| 8 #include "base/memory/scoped_ptr.h" | 8 #include "base/memory/scoped_ptr.h" |
| 9 #include "net/quic/quic_packet_writer.h" | 9 #include "net/quic/quic_packet_writer.h" |
| 10 | 10 |
| 11 namespace net { | 11 namespace net { |
| 12 namespace test { | 12 namespace test { |
| 13 | 13 |
| 14 // Allows setting a writer for a QuicConnection, to allow | 14 // Allows setting a writer for a QuicConnection, to allow |
| 15 // fine-grained control of writes. | 15 // fine-grained control of writes. |
| 16 class QuicTestWriter : public QuicPacketWriter { | 16 class QuicTestWriter : public QuicPacketWriter { |
| 17 public: | 17 public: |
| 18 QuicTestWriter(); | 18 QuicTestWriter(); |
| 19 virtual ~QuicTestWriter(); | 19 virtual ~QuicTestWriter() OVERRIDE; |
| 20 // Takes ownership of |writer|. | 20 // Takes ownership of |writer|. |
| 21 void set_writer(QuicPacketWriter* writer); | 21 void set_writer(QuicPacketWriter* writer); |
| 22 | 22 |
| 23 protected: | 23 protected: |
| 24 QuicPacketWriter* writer(); | 24 QuicPacketWriter* writer(); |
| 25 | 25 |
| 26 private: | 26 private: |
| 27 scoped_ptr<QuicPacketWriter> writer_; | 27 scoped_ptr<QuicPacketWriter> writer_; |
| 28 }; | 28 }; |
| 29 | 29 |
| 30 } // namespace test | 30 } // namespace test |
| 31 } // namespace net | 31 } // namespace net |
| 32 | 32 |
| 33 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_ | 33 #endif // NET_QUIC_TEST_TOOLS_QUIC_TEST_WRITER_H_ |
| OLD | NEW |