Index: net/base/file_stream_unittest.cc |
diff --git a/net/base/file_stream_unittest.cc b/net/base/file_stream_unittest.cc |
index a091e7a42aab291ded75234a57a9945a0f5e8220..ee0bbcba4067c425dedefccb3507ed8c75ce39e2 100644 |
--- a/net/base/file_stream_unittest.cc |
+++ b/net/base/file_stream_unittest.cc |
@@ -10,7 +10,9 @@ |
#include "base/message_loop/message_loop.h" |
#include "base/path_service.h" |
#include "base/platform_file.h" |
+#include "base/run_loop.h" |
#include "base/synchronization/waitable_event.h" |
+#include "base/task_runner.h" |
#include "base/test/test_timeouts.h" |
#include "net/base/capturing_net_log.h" |
#include "net/base/io_buffer.h" |
@@ -46,11 +48,16 @@ class FileStreamTest : public PlatformTest { |
virtual void TearDown() { |
EXPECT_TRUE(base::DeleteFile(temp_file_path_, false)); |
+ base::RunLoop().RunUntilIdle(); |
Ryan Sleevi
2013/09/06 23:27:28
I'm not really thrilled with adding this to test s
earthdok
2013/09/07 00:20:03
The tasks are posted from async FileStreams on des
|
PlatformTest::TearDown(); |
} |
const base::FilePath temp_file_path() const { return temp_file_path_; } |
+ scoped_refptr<base::TaskRunner> file_task_runner() const { |
+ return base::MessageLoop::current()->message_loop_proxy(); |
Ryan Sleevi
2013/09/06 23:27:28
Just drop this method and use base::MessageLoopPro
|
+ } |
+ |
private: |
base::FilePath temp_file_path_; |
}; |
@@ -60,7 +67,7 @@ namespace { |
TEST_F(FileStreamTest, BasicOpenClose) { |
base::PlatformFile file = base::kInvalidPlatformFileValue; |
{ |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int rv = stream.OpenSync(temp_file_path(), |
base::PLATFORM_FILE_OPEN | base::PLATFORM_FILE_READ); |
EXPECT_EQ(OK, rv); |
@@ -83,7 +90,7 @@ TEST_F(FileStreamTest, FileHandleNotLeftOpen) { |
{ |
// Seek to the beginning of the file and read. |
- FileStream read_stream(file, flags, NULL); |
+ FileStream read_stream(file, flags, NULL, file_task_runner()); |
EXPECT_TRUE(read_stream.IsOpen()); |
} |
@@ -105,7 +112,8 @@ TEST_F(FileStreamTest, UseFileHandle) { |
temp_file_path(), flags, &created, NULL); |
// Seek to the beginning of the file and read. |
- scoped_ptr<FileStream> read_stream(new FileStream(file, flags, NULL)); |
+ scoped_ptr<FileStream> read_stream( |
+ new FileStream(file, flags, NULL, file_task_runner())); |
ASSERT_EQ(0, read_stream->SeekSync(FROM_BEGIN, 0)); |
ASSERT_EQ(kTestDataSize, read_stream->Available()); |
// Read into buffer and compare. |
@@ -120,7 +128,8 @@ TEST_F(FileStreamTest, UseFileHandle) { |
flags = base::PLATFORM_FILE_OPEN_ALWAYS | base::PLATFORM_FILE_WRITE; |
file = base::CreatePlatformFile(temp_file_path(), flags, &created, NULL); |
- scoped_ptr<FileStream> write_stream(new FileStream(file, flags, NULL)); |
+ scoped_ptr<FileStream> write_stream( |
+ new FileStream(file, flags, NULL, file_task_runner())); |
ASSERT_EQ(0, write_stream->SeekSync(FROM_BEGIN, 0)); |
ASSERT_EQ(kTestDataSize, |
write_stream->WriteSync(kTestData, kTestDataSize)); |
@@ -133,7 +142,7 @@ TEST_F(FileStreamTest, UseFileHandle) { |
} |
TEST_F(FileStreamTest, UseClosedStream) { |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
EXPECT_FALSE(stream.IsOpen()); |
@@ -156,7 +165,7 @@ TEST_F(FileStreamTest, BasicRead) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ; |
int rv = stream.OpenSync(temp_file_path(), flags); |
@@ -186,7 +195,7 @@ TEST_F(FileStreamTest, AsyncRead) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_ASYNC; |
@@ -221,7 +230,7 @@ TEST_F(FileStreamTest, AsyncRead_EarlyDelete) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_ASYNC; |
@@ -239,7 +248,7 @@ TEST_F(FileStreamTest, AsyncRead_EarlyDelete) { |
if (rv < 0) { |
EXPECT_EQ(ERR_IO_PENDING, rv); |
// The callback should not be called if the request is cancelled. |
- base::MessageLoop::current()->RunUntilIdle(); |
+ base::RunLoop().RunUntilIdle(); |
EXPECT_FALSE(callback.have_result()); |
} else { |
EXPECT_EQ(std::string(kTestData, rv), std::string(buf->data(), rv)); |
@@ -251,7 +260,7 @@ TEST_F(FileStreamTest, BasicRead_FromOffset) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ; |
int rv = stream.OpenSync(temp_file_path(), flags); |
@@ -286,7 +295,7 @@ TEST_F(FileStreamTest, AsyncRead_FromOffset) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_ASYNC; |
@@ -324,7 +333,7 @@ TEST_F(FileStreamTest, AsyncRead_FromOffset) { |
} |
TEST_F(FileStreamTest, SeekAround) { |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ; |
int rv = stream.OpenSync(temp_file_path(), flags); |
@@ -347,7 +356,7 @@ TEST_F(FileStreamTest, SeekAround) { |
} |
TEST_F(FileStreamTest, AsyncSeekAround) { |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_ASYNC | |
base::PLATFORM_FILE_READ; |
@@ -383,7 +392,7 @@ TEST_F(FileStreamTest, AsyncSeekAround) { |
} |
TEST_F(FileStreamTest, BasicWrite) { |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_CREATE_ALWAYS | |
base::PLATFORM_FILE_WRITE; |
int rv = stream->OpenSync(temp_file_path(), flags); |
@@ -404,7 +413,7 @@ TEST_F(FileStreamTest, BasicWrite) { |
} |
TEST_F(FileStreamTest, AsyncWrite) { |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_CREATE_ALWAYS | |
base::PLATFORM_FILE_WRITE | |
base::PLATFORM_FILE_ASYNC; |
@@ -440,7 +449,7 @@ TEST_F(FileStreamTest, AsyncWrite) { |
} |
TEST_F(FileStreamTest, AsyncWrite_EarlyDelete) { |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_CREATE_ALWAYS | |
base::PLATFORM_FILE_WRITE | |
base::PLATFORM_FILE_ASYNC; |
@@ -460,7 +469,7 @@ TEST_F(FileStreamTest, AsyncWrite_EarlyDelete) { |
if (rv < 0) { |
EXPECT_EQ(ERR_IO_PENDING, rv); |
// The callback should not be called if the request is cancelled. |
- base::MessageLoop::current()->RunUntilIdle(); |
+ base::RunLoop().RunUntilIdle(); |
EXPECT_FALSE(callback.have_result()); |
} else { |
ok = file_util::GetFileSize(temp_file_path(), &file_size); |
@@ -470,7 +479,7 @@ TEST_F(FileStreamTest, AsyncWrite_EarlyDelete) { |
} |
TEST_F(FileStreamTest, BasicWrite_FromOffset) { |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_WRITE; |
int rv = stream->OpenSync(temp_file_path(), flags); |
@@ -499,7 +508,7 @@ TEST_F(FileStreamTest, AsyncWrite_FromOffset) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- FileStream stream(NULL); |
+ FileStream stream(NULL, file_task_runner()); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_WRITE | |
base::PLATFORM_FILE_ASYNC; |
@@ -541,7 +550,7 @@ TEST_F(FileStreamTest, BasicReadWrite) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_WRITE; |
@@ -580,7 +589,7 @@ TEST_F(FileStreamTest, BasicWriteRead) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_WRITE; |
@@ -628,7 +637,7 @@ TEST_F(FileStreamTest, BasicAsyncReadWrite) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_WRITE | |
@@ -687,7 +696,7 @@ TEST_F(FileStreamTest, BasicAsyncWriteRead) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_WRITE | |
@@ -778,7 +787,7 @@ class TestWriteReadCompletionCallback { |
DCHECK(!waiting_for_result_); |
while (!have_result_) { |
waiting_for_result_ = true; |
- base::MessageLoop::current()->Run(); |
+ base::RunLoop().Run(); |
waiting_for_result_ = false; |
} |
have_result_ = false; // auto-reset for next callback |
@@ -853,7 +862,7 @@ TEST_F(FileStreamTest, AsyncWriteRead) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_WRITE | |
@@ -911,7 +920,7 @@ class TestWriteCloseCompletionCallback { |
DCHECK(!waiting_for_result_); |
while (!have_result_) { |
waiting_for_result_ = true; |
- base::MessageLoop::current()->Run(); |
+ base::RunLoop().Run(); |
waiting_for_result_ = false; |
} |
have_result_ = false; // auto-reset for next callback |
@@ -962,7 +971,7 @@ TEST_F(FileStreamTest, AsyncWriteClose) { |
bool ok = file_util::GetFileSize(temp_file_path(), &file_size); |
EXPECT_TRUE(ok); |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_WRITE | |
@@ -999,7 +1008,7 @@ TEST_F(FileStreamTest, AsyncWriteClose) { |
TEST_F(FileStreamTest, Truncate) { |
int flags = base::PLATFORM_FILE_CREATE_ALWAYS | base::PLATFORM_FILE_WRITE; |
- scoped_ptr<FileStream> write_stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> write_stream(new FileStream(NULL, file_task_runner())); |
ASSERT_EQ(OK, write_stream->OpenSync(temp_file_path(), flags)); |
// Write some data to the file. |
@@ -1023,7 +1032,7 @@ TEST_F(FileStreamTest, Truncate) { |
} |
TEST_F(FileStreamTest, AsyncOpenAndDelete) { |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_WRITE | |
base::PLATFORM_FILE_ASYNC; |
@@ -1035,13 +1044,13 @@ TEST_F(FileStreamTest, AsyncOpenAndDelete) { |
// complete. Should be safe. |
stream.reset(); |
// open_callback won't be called. |
- base::MessageLoop::current()->RunUntilIdle(); |
+ base::RunLoop().RunUntilIdle(); |
EXPECT_FALSE(open_callback.have_result()); |
} |
// Verify that async Write() errors are mapped correctly. |
TEST_F(FileStreamTest, AsyncWriteError) { |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_CREATE_ALWAYS | |
base::PLATFORM_FILE_WRITE | |
base::PLATFORM_FILE_ASYNC; |
@@ -1060,7 +1069,7 @@ TEST_F(FileStreamTest, AsyncWriteError) { |
// Verify that async Read() errors are mapped correctly. |
TEST_F(FileStreamTest, AsyncReadError) { |
- scoped_ptr<FileStream> stream(new FileStream(NULL)); |
+ scoped_ptr<FileStream> stream(new FileStream(NULL, file_task_runner())); |
int flags = base::PLATFORM_FILE_OPEN | |
base::PLATFORM_FILE_READ | |
base::PLATFORM_FILE_ASYNC; |