Index: net/base/file_stream_unittest.cc |
diff --git a/net/base/file_stream_unittest.cc b/net/base/file_stream_unittest.cc |
index fc9257e5d1c0af8d6f541191f02415862cc57da3..8b47eccf4144185da0a990c7f60a85fa848b7080 100644 |
--- a/net/base/file_stream_unittest.cc |
+++ b/net/base/file_stream_unittest.cc |
@@ -546,6 +546,25 @@ class TestWriteReadCompletionCallback { |
const CompletionCallback& callback() const { return callback_; } |
+ void ValidateWrittenData() { |
+ TestCompletionCallback callback; |
+ int rv = 0; |
+ for (;;) { |
+ scoped_refptr<IOBufferWithSize> buf = new IOBufferWithSize(4); |
+ rv = stream_->Read(buf.get(), buf->size(), callback.callback()); |
+ if (rv == ERR_IO_PENDING) { |
+ base::MessageLoop::ScopedNestableTaskAllower allow( |
+ base::MessageLoop::current()); |
+ rv = callback.WaitForResult(); |
+ } |
+ EXPECT_LE(0, rv); |
+ if (rv <= 0) |
+ break; |
+ *total_bytes_read_ += rv; |
+ data_read_->append(buf->data(), rv); |
+ } |
+ } |
+ |
private: |
void OnComplete(int result) { |
DCHECK_LT(0, result); |
@@ -577,22 +596,6 @@ class TestWriteReadCompletionCallback { |
base::MessageLoop::current()); |
EXPECT_LE(0, callback64.WaitForResult()); |
} |
- |
- TestCompletionCallback callback; |
- for (;;) { |
- scoped_refptr<IOBufferWithSize> buf = new IOBufferWithSize(4); |
- rv = stream_->Read(buf.get(), buf->size(), callback.callback()); |
- if (rv == ERR_IO_PENDING) { |
- base::MessageLoop::ScopedNestableTaskAllower allow( |
- base::MessageLoop::current()); |
- rv = callback.WaitForResult(); |
- } |
- EXPECT_LE(0, rv); |
- if (rv <= 0) |
- break; |
- *total_bytes_read_ += rv; |
- data_read_->append(buf->data(), rv); |
- } |
} |
result_ = *total_bytes_written_; |
@@ -646,6 +649,8 @@ TEST_F(FileStreamTest, WriteRead) { |
EXPECT_LT(0, rv); |
EXPECT_EQ(kTestDataSize, total_bytes_written); |
+ callback.ValidateWrittenData(); |
+ |
stream.reset(); |
EXPECT_TRUE(base::GetFileSize(temp_file_path(), &file_size)); |