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; |