Add net::SourceStream and net::FilterSourceStream

net/filter/mock_source_stream.cc

Index: net/filter/mock_source_stream.cc
diff --git a/net/filter/mock_source_stream.cc b/net/filter/mock_source_stream.cc
new file mode 100644
index 0000000000000000000000000000000000000000..dd3abc422c5ea89b68a7ec44d0c3bc3ece9fe557
--- /dev/null
+++ b/net/filter/mock_source_stream.cc
@@ -0,0 +1,72 @@
+// Copyright 2015 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.
+
+#include "net/filter/mock_source_stream.h"
+
+#include "net/base/io_buffer.h"
+
+namespace net {
+
+MockSourceStream::MockSourceStream()
+    : SourceStream(SourceStream::TYPE_NONE),
+      awaiting_completion_(false),
+      dest_buffer_(nullptr),
+      dest_buffer_size_(0) {}
+
+MockSourceStream::~MockSourceStream() {}

[mmenke, 2016/08/17 19:01:56]

DCHECK(results_.empty())?

+
+int MockSourceStream::Read(IOBuffer* dest_buffer,
+                           size_t buffer_size,
+                           const CompletionCallback& callback) {
+  DCHECK(!awaiting_completion_);
+  if (results_.empty()) {
+    return OK;

[mmenke, 2016/08/17 19:01:56]

ERR_UNEXPECTED, and a DCHECK?  Then the consumer would have to add the last
read, too.

[xunjieli, 2016/08/17 20:00:20]

Done.

+  }

[mmenke, 2016/08/17 19:01:56]

nit:  Remove braces

[xunjieli, 2016/08/17 20:00:20]

Done.

+
+  QueuedResult r = results_.front();
+  if (!r.sync) {
+    awaiting_completion_ = true;
+    dest_buffer_ = dest_buffer;
+    dest_buffer_size_ = buffer_size;
+    callback_ = callback;
+    return ERR_IO_PENDING;
+  }
+
+  results_.pop();
+  DCHECK_GE(buffer_size, r.len);

[mmenke, 2016/08/17 19:01:56]

May want to move this up below the !r.sync check - it gives us a more useful
stack on failure.

[mmenke, 2016/08/17 19:05:42]

Er...up above

[xunjieli, 2016/08/17 20:00:20]

Acknowledged.

[xunjieli, 2016/08/17 20:00:20]

Done.

+  memcpy(dest_buffer->data(), r.data, r.len);
+  return r.error == OK ? r.len : r.error;
+}
+
+std::string MockSourceStream::OrderedTypeStringList() const {
+  return "";
+}
+
+MockSourceStream::QueuedResult::QueuedResult(const char* data,
+                                             size_t len,
+                                             Error error,
+                                             bool sync)
+    : data(data), len(len), error(error), sync(sync) {}
+
+void MockSourceStream::AddReadResult(const char* data,
+                                     size_t len,
+                                     Error error,
+                                     bool sync) {
+  QueuedResult result(data, len, error, sync);
+  results_.push(result);
+}
+
+void MockSourceStream::CompleteNextRead() {
+  DCHECK(awaiting_completion_);

[mmenke, 2016/08/17 19:01:56]

Suggest a blank line or two here, to increase readability.

[xunjieli, 2016/08/17 20:00:20]

Done.

+  awaiting_completion_ = false;
+  QueuedResult r = results_.front();
+  DCHECK(!r.sync);
+  results_.pop();
+  DCHECK_GE(dest_buffer_size_, r.len);
+  memcpy(dest_buffer_->data(), r.data, r.len);
+  dest_buffer_ = nullptr;
+  callback_.Run(r.error == OK ? r.len : r.error);
+}
+
+}  // namespace net