headless: Allow per-context protocol handlers

headless/test/test_url_request_job.cc

+// 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.
+
+#include "headless/test/test_url_request_job.h"
+
+#include "net/base/io_buffer.h"
+#include "net/http/http_response_headers.h"
+
+namespace headless {
+
+TestURLRequestJob::TestURLRequestJob(net::URLRequest* request,
+                                     net::NetworkDelegate* network_delegate,
+                                     const std::string& body)
+    : net::URLRequestJob(request, network_delegate),
+      body_(new net::StringIOBuffer(body)),
+      src_buf_(new net::DrainableIOBuffer(body_.get(), body_->size())) {}
+
+TestURLRequestJob::~TestURLRequestJob() {}
+
+void TestURLRequestJob::Start() {
+  NotifyHeadersComplete();
+}
+
+void TestURLRequestJob::GetResponseInfo(net::HttpResponseInfo* info) {
+  info->headers =
+      new net::HttpResponseHeaders("Content-Type: text/html\r\n\r\n");
+}
+
+int TestURLRequestJob::ReadRawData(net::IOBuffer* buf, int buf_size) {
+  scoped_refptr<net::DrainableIOBuffer> dest_buf(
+      new net::DrainableIOBuffer(buf, buf_size));
+  while (src_buf_->BytesRemaining() > 0 && dest_buf->BytesRemaining() > 0) {
+    *dest_buf->data() = *src_buf_->data();
+    src_buf_->DidConsume(1);
+    dest_buf->DidConsume(1);
+  }
+  return dest_buf->BytesConsumed();
+}
+
+}  // namespace headless