| Index: content/browser/fileapi/fileapi_message_filter_unittest.cc
|
| diff --git a/content/browser/fileapi/fileapi_message_filter_unittest.cc b/content/browser/fileapi/fileapi_message_filter_unittest.cc
|
| index 2115180e606bd48587ad80f06332d5e111a03143..61590a94791e45986b64990769999c1193b3b5e4 100644
|
| --- a/content/browser/fileapi/fileapi_message_filter_unittest.cc
|
| +++ b/content/browser/fileapi/fileapi_message_filter_unittest.cc
|
| @@ -18,11 +18,14 @@
|
| #include "content/browser/streams/stream_registry.h"
|
| #include "content/common/fileapi/file_system_messages.h"
|
| #include "content/common/fileapi/webblob_messages.h"
|
| +#include "content/public/browser/browser_context.h"
|
| #include "content/public/browser/browser_thread.h"
|
| +#include "content/public/browser/storage_partition.h"
|
| #include "content/public/common/common_param_traits.h"
|
| #include "content/public/test/mock_render_process_host.h"
|
| #include "content/public/test/test_browser_context.h"
|
| #include "content/public/test/test_browser_thread.h"
|
| +#include "content/public/test/test_browser_thread_bundle.h"
|
| #include "content/public/test/test_file_system_context.h"
|
| #include "net/base/io_buffer.h"
|
| #include "storage/browser/blob/blob_storage_context.h"
|
| @@ -46,7 +49,7 @@ const char kFakeContentType[] = "fake/type";
|
| class FileAPIMessageFilterTest : public testing::Test {
|
| public:
|
| FileAPIMessageFilterTest()
|
| - : io_browser_thread_(BrowserThread::IO, &message_loop_) {
|
| + : browser_thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP) {
|
| }
|
|
|
| protected:
|
| @@ -68,18 +71,17 @@ class FileAPIMessageFilterTest : public testing::Test {
|
|
|
| filter_ = new FileAPIMessageFilter(
|
| 0 /* process_id */,
|
| - browser_context_.GetRequestContext(),
|
| + BrowserContext::GetDefaultStoragePartition(&browser_context_)->
|
| + GetURLRequestContext(),
|
| file_system_context_.get(),
|
| blob_storage_context_,
|
| stream_context_);
|
|
|
| // Complete initialization.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| - base::MessageLoop message_loop_;
|
| - TestBrowserThread io_browser_thread_;
|
| -
|
| + TestBrowserThreadBundle browser_thread_bundle_;
|
| TestBrowserContext browser_context_;
|
| scoped_refptr<storage::FileSystemContext> file_system_context_;
|
| StreamContext* stream_context_;
|
| @@ -92,14 +94,15 @@ TEST_F(FileAPIMessageFilterTest, CloseChannelWithInflightRequest) {
|
| scoped_refptr<FileAPIMessageFilter> filter(
|
| new FileAPIMessageFilter(
|
| 0 /* process_id */,
|
| - browser_context_.GetRequestContext(),
|
| + BrowserContext::GetDefaultStoragePartition(&browser_context_)->
|
| + GetURLRequestContext(),
|
| file_system_context_.get(),
|
| ChromeBlobStorageContext::GetFor(&browser_context_),
|
| StreamContext::GetFor(&browser_context_)));
|
| filter->OnChannelConnected(0);
|
|
|
| // Complete initialization.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
|
|
| int request_id = 0;
|
| const GURL kUrl("filesystem:http://example.com/temporary/foo");
|
| @@ -110,21 +113,23 @@ TEST_F(FileAPIMessageFilterTest, CloseChannelWithInflightRequest) {
|
| filter->OnChannelClosing();
|
|
|
| // This shouldn't cause DCHECK failure.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| TEST_F(FileAPIMessageFilterTest, MultipleFilters) {
|
| scoped_refptr<FileAPIMessageFilter> filter1(
|
| new FileAPIMessageFilter(
|
| 0 /* process_id */,
|
| - browser_context_.GetRequestContext(),
|
| + BrowserContext::GetDefaultStoragePartition(&browser_context_)->
|
| + GetURLRequestContext(),
|
| file_system_context_.get(),
|
| ChromeBlobStorageContext::GetFor(&browser_context_),
|
| StreamContext::GetFor(&browser_context_)));
|
| scoped_refptr<FileAPIMessageFilter> filter2(
|
| new FileAPIMessageFilter(
|
| 1 /* process_id */,
|
| - browser_context_.GetRequestContext(),
|
| + BrowserContext::GetDefaultStoragePartition(&browser_context_)->
|
| + GetURLRequestContext(),
|
| file_system_context_.get(),
|
| ChromeBlobStorageContext::GetFor(&browser_context_),
|
| StreamContext::GetFor(&browser_context_)));
|
| @@ -132,7 +137,7 @@ TEST_F(FileAPIMessageFilterTest, MultipleFilters) {
|
| filter2->OnChannelConnected(1);
|
|
|
| // Complete initialization.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
|
|
| int request_id = 0;
|
| const GURL kUrl("filesystem:http://example.com/temporary/foo");
|
| @@ -143,7 +148,7 @@ TEST_F(FileAPIMessageFilterTest, MultipleFilters) {
|
| filter2->OnChannelClosing();
|
|
|
| // This shouldn't cause DCHECK failure.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
| }
|
|
|
| TEST_F(FileAPIMessageFilterTest, BuildEmptyStream) {
|
| @@ -179,7 +184,7 @@ TEST_F(FileAPIMessageFilterTest, BuildEmptyStream) {
|
| EXPECT_EQ(0, bytes_read);
|
|
|
| // Run loop to finish transfer.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
|
|
| EXPECT_EQ(Stream::STREAM_COMPLETE,
|
| stream->ReadRawData(buffer.get(), kBufferSize, &bytes_read));
|
| @@ -209,7 +214,7 @@ TEST_F(FileAPIMessageFilterTest, BuildNonEmptyStream) {
|
| EXPECT_TRUE(filter_->OnMessageReceived(finish_message));
|
|
|
| // Run loop to finish transfer and commit finalize command.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
|
|
| scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(kFakeData.size()));
|
| int bytes_read = 0;
|
| @@ -256,7 +261,7 @@ TEST_F(FileAPIMessageFilterTest, BuildStreamWithSharedMemory) {
|
| EXPECT_TRUE(filter_->OnMessageReceived(finish_message));
|
|
|
| // Run loop to finish transfer and commit finalize command.
|
| - message_loop_.RunUntilIdle();
|
| + base::MessageLoop::current()->RunUntilIdle();
|
|
|
| scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(kFakeData.size()));
|
| int bytes_read = 0;
|
|
|