Fix various issues in RedirectToFileResourceHandler.

net/base/mock_file_stream.cc

 // Copyright (c) 2012 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/base/mock_file_stream.h"
 
+#include "base/bind.h"
+#include "base/message_loop/message_loop.h"
+
 namespace net {
 
 namespace testing {
 
+MockFileStream::MockFileStream(net::NetLog* net_log)
+    : net::FileStream(net_log),
+      forced_error_(net::OK),
+      async_error_(false),
+      throttled_(false),
+      weak_factory_(this) {
+}
+
+MockFileStream::MockFileStream(base::PlatformFile file,
+                               int flags,
+                               net::NetLog* net_log)
+    : net::FileStream(file, flags, net_log),
+      forced_error_(net::OK),
+      async_error_(false),
+      throttled_(false),
+      weak_factory_(this) {
+}
+
+MockFileStream::MockFileStream(
+    base::PlatformFile file,
+    int flags,
+    net::NetLog* net_log,
+    const scoped_refptr<base::TaskRunner>& task_runner)
+    : net::FileStream(file, flags, net_log, task_runner),
+      forced_error_(net::OK),
+      async_error_(false),
+      throttled_(false),
+      weak_factory_(this) {
+}
+
+MockFileStream::~MockFileStream() {
+}
+
 int MockFileStream::OpenSync(const base::FilePath& path, int open_flags) {
   path_ = path;
   return ReturnError(FileStream::OpenSync(path, open_flags));
 }
 
 int MockFileStream::Seek(Whence whence, int64 offset,
                          const Int64CompletionCallback& callback) {
-  return ReturnError(FileStream::Seek(whence, offset, callback));
+  Int64CompletionCallback wrapped_callback =
+      base::Bind(&MockFileStream::DoCallback64,
+                 weak_factory_.GetWeakPtr(), callback);
+  if (forced_error_ == net::OK)
+    return FileStream::Seek(whence, offset, wrapped_callback);
+  return ErrorCallback64(wrapped_callback);
 }
 
 int64 MockFileStream::SeekSync(Whence whence, int64 offset) {
   return ReturnError64(FileStream::SeekSync(whence, offset));
 }
 
 int64 MockFileStream::Available() {
   return ReturnError64(FileStream::Available());
 }
 
 int MockFileStream::Read(IOBuffer* buf,
                          int buf_len,
                          const CompletionCallback& callback) {
-  return ReturnError(FileStream::Read(buf, buf_len, callback));
+  CompletionCallback wrapped_callback = base::Bind(&MockFileStream::DoCallback,
+                                                   weak_factory_.GetWeakPtr(),
+                                                   callback);
+  if (forced_error_ == net::OK)
+    return FileStream::Read(buf, buf_len, wrapped_callback);
+  return ErrorCallback(wrapped_callback);
 }
 
 int MockFileStream::ReadSync(char* buf, int buf_len) {
   return ReturnError(FileStream::ReadSync(buf, buf_len));
 }
 
 int MockFileStream::ReadUntilComplete(char *buf, int buf_len) {
   return ReturnError(FileStream::ReadUntilComplete(buf, buf_len));
 }
 
 int MockFileStream::Write(IOBuffer* buf,
                           int buf_len,
                           const CompletionCallback& callback) {
-  return ReturnError(FileStream::Write(buf, buf_len, callback));
+  CompletionCallback wrapped_callback = base::Bind(&MockFileStream::DoCallback,
+                                                   weak_factory_.GetWeakPtr(),
+                                                   callback);
+  if (forced_error_ == net::OK)
+    return FileStream::Write(buf, buf_len, wrapped_callback);
+  return ErrorCallback(wrapped_callback);
 }
 
 int MockFileStream::WriteSync(const char* buf, int buf_len) {
   return ReturnError(FileStream::WriteSync(buf, buf_len));
 }
 
 int64 MockFileStream::Truncate(int64 bytes) {
   return ReturnError64(FileStream::Truncate(bytes));
 }
 
 int MockFileStream::Flush(const CompletionCallback& callback) {
-  return ReturnError(FileStream::Flush(callback));
+  CompletionCallback wrapped_callback = base::Bind(&MockFileStream::DoCallback,
+                                                   weak_factory_.GetWeakPtr(),
+                                                   callback);
+  if (forced_error_ == net::OK)
+    return FileStream::Flush(wrapped_callback);
+  return ErrorCallback(wrapped_callback);
 }
 
 int MockFileStream::FlushSync() {
   return ReturnError(FileStream::FlushSync());
 }
 
+void MockFileStream::ThrottleCallbacks() {
+  CHECK(!throttled_);
+  throttled_ = true;
+}
+
+void MockFileStream::ReleaseCallbacks() {
+  CHECK(throttled_);
+  throttled_ = false;
+
+  if (!throttled_task_.is_null()) {
+    base::Closure throttled_task = throttled_task_;
+    throttled_task_.Reset();
+    base::MessageLoop::current()->PostTask(FROM_HERE, throttled_task);
+  }
+}
+
+void MockFileStream::DoCallback(const CompletionCallback& callback,
+                                int result) {
+  if (!throttled_) {
+    callback.Run(result);
+    return;
+  }
+  CHECK(throttled_task_.is_null());
+  throttled_task_ = base::Bind(callback, result);
+}
+
+void MockFileStream::DoCallback64(const Int64CompletionCallback& callback,
+                                  int64 result) {
+  if (!throttled_) {
+    callback.Run(result);
+    return;
+  }
+  CHECK(throttled_task_.is_null());
+  throttled_task_ = base::Bind(callback, result);
+}
+
+int MockFileStream::ErrorCallback(const CompletionCallback& callback) {
+  CHECK_NE(net::OK, forced_error_);
+  if (async_error_) {
+    base::MessageLoop::current()->PostTask(
+        FROM_HERE, base::Bind(callback, forced_error_));
+    clear_forced_error();
+    return net::ERR_IO_PENDING;
+  }
+  int ret = forced_error_;
+  clear_forced_error();
+  return ret;
+}
+
+int64 MockFileStream::ErrorCallback64(const Int64CompletionCallback& callback) {
+  CHECK_NE(net::OK, forced_error_);
+  if (async_error_) {
+    base::MessageLoop::current()->PostTask(
+        FROM_HERE, base::Bind(callback, forced_error_));
+    clear_forced_error();
+    return net::ERR_IO_PENDING;
+  }
+  int64 ret = forced_error_;
+  clear_forced_error();
+  return ret;
+}
+
 }  // namespace testing
 
 }  // namespace net