| Index: chrome/browser/ui/webui/media_router/query_result_manager_unittest.cc
|
| diff --git a/chrome/browser/ui/webui/media_router/query_result_manager_unittest.cc b/chrome/browser/ui/webui/media_router/query_result_manager_unittest.cc
|
| index 8a9bbe982c9499e7d2b4c79e75732190d68716fd..8e019552ce810cca102c1b23653778d1c7e2c142 100644
|
| --- a/chrome/browser/ui/webui/media_router/query_result_manager_unittest.cc
|
| +++ b/chrome/browser/ui/webui/media_router/query_result_manager_unittest.cc
|
| @@ -47,7 +47,7 @@
|
| .WillOnce(Return(true));
|
| EXPECT_CALL(mock_observer_, OnResultsUpdated(_)).Times(1);
|
| query_result_manager_.SetSourcesForCastMode(cast_mode, {source},
|
| - url::Origin(GURL(kOrigin)));
|
| + GURL(kOrigin));
|
| }
|
|
|
| bool IsDefaultSourceForSink(const MediaSource* source,
|
| @@ -108,6 +108,7 @@
|
| }
|
|
|
| TEST_F(QueryResultManagerTest, StartStopSinksQuery) {
|
| + GURL origin(kOrigin);
|
| CastModeSet cast_modes = query_result_manager_.GetSupportedCastModes();
|
| EXPECT_TRUE(cast_modes.empty());
|
| std::vector<MediaSource> actual_sources =
|
| @@ -118,7 +119,7 @@
|
| EXPECT_CALL(mock_router_, RegisterMediaSinksObserver(_))
|
| .WillOnce(Return(true));
|
| query_result_manager_.SetSourcesForCastMode(MediaCastMode::DEFAULT, {source},
|
| - url::Origin(GURL(kOrigin)));
|
| + origin);
|
|
|
| cast_modes = query_result_manager_.GetSupportedCastModes();
|
| EXPECT_EQ(1u, cast_modes.size());
|
| @@ -134,8 +135,8 @@
|
| EXPECT_CALL(mock_router_, UnregisterMediaSinksObserver(_)).Times(1);
|
| EXPECT_CALL(mock_router_, RegisterMediaSinksObserver(_))
|
| .WillOnce(Return(true));
|
| - query_result_manager_.SetSourcesForCastMode(
|
| - MediaCastMode::DEFAULT, {another_source}, url::Origin(GURL(kOrigin)));
|
| + query_result_manager_.SetSourcesForCastMode(MediaCastMode::DEFAULT,
|
| + {another_source}, origin);
|
|
|
| cast_modes = query_result_manager_.GetSupportedCastModes();
|
| EXPECT_EQ(1u, cast_modes.size());
|
| @@ -166,6 +167,7 @@
|
| MediaSource default_source2 =
|
| MediaSourceForPresentationUrl(GURL("http://baz.com"));
|
| MediaSource tab_source = MediaSourceForTab(123);
|
| + GURL origin(kOrigin);
|
|
|
| query_result_manager_.AddObserver(&mock_observer_);
|
| DiscoverSinks(MediaCastMode::DEFAULT, default_source1);
|
| @@ -195,7 +197,7 @@
|
| EXPECT_CALL(mock_observer_,
|
| OnResultsUpdated(VectorEquals(expected_sinks))).Times(1);
|
| sinks_observer_it->second->OnSinksUpdated(sinks_query_result,
|
| - std::vector<url::Origin>());
|
| + std::vector<GURL>());
|
|
|
| // Action: TAB_MIRROR -> [2, 3, 4]
|
| // Expected result:
|
| @@ -224,7 +226,7 @@
|
| EXPECT_CALL(mock_observer_,
|
| OnResultsUpdated(VectorEquals(expected_sinks))).Times(1);
|
| sinks_observer_it->second->OnSinksUpdated(sinks_query_result,
|
| - {url::Origin(GURL(kOrigin))});
|
| + {GURL(kOrigin)});
|
|
|
| // Action: Update default presentation URL
|
| // Expected result:
|
| @@ -244,8 +246,8 @@
|
| .WillOnce(Return(true));
|
| EXPECT_CALL(mock_observer_,
|
| OnResultsUpdated(VectorEquals(expected_sinks))).Times(1);
|
| - query_result_manager_.SetSourcesForCastMode(
|
| - MediaCastMode::DEFAULT, {default_source2}, url::Origin(GURL(kOrigin)));
|
| + query_result_manager_.SetSourcesForCastMode(MediaCastMode::DEFAULT,
|
| + {default_source2}, origin);
|
|
|
| // Action: DEFAULT -> [1], origins don't match
|
| // Expected result: [2 -> {TAB_MIRROR}, 3 -> {TAB_MIRROR}, 4 -> {TAB_MIRROR}]
|
| @@ -258,7 +260,7 @@
|
| EXPECT_CALL(mock_observer_, OnResultsUpdated(VectorEquals(expected_sinks)))
|
| .Times(1);
|
| sinks_observer_it->second->OnSinksUpdated(
|
| - sinks_query_result, {url::Origin(GURL("https://differentOrigin.com"))});
|
| + sinks_query_result, {GURL("https://differentOrigin.com")});
|
|
|
| // Action: Remove TAB_MIRROR observer
|
| // Expected result:
|
| @@ -289,16 +291,17 @@
|
| const std::vector<MediaSource> default_sources = {source_a, source_b,
|
| source_c};
|
| const std::vector<MediaSource> tab_sources = {source_tab};
|
| + const GURL origin(kOrigin);
|
| const auto& sinks_observers = query_result_manager_.sinks_observers_;
|
|
|
| // There should be one MediaSinksObserver per source.
|
| EXPECT_CALL(mock_router_, RegisterMediaSinksObserver(_))
|
| .Times(4)
|
| .WillRepeatedly(Return(true));
|
| - query_result_manager_.SetSourcesForCastMode(
|
| - MediaCastMode::DEFAULT, default_sources, url::Origin(GURL(kOrigin)));
|
| - query_result_manager_.SetSourcesForCastMode(
|
| - MediaCastMode::TAB_MIRROR, tab_sources, url::Origin(GURL(kOrigin)));
|
| + query_result_manager_.SetSourcesForCastMode(MediaCastMode::DEFAULT,
|
| + default_sources, origin);
|
| + query_result_manager_.SetSourcesForCastMode(MediaCastMode::TAB_MIRROR,
|
| + tab_sources, origin);
|
|
|
| // Scenario (results in this order):
|
| // Action: URL_B -> [2, 4]
|
| @@ -312,7 +315,7 @@
|
| ASSERT_TRUE(sinks_observer_it->second.get());
|
|
|
| auto& source_b_observer = sinks_observer_it->second;
|
| - source_b_observer->OnSinksUpdated({sink2, sink4}, std::vector<url::Origin>());
|
| + source_b_observer->OnSinksUpdated({sink2, sink4}, std::vector<GURL>());
|
| EXPECT_TRUE(IsDefaultSourceForSink(nullptr, sink1));
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_b, sink2));
|
| EXPECT_TRUE(IsDefaultSourceForSink(nullptr, sink3));
|
| @@ -329,8 +332,7 @@
|
| ASSERT_TRUE(sinks_observer_it->second.get());
|
|
|
| auto& source_c_observer = sinks_observer_it->second;
|
| - source_c_observer->OnSinksUpdated({sink1, sink2, sink3},
|
| - std::vector<url::Origin>());
|
| + source_c_observer->OnSinksUpdated({sink1, sink2, sink3}, std::vector<GURL>());
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_c, sink1));
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_b, sink2));
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_c, sink3));
|
| @@ -347,8 +349,7 @@
|
| ASSERT_TRUE(sinks_observer_it->second.get());
|
|
|
| auto& source_a_observer = sinks_observer_it->second;
|
| - source_a_observer->OnSinksUpdated({sink2, sink3, sink4},
|
| - std::vector<url::Origin>());
|
| + source_a_observer->OnSinksUpdated({sink2, sink3, sink4}, std::vector<GURL>());
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_c, sink1));
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_a, sink2));
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_a, sink3));
|
| @@ -365,8 +366,7 @@
|
| ASSERT_TRUE(sinks_observer_it->second.get());
|
|
|
| auto& source_tab_observer = sinks_observer_it->second;
|
| - source_tab_observer->OnSinksUpdated({sink1, sink2},
|
| - std::vector<url::Origin>());
|
| + source_tab_observer->OnSinksUpdated({sink1, sink2}, std::vector<GURL>());
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_c, sink1));
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_a, sink2));
|
| EXPECT_TRUE(IsDefaultSourceForSink(&source_a, sink3));
|
| @@ -385,16 +385,17 @@
|
| TEST_F(QueryResultManagerTest, AddInvalidSource) {
|
| const MediaSource source(
|
| MediaSourceForPresentationUrl(GURL("http://url.com")));
|
| + const GURL origin(kOrigin);
|
|
|
| EXPECT_CALL(mock_router_, RegisterMediaSinksObserver(_))
|
| .Times(1)
|
| .WillRepeatedly(Return(true));
|
| query_result_manager_.SetSourcesForCastMode(MediaCastMode::DEFAULT, {source},
|
| - url::Origin(GURL(kOrigin)));
|
| + origin);
|
| // |source| has already been registered with the default cast mode, so it
|
| // shouldn't get registered with tab mirroring.
|
| - query_result_manager_.SetSourcesForCastMode(
|
| - MediaCastMode::TAB_MIRROR, {source}, url::Origin(GURL(kOrigin)));
|
| + query_result_manager_.SetSourcesForCastMode(MediaCastMode::TAB_MIRROR,
|
| + {source}, origin);
|
|
|
| const auto& cast_mode_sources = query_result_manager_.cast_mode_sources_;
|
| const auto& default_sources = cast_mode_sources.at(MediaCastMode::DEFAULT);
|
|
|