Chromium Code Reviews| 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..47fbba2b7b38aab4659044880d411bca9157a4be |
| --- /dev/null |
| +++ b/content/browser/fileapi/mock_url_request_delegate.cc |
| @@ -0,0 +1,73 @@ |
| +// 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() |
| + : received_data_(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(received_data_.get(), kBufferSize, &bytes_read)) { |
| + if (!request->status().is_io_pending()) { |
|
michaeln
2014/05/29 01:09:04
nit: brackets not needed
falken
2014/05/29 02:10:46
Done.
|
| + RequestComplete(); |
| + } |
| + return; |
| + } |
| + |
| + ReceiveData(request, bytes_read); |
| +} |
| + |
| +void MockURLRequestDelegate::ReceiveData(net::URLRequest* request, |
| + int bytes_read) { |
| + if (bytes_read) { |
| + response_data_.append(received_data_->data(), |
| + static_cast<size_t>(bytes_read)); |
| + ReadSome(request); |
| + } else { |
| + RequestComplete(); |
| + } |
| +} |
| + |
| +void MockURLRequestDelegate::RequestComplete() { |
| + base::MessageLoop::current()->Quit(); |
| +} |
| + |
| +} // namespace |