Add net::SourceStream and net::FilterSourceStream

net/filter/mock_source_stream.h

Index: net/filter/mock_source_stream.h
diff --git a/net/filter/mock_source_stream.h b/net/filter/mock_source_stream.h
new file mode 100644
index 0000000000000000000000000000000000000000..eff5d914a194d23be8eaf1d589e10e745564108c
--- /dev/null
+++ b/net/filter/mock_source_stream.h
@@ -0,0 +1,63 @@
+// Copyright 2016 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_FILTER_MOCK_SOURCE_STREAM_H_
+#define NET_FILTER_MOCK_SOURCE_STREAM_H_
+
+#include <queue>
+
+#include "base/macros.h"
+#include "net/base/completion_callback.h"
+#include "net/base/io_buffer.h"
+#include "net/filter/source_stream.h"
+
+namespace net {
+
+class IOBuffer;
+
+// A SourceStream implementation used in tests. This allows tests to specify
+// what data to return for each Read() call.
+class MockSourceStream : public SourceStream {
+ public:
+  MockSourceStream();
+  ~MockSourceStream() override;
+
+  // SourceStream implementation
+  int Read(IOBuffer* dest_buffer,
+           int buffer_size,
+           const CompletionCallback& callback) override;
+
+  std::string OrderedTypeStringList() const override;

[mmenke, 2016/08/18 16:18:43]

include <string>

[mmenke, 2016/08/18 16:18:43]

nit:  Remove blank line between override method.  That's just the common
convention, not official rule about it.

[xunjieli, 2016/08/19 14:31:40]

Done.

[xunjieli, 2016/08/19 14:31:40]

Done.

+
+  // Enqueues a result to be returned by |Read|. This method does not make a
+  // copy of |data|, so |data| must outlive this object. If |sync| is true,
+  // |Read| will return the supplied data synchronously; otherwise, consumer
+  // needs to call |CompleteNextRead|
+  void AddReadResult(const char* data, int len, Error error, bool sync);

[mmenke, 2016/08/18 16:18:43]

include net_error.h

[xunjieli, 2016/08/19 14:31:40]

Done.

+
+  // Completes a pending Read() call.
+  void CompleteNextRead();
+
+ private:
+  struct QueuedResult {
+    QueuedResult(const char* data, int len, Error error, bool sync);
+
+    const char* data;
+    const int len;
+    const Error error;
+    const bool sync;
+  };
+
+  std::queue<QueuedResult> results_;
+  bool awaiting_completion_;
+  scoped_refptr<IOBuffer> dest_buffer_;

[mmenke, 2016/08/18 16:18:43]

include ref_counted.h

[xunjieli, 2016/08/19 14:31:40]

Done.

+  CompletionCallback callback_;
+  int dest_buffer_size_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockSourceStream);
+};
+
+}  // namespace net
+
+#endif  // NET_FILTER_MOCK_SOURCE_STREAM_H_