| Index: net/quic/test_tools/simple_data_producer.h
|
| diff --git a/net/quic/test_tools/simple_data_producer.h b/net/quic/test_tools/simple_data_producer.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..c830cc10fd8727b6cf815cf149a717d465f29aef
|
| --- /dev/null
|
| +++ b/net/quic/test_tools/simple_data_producer.h
|
| @@ -0,0 +1,49 @@
|
| +// Copyright (c) 2017 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_QUIC_TEST_TOOLS_SIMPLE_DATA_PRODUCER_H_
|
| +#define NET_QUIC_TEST_TOOLS_SIMPLE_DATA_PRODUCER_H_
|
| +
|
| +#include <unordered_map>
|
| +
|
| +#include "net/quic/core/quic_simple_buffer_allocator.h"
|
| +#include "net/quic/core/quic_stream_frame_data_producer.h"
|
| +#include "net/quic/core/quic_stream_send_buffer.h"
|
| +
|
| +namespace net {
|
| +
|
| +namespace test {
|
| +
|
| +// A simple data producer which copies stream data into a map from stream
|
| +// id to send buffer.
|
| +class SimpleDataProducer : public QuicStreamFrameDataProducer {
|
| + public:
|
| + SimpleDataProducer();
|
| + ~SimpleDataProducer() override;
|
| +
|
| + // QuicStreamFrameDataProducer methods:
|
| + void SaveStreamData(QuicStreamId id,
|
| + QuicIOVector iov,
|
| + size_t iov_offset,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length) override;
|
| + bool WriteStreamData(QuicStreamId id,
|
| + QuicStreamOffset offset,
|
| + QuicByteCount data_length,
|
| + QuicDataWriter* writer) override;
|
| +
|
| + private:
|
| + using SendBufferMap =
|
| + std::unordered_map<QuicStreamId, std::unique_ptr<QuicStreamSendBuffer>>;
|
| +
|
| + SendBufferMap send_buffer_map_;
|
| +
|
| + SimpleBufferAllocator allocator_;
|
| +};
|
| +
|
| +} // namespace test
|
| +
|
| +} // namespace net
|
| +
|
| +#endif // NET_QUIC_TEST_TOOLS_SIMPLE_DATA_PRODUCER_H_
|
|
|