Chromium Code Reviews| 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; |