Index: trunk/src/net/base/mock_file_stream.cc |
=================================================================== |
--- trunk/src/net/base/mock_file_stream.cc (revision 256703) |
+++ trunk/src/net/base/mock_file_stream.cc (working copy) |
@@ -4,46 +4,10 @@ |
#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)); |
@@ -51,12 +15,7 @@ |
int MockFileStream::Seek(Whence whence, int64 offset, |
const Int64CompletionCallback& 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); |
+ return ReturnError(FileStream::Seek(whence, offset, callback)); |
} |
int64 MockFileStream::SeekSync(Whence whence, int64 offset) { |
@@ -70,12 +29,7 @@ |
int MockFileStream::Read(IOBuffer* buf, |
int buf_len, |
const CompletionCallback& 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); |
+ return ReturnError(FileStream::Read(buf, buf_len, callback)); |
} |
int MockFileStream::ReadSync(char* buf, int buf_len) { |
@@ -89,12 +43,7 @@ |
int MockFileStream::Write(IOBuffer* buf, |
int buf_len, |
const CompletionCallback& 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); |
+ return ReturnError(FileStream::Write(buf, buf_len, callback)); |
} |
int MockFileStream::WriteSync(const char* buf, int buf_len) { |
@@ -106,80 +55,13 @@ |
} |
int MockFileStream::Flush(const CompletionCallback& 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); |
+ return ReturnError(FileStream::Flush(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 |