Index: extensions/browser/api/mime_handler_private/mime_handler_private_unittest.cc |
diff --git a/extensions/browser/api/mime_handler_private/mime_handler_private_unittest.cc b/extensions/browser/api/mime_handler_private/mime_handler_private_unittest.cc |
index d8e1035097b536a1481d3c5247d7c312c75889e4..47df5cb4440ea2c14df15f9bdc439c2999082d7e 100644 |
--- a/extensions/browser/api/mime_handler_private/mime_handler_private_unittest.cc |
+++ b/extensions/browser/api/mime_handler_private/mime_handler_private_unittest.cc |
@@ -12,6 +12,7 @@ |
#include "content/public/browser/stream_handle.h" |
#include "content/public/browser/stream_info.h" |
#include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.h" |
+#include "mojo/public/cpp/bindings/strong_binding.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace extensions { |
@@ -47,11 +48,13 @@ class MimeHandlerServiceImplTest : public testing::Test { |
stream_info->original_url = GURL("test://extensions_unittests"); |
stream_container_.reset( |
new StreamContainer(std::move(stream_info), 1, true, GURL(), "")); |
- service_.reset(new MimeHandlerServiceImpl(stream_container_->GetWeakPtr(), |
- mojo::GetProxy(&service_ptr_))); |
+ service_binding_ = |
+ mojo::MakeStrongBinding(base::MakeUnique<MimeHandlerServiceImpl>( |
+ stream_container_->GetWeakPtr()), |
+ mojo::GetProxy(&service_ptr_)); |
} |
void TearDown() override { |
- service_.reset(); |
+ service_binding_->Close(); |
stream_container_.reset(); |
} |
@@ -63,30 +66,30 @@ class MimeHandlerServiceImplTest : public testing::Test { |
base::MessageLoop message_loop_; |
std::unique_ptr<StreamContainer> stream_container_; |
mime_handler::MimeHandlerServicePtr service_ptr_; |
- std::unique_ptr<mime_handler::MimeHandlerService> service_; |
+ mojo::StrongBindingPtr<mime_handler::MimeHandlerService> service_binding_; |
bool abort_called_ = false; |
mime_handler::StreamInfoPtr stream_info_; |
}; |
TEST_F(MimeHandlerServiceImplTest, Abort) { |
- service_->AbortStream(base::Bind(&MimeHandlerServiceImplTest::AbortCallback, |
- base::Unretained(this))); |
+ service_binding_->impl()->AbortStream(base::Bind( |
+ &MimeHandlerServiceImplTest::AbortCallback, base::Unretained(this))); |
EXPECT_TRUE(abort_called_); |
abort_called_ = false; |
- service_->AbortStream(base::Bind(&MimeHandlerServiceImplTest::AbortCallback, |
- base::Unretained(this))); |
+ service_binding_->impl()->AbortStream(base::Bind( |
+ &MimeHandlerServiceImplTest::AbortCallback, base::Unretained(this))); |
EXPECT_TRUE(abort_called_); |
stream_container_.reset(); |
abort_called_ = false; |
- service_->AbortStream(base::Bind(&MimeHandlerServiceImplTest::AbortCallback, |
- base::Unretained(this))); |
+ service_binding_->impl()->AbortStream(base::Bind( |
+ &MimeHandlerServiceImplTest::AbortCallback, base::Unretained(this))); |
EXPECT_TRUE(abort_called_); |
} |
TEST_F(MimeHandlerServiceImplTest, GetStreamInfo) { |
- service_->GetStreamInfo( |
+ service_binding_->impl()->GetStreamInfo( |
base::Bind(&MimeHandlerServiceImplTest::GetStreamInfoCallback, |
base::Unretained(this))); |
ASSERT_TRUE(stream_info_); |
@@ -96,16 +99,16 @@ TEST_F(MimeHandlerServiceImplTest, GetStreamInfo) { |
EXPECT_EQ("test://extensions_unittests", stream_info_->original_url); |
EXPECT_EQ("stream://url", stream_info_->stream_url); |
- service_->AbortStream(base::Bind(&MimeHandlerServiceImplTest::AbortCallback, |
- base::Unretained(this))); |
+ service_binding_->impl()->AbortStream(base::Bind( |
+ &MimeHandlerServiceImplTest::AbortCallback, base::Unretained(this))); |
EXPECT_TRUE(abort_called_); |
- service_->GetStreamInfo( |
+ service_binding_->impl()->GetStreamInfo( |
base::Bind(&MimeHandlerServiceImplTest::GetStreamInfoCallback, |
base::Unretained(this))); |
ASSERT_FALSE(stream_info_); |
stream_container_.reset(); |
- service_->GetStreamInfo( |
+ service_binding_->impl()->GetStreamInfo( |
base::Bind(&MimeHandlerServiceImplTest::GetStreamInfoCallback, |
base::Unretained(this))); |
ASSERT_FALSE(stream_info_); |