| 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 3a4c47e93eb758cd66dac8d55b1a67cf9aad35e1..fee38393fed9989e0a494da8d602836a28bd2f67 100644
|
| --- a/content/browser/fileapi/fileapi_message_filter_unittest.cc
|
| +++ b/content/browser/fileapi/fileapi_message_filter_unittest.cc
|
| @@ -11,6 +11,7 @@
|
| #include "base/memory/shared_memory.h"
|
| #include "base/message_loop/message_loop.h"
|
| #include "base/process/process.h"
|
| +#include "content/browser/browser_message_filter_peer.h"
|
| #include "content/browser/child_process_security_policy_impl.h"
|
| #include "content/browser/fileapi/chrome_blob_storage_context.h"
|
| #include "content/browser/streams/stream_registry.h"
|
| @@ -84,6 +85,7 @@ class FileAPIMessageFilterTest : public testing::Test {
|
| ChromeBlobStorageContext* blob_storage_context_;
|
|
|
| scoped_refptr<FileAPIMessageFilter> filter_;
|
| + scoped_ptr<BrowserMessageFilterPeer> filter_peer_;
|
| };
|
|
|
| TEST_F(FileAPIMessageFilterTest, CloseChannelWithInflightRequest) {
|
| @@ -94,7 +96,7 @@ TEST_F(FileAPIMessageFilterTest, CloseChannelWithInflightRequest) {
|
| file_system_context_.get(),
|
| ChromeBlobStorageContext::GetFor(&browser_context_),
|
| StreamContext::GetFor(&browser_context_)));
|
| - filter->OnChannelConnected(0);
|
| + filter->OnSenderConnected();
|
|
|
| // Complete initialization.
|
| message_loop_.RunUntilIdle();
|
| @@ -105,7 +107,7 @@ TEST_F(FileAPIMessageFilterTest, CloseChannelWithInflightRequest) {
|
| EXPECT_TRUE(filter->OnMessageReceived(read_metadata));
|
|
|
| // Close the filter while it has inflight request.
|
| - filter->OnChannelClosing();
|
| + filter->OnSenderClosing();
|
|
|
| // This shouldn't cause DCHECK failure.
|
| message_loop_.RunUntilIdle();
|
| @@ -126,8 +128,8 @@ TEST_F(FileAPIMessageFilterTest, MultipleFilters) {
|
| file_system_context_.get(),
|
| ChromeBlobStorageContext::GetFor(&browser_context_),
|
| StreamContext::GetFor(&browser_context_)));
|
| - filter1->OnChannelConnected(0);
|
| - filter2->OnChannelConnected(1);
|
| + filter1->OnSenderConnected();
|
| + filter2->OnSenderConnected();
|
|
|
| // Complete initialization.
|
| message_loop_.RunUntilIdle();
|
| @@ -138,7 +140,7 @@ TEST_F(FileAPIMessageFilterTest, MultipleFilters) {
|
| EXPECT_TRUE(filter1->OnMessageReceived(read_metadata));
|
|
|
| // Close the other filter before the request for filter1 is processed.
|
| - filter2->OnChannelClosing();
|
| + filter2->OnSenderClosing();
|
|
|
| // This shouldn't cause DCHECK failure.
|
| message_loop_.RunUntilIdle();
|
| @@ -236,7 +238,8 @@ TEST_F(FileAPIMessageFilterTest, BuildStreamWithSharedMemory) {
|
| // OnAppendSharedMemoryToStream passes the peer process's handle to
|
| // SharedMemory's constructor. If it's incorrect, DuplicateHandle won't work
|
| // correctly.
|
| - filter_->set_peer_pid_for_testing(base::Process::Current().pid());
|
| + filter_peer_.reset(new BrowserMessageFilterPeer(base::GetCurrentProcId()));
|
| + filter_->OnFilterAddedToHost(filter_peer_.get());
|
|
|
| StreamHostMsg_StartBuilding start_message(kUrl, kFakeContentType);
|
| EXPECT_TRUE(filter_->OnMessageReceived(start_message));
|
| @@ -272,7 +275,7 @@ TEST_F(FileAPIMessageFilterTest, BuildStreamWithSharedMemory) {
|
| EXPECT_EQ(0, bytes_read);
|
| }
|
|
|
| -TEST_F(FileAPIMessageFilterTest, BuildStreamAndCallOnChannelClosing) {
|
| +TEST_F(FileAPIMessageFilterTest, BuildStreamAndCallOnSenderClosing) {
|
| StreamRegistry* stream_registry = stream_context_->registry();
|
|
|
| const GURL kUrl(kFakeBlobInternalUrlSpec);
|
| @@ -282,7 +285,7 @@ TEST_F(FileAPIMessageFilterTest, BuildStreamAndCallOnChannelClosing) {
|
|
|
| ASSERT_FALSE(stream_registry->GetStream(kUrl).get() == NULL);
|
|
|
| - filter_->OnChannelClosing();
|
| + filter_->OnSenderClosing();
|
|
|
| ASSERT_EQ(NULL, stream_registry->GetStream(kUrl).get());
|
| }
|
|
|