Index: chrome/browser/media/router/mojo/media_router_mojo_impl_unittest.cc |
diff --git a/chrome/browser/media/router/mojo/media_router_mojo_impl_unittest.cc b/chrome/browser/media/router/mojo/media_router_mojo_impl_unittest.cc |
index 2d18b4558b9cad21ef01775a4b5ee0ac0435ef0d..8668d036eebd15c9e5e253a96eddfc20cff3fa6c 100644 |
--- a/chrome/browser/media/router/mojo/media_router_mojo_impl_unittest.cc |
+++ b/chrome/browser/media/router/mojo/media_router_mojo_impl_unittest.cc |
@@ -187,7 +187,9 @@ class TestProcessManager : public extensions::ProcessManager { |
// Mockable class for awaiting RegisterMediaRouteProvider callbacks. |
class RegisterMediaRouteProviderHandler { |
public: |
- MOCK_METHOD1(Invoke, void(const std::string& instance_id)); |
+ MOCK_METHOD2(Invoke, |
+ void(const std::string& instance_id, |
+ bool enable_browser_side_discovery)); |
}; |
TEST_F(MediaRouterMojoImplTest, CreateRoute) { |
@@ -1411,7 +1413,8 @@ class MediaRouterMojoExtensionTest : public ::testing::Test { |
DCHECK(process_manager_); |
// Create MR and its proxy, so that it can be accessed through Mojo. |
- media_router_.reset(new MediaRouterMojoImpl(process_manager_)); |
+ media_router_.reset( |
+ new MediaRouterMojoImpl(process_manager_, profile_.get())); |
ProcessEventLoop(); |
} |
@@ -1496,7 +1499,7 @@ TEST_F(MediaRouterMojoExtensionTest, DeferredBindingAndSuspension) { |
// itself via RegisterMediaRouteProvider(). |
// Now that the |media_router| and |mojo_media_router| are fully initialized, |
// the queued DetachRoute() call should be executed. |
- EXPECT_CALL(provide_handler_, Invoke(testing::Not(""))) |
+ EXPECT_CALL(provide_handler_, Invoke(testing::Not(""), false)) |
.WillOnce(InvokeWithoutArgs([&run_loop]() { run_loop.Quit(); })); |
EXPECT_CALL(*process_manager_, IsEventPageSuspended(extension_->id())) |
.WillOnce(Return(false)); |
@@ -1524,7 +1527,7 @@ TEST_F(MediaRouterMojoExtensionTest, DeferredBindingAndSuspension) { |
base::RunLoop run_loop4, run_loop5; |
// RegisterMediaRouteProvider() is called. |
// The queued DetachRoute(kRouteId2) call should be executed. |
- EXPECT_CALL(provide_handler_, Invoke(testing::Not(""))) |
+ EXPECT_CALL(provide_handler_, Invoke(testing::Not(""), false)) |
.WillOnce(InvokeWithoutArgs([&run_loop4]() { run_loop4.Quit(); })); |
EXPECT_CALL(*process_manager_, IsEventPageSuspended(extension_->id())) |
.WillOnce(Return(false)); |
@@ -1634,7 +1637,7 @@ TEST_F(MediaRouterMojoExtensionTest, DropOldestPendingRequest) { |
// The oldest request should have been dropped, so we don't expect to see |
// DetachRoute(kRouteId) here. |
BindMediaRouteProvider(); |
- EXPECT_CALL(provide_handler_, Invoke(testing::Not(""))) |
+ EXPECT_CALL(provide_handler_, Invoke(testing::Not(""), false)) |
.WillOnce(InvokeWithoutArgs([&run_loop]() { run_loop.Quit(); })); |
EXPECT_CALL(*process_manager_, IsEventPageSuspended(extension_->id())); |
EXPECT_CALL(mock_media_route_provider_, EnableMdnsDiscovery()) |
@@ -1731,10 +1734,8 @@ TEST_F(MediaRouterMojoExtensionTest, UpdateMediaSinksOnUserGesture) { |
BindMediaRouteProvider(); |
base::RunLoop run_loop; |
- EXPECT_CALL(provide_handler_, Invoke(testing::Not(""))) |
- .WillOnce(InvokeWithoutArgs([&run_loop]() { |
- run_loop.Quit(); |
- })); |
+ EXPECT_CALL(provide_handler_, Invoke(testing::Not(""), false)) |
+ .WillOnce(InvokeWithoutArgs([&run_loop]() { run_loop.Quit(); })); |
EXPECT_CALL(*process_manager_, IsEventPageSuspended(extension_->id())) |
#if defined(OS_WIN) |
// Windows calls once for EnableMdnsDiscovery |