Index: content/browser/fileapi/mock_url_request_delegate.cc |
diff --git a/content/browser/fileapi/mock_url_request_delegate.cc b/content/browser/fileapi/mock_url_request_delegate.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..97e09f0df6f6285601f55bf3f98337f6399302d4 |
--- /dev/null |
+++ b/content/browser/fileapi/mock_url_request_delegate.cc |
@@ -0,0 +1,72 @@ |
+// Copyright 2014 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 "mock_url_request_delegate.h" |
+ |
+#include "base/run_loop.h" |
+#include "net/base/io_buffer.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+namespace { |
+const int kBufferSize = 1024; |
+} |
+ |
+namespace content { |
+ |
+MockURLRequestDelegate::MockURLRequestDelegate() |
+ : io_buffer_(new net::IOBuffer(kBufferSize)) { |
+} |
+ |
+MockURLRequestDelegate::~MockURLRequestDelegate() { |
+} |
+ |
+void MockURLRequestDelegate::OnResponseStarted(net::URLRequest* request) { |
+ if (request->status().is_success()) { |
+ EXPECT_TRUE(request->response_headers()); |
+ ReadSome(request); |
+ } else { |
+ RequestComplete(); |
+ } |
+} |
+ |
+void MockURLRequestDelegate::OnReadCompleted(net::URLRequest* request, |
+ int bytes_read) { |
+ if (bytes_read > 0) |
+ ReceiveData(request, bytes_read); |
+ else |
+ RequestComplete(); |
+} |
+ |
+void MockURLRequestDelegate::ReadSome(net::URLRequest* request) { |
+ if (!request->is_pending()) { |
+ RequestComplete(); |
+ return; |
+ } |
+ |
+ int bytes_read = 0; |
+ if (!request->Read(io_buffer_.get(), kBufferSize, &bytes_read)) { |
+ if (!request->status().is_io_pending()) |
+ RequestComplete(); |
+ return; |
+ } |
+ |
+ ReceiveData(request, bytes_read); |
+} |
+ |
+void MockURLRequestDelegate::ReceiveData(net::URLRequest* request, |
+ int bytes_read) { |
+ if (bytes_read) { |
+ response_data_.append(io_buffer_->data(), |
+ static_cast<size_t>(bytes_read)); |
+ ReadSome(request); |
+ } else { |
+ RequestComplete(); |
+ } |
+} |
+ |
+void MockURLRequestDelegate::RequestComplete() { |
+ base::MessageLoop::current()->Quit(); |
+} |
+ |
+} // namespace |