| Index: chrome/browser/media/router/media_router_mojo_impl_unittest.cc
|
| diff --git a/chrome/browser/media/router/media_router_mojo_impl_unittest.cc b/chrome/browser/media/router/media_router_mojo_impl_unittest.cc
|
| index c9cfaac491883f4249ee857d319c0de2989ad2f1..a7e93d0fb770eeb9defcb3b1c5739441e61c5e2f 100644
|
| --- a/chrome/browser/media/router/media_router_mojo_impl_unittest.cc
|
| +++ b/chrome/browser/media/router/media_router_mojo_impl_unittest.cc
|
| @@ -482,13 +482,18 @@ TEST_F(MediaRouterMojoImplTest, RegisterAndUnregisterMediaSinksObserver) {
|
| router()->OnSinkAvailabilityUpdated(
|
| interfaces::MediaRouter::SinkAvailability::AVAILABLE);
|
| MediaSource media_source(kSource);
|
| + media_source.set_origin(kOrigin);
|
| + GURL origin(kOrigin);
|
| + MediaSource media_source2(kSource2);
|
|
|
| // These should only be called once even if there is more than one observer
|
| // for a given source.
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource)));
|
| + StartObservingMediaSinks(mojo::String(kSource),
|
| + mojo::String(kOrigin)));
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource2)));
|
| + StartObservingMediaSinks(mojo::String(kSource2),
|
| + mojo::String("")));
|
|
|
| scoped_ptr<MockMediaSinksObserver> sinks_observer(
|
| new MockMediaSinksObserver(router(), media_source));
|
| @@ -497,7 +502,7 @@ TEST_F(MediaRouterMojoImplTest, RegisterAndUnregisterMediaSinksObserver) {
|
| new MockMediaSinksObserver(router(), media_source));
|
| EXPECT_TRUE(extra_sinks_observer->Init());
|
| scoped_ptr<MockMediaSinksObserver> unrelated_sinks_observer(
|
| - new MockMediaSinksObserver(router(), MediaSource(kSource2)));
|
| + new MockMediaSinksObserver(router(), media_source2));
|
| EXPECT_TRUE(unrelated_sinks_observer->Init());
|
| ProcessEventLoop();
|
|
|
| @@ -520,11 +525,13 @@ TEST_F(MediaRouterMojoImplTest, RegisterAndUnregisterMediaSinksObserver) {
|
| media_router::interfaces::MediaSink::IconType::CAST;
|
|
|
| base::RunLoop run_loop;
|
| - EXPECT_CALL(*sinks_observer, OnSinksReceived(SequenceEquals(expected_sinks)));
|
| + EXPECT_CALL(*sinks_observer, OnSinksReceived(media_source,
|
| + SequenceEquals(expected_sinks)));
|
| EXPECT_CALL(*extra_sinks_observer,
|
| - OnSinksReceived(SequenceEquals(expected_sinks)))
|
| + OnSinksReceived(media_source, SequenceEquals(expected_sinks)))
|
| .WillOnce(InvokeWithoutArgs([&run_loop]() { run_loop.Quit(); }));
|
| media_router_proxy_->OnSinksReceived(media_source.id(),
|
| + media_source.origin(),
|
| std::move(mojo_sinks));
|
| run_loop.Run();
|
|
|
| @@ -534,44 +541,56 @@ TEST_F(MediaRouterMojoImplTest, RegisterAndUnregisterMediaSinksObserver) {
|
| scoped_ptr<MockMediaSinksObserver> cached_sinks_observer(
|
| new MockMediaSinksObserver(router(), media_source));
|
| EXPECT_CALL(*cached_sinks_observer,
|
| - OnSinksReceived(SequenceEquals(expected_sinks)));
|
| + OnSinksReceived(media_source, SequenceEquals(expected_sinks)));
|
| EXPECT_TRUE(cached_sinks_observer->Init());
|
|
|
| base::RunLoop run_loop2;
|
| + base::RunLoop run_loop3;
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StopObservingMediaSinks(mojo::String(kSource)));
|
| - EXPECT_CALL(mock_media_route_provider_,
|
| - StopObservingMediaSinks(mojo::String(kSource2)))
|
| + StopObservingMediaSinks(mojo::String(kSource),
|
| + mojo::String(kOrigin)))
|
| .WillOnce(InvokeWithoutArgs([&run_loop2]() {
|
| run_loop2.Quit();
|
| }));
|
| + EXPECT_CALL(mock_media_route_provider_,
|
| + StopObservingMediaSinks(mojo::String(kSource2),
|
| + mojo::String("")))
|
| + .WillOnce(InvokeWithoutArgs([&run_loop3]() {
|
| + run_loop3.Quit();
|
| + }));
|
| sinks_observer.reset();
|
| extra_sinks_observer.reset();
|
| unrelated_sinks_observer.reset();
|
| cached_sinks_observer.reset();
|
| run_loop2.Run();
|
| + run_loop3.Run();
|
| }
|
|
|
| TEST_F(MediaRouterMojoImplTest,
|
| RegisterMediaSinksObserverWithAvailabilityChange) {
|
| + GURL origin(kOrigin);
|
| // When availability is UNAVAILABLE, no calls should be made to MRPM.
|
| router()->OnSinkAvailabilityUpdated(
|
| interfaces::MediaRouter::SinkAvailability::UNAVAILABLE);
|
| MediaSource media_source(kSource);
|
| + media_source.set_origin(kOrigin);
|
| scoped_ptr<MockMediaSinksObserver> sinks_observer(
|
| new MockMediaSinksObserver(router(), media_source));
|
| - EXPECT_CALL(*sinks_observer, OnSinksReceived(IsEmpty()));
|
| + EXPECT_CALL(*sinks_observer, OnSinksReceived(media_source, IsEmpty()));
|
| EXPECT_TRUE(sinks_observer->Init());
|
| MediaSource media_source2(kSource2);
|
| + media_source2.set_origin(kOrigin);
|
| scoped_ptr<MockMediaSinksObserver> sinks_observer2(
|
| new MockMediaSinksObserver(router(), media_source2));
|
| - EXPECT_CALL(*sinks_observer2, OnSinksReceived(IsEmpty()));
|
| + EXPECT_CALL(*sinks_observer2, OnSinksReceived(media_source2, IsEmpty()));
|
| EXPECT_TRUE(sinks_observer2->Init());
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource)))
|
| + StartObservingMediaSinks(mojo::String(kSource),
|
| + mojo::String(kOrigin)))
|
| .Times(0);
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource2)))
|
| + StartObservingMediaSinks(mojo::String(kSource2),
|
| + mojo::String(kOrigin)))
|
| .Times(0);
|
| ProcessEventLoop();
|
| EXPECT_TRUE(Mock::VerifyAndClearExpectations(&mock_media_route_provider_));
|
| @@ -581,10 +600,12 @@ TEST_F(MediaRouterMojoImplTest,
|
| router()->OnSinkAvailabilityUpdated(
|
| interfaces::MediaRouter::SinkAvailability::AVAILABLE);
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource)))
|
| + StartObservingMediaSinks(mojo::String(kSource),
|
| + mojo::String(kOrigin)))
|
| .Times(1);
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource2)))
|
| + StartObservingMediaSinks(mojo::String(kSource2),
|
| + mojo::String(kOrigin)))
|
| .Times(1);
|
| ProcessEventLoop();
|
| EXPECT_TRUE(Mock::VerifyAndClearExpectations(&mock_media_route_provider_));
|
| @@ -593,10 +614,12 @@ TEST_F(MediaRouterMojoImplTest,
|
| router()->OnSinkAvailabilityUpdated(
|
| interfaces::MediaRouter::SinkAvailability::AVAILABLE);
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource)))
|
| + StartObservingMediaSinks(mojo::String(kSource),
|
| + mojo::String(kOrigin)))
|
| .Times(0);
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StartObservingMediaSinks(mojo::String(kSource2)))
|
| + StartObservingMediaSinks(mojo::String(kSource2),
|
| + mojo::String(kOrigin)))
|
| .Times(0);
|
| ProcessEventLoop();
|
| EXPECT_TRUE(Mock::VerifyAndClearExpectations(&mock_media_route_provider_));
|
| @@ -606,7 +629,8 @@ TEST_F(MediaRouterMojoImplTest,
|
| router()->OnSinkAvailabilityUpdated(
|
| interfaces::MediaRouter::SinkAvailability::UNAVAILABLE);
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StopObservingMediaSinks(mojo::String(kSource)))
|
| + StopObservingMediaSinks(mojo::String(kSource),
|
| + mojo::String(kOrigin)))
|
| .Times(0);
|
| sinks_observer.reset();
|
| ProcessEventLoop();
|
| @@ -617,7 +641,8 @@ TEST_F(MediaRouterMojoImplTest,
|
| router()->OnSinkAvailabilityUpdated(
|
| interfaces::MediaRouter::SinkAvailability::AVAILABLE);
|
| EXPECT_CALL(mock_media_route_provider_,
|
| - StopObservingMediaSinks(mojo::String(kSource2)));
|
| + StopObservingMediaSinks(mojo::String(kSource2),
|
| + mojo::String(kOrigin)));
|
| sinks_observer2.reset();
|
| ProcessEventLoop();
|
| }
|
|
|