OLD | NEW |
(Empty) | |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. |
| 4 |
| 5 #include "chrome/browser/media/router/discovery/dial/dial_media_sink_service.h" |
| 6 #include "base/test/mock_callback.h" |
| 7 #include "chrome/browser/media/router/discovery/dial/dial_device_data.h" |
| 8 #include "chrome/browser/media/router/discovery/dial/dial_media_sink_cache_servi
ce.h" |
| 9 #include "chrome/browser/media/router/discovery/dial/dial_registry.h" |
| 10 #include "chrome/browser/media/router/discovery/dial/mock_device_description_ser
vice.h" |
| 11 #include "chrome/browser/media/router/test_helper.h" |
| 12 #include "chrome/test/base/testing_profile.h" |
| 13 #include "content/public/test/test_browser_thread_bundle.h" |
| 14 #include "testing/gmock/include/gmock/gmock.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" |
| 16 |
| 17 using ::testing::_; |
| 18 using ::testing::Return; |
| 19 |
| 20 namespace media_router { |
| 21 |
| 22 class TestDialRegistry : public DialRegistry { |
| 23 public: |
| 24 TestDialRegistry() {} |
| 25 ~TestDialRegistry() {} |
| 26 |
| 27 MOCK_METHOD1(RegisterObserver, void(DialRegistry::Observer* observer)); |
| 28 MOCK_METHOD1(UnregisterObserver, void(DialRegistry::Observer* observer)); |
| 29 }; |
| 30 |
| 31 class MockDialMediaSinkCacheService : public DialMediaSinkCacheService { |
| 32 public: |
| 33 MOCK_METHOD0(GetAliveSinks, std::vector<MediaSinkInternal>()); |
| 34 }; |
| 35 |
| 36 class TestDialMediaSinkService : public DialMediaSinkService { |
| 37 public: |
| 38 TestDialMediaSinkService( |
| 39 const MediaSinkService::OnSinksDiscoveredCallback& callback, |
| 40 content::BrowserContext* browser_context, |
| 41 TestDialRegistry* test_dial_registry, |
| 42 MockDeviceDescriptionService* mock_description_service, |
| 43 MockDialMediaSinkCacheService* mock_cache_service) |
| 44 : DialMediaSinkService(callback, browser_context), |
| 45 test_dial_registry_(test_dial_registry), |
| 46 mock_description_service_(mock_description_service), |
| 47 mock_cache_service_(mock_cache_service) {} |
| 48 |
| 49 DialRegistry* dial_registry() override { return test_dial_registry_; } |
| 50 |
| 51 DeviceDescriptionService* description_service() override { |
| 52 return mock_description_service_; |
| 53 } |
| 54 |
| 55 DialMediaSinkCacheService* cache_service() override { |
| 56 return mock_cache_service_; |
| 57 } |
| 58 |
| 59 private: |
| 60 TestDialRegistry* test_dial_registry_; |
| 61 MockDeviceDescriptionService* mock_description_service_; |
| 62 MockDialMediaSinkCacheService* mock_cache_service_; |
| 63 }; |
| 64 |
| 65 class DialMediaSinkServiceTest : public ::testing::Test { |
| 66 public: |
| 67 DialMediaSinkServiceTest() |
| 68 : thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP), |
| 69 mock_description_service_(mock_success_cb_.Get(), mock_error_cb_.Get()), |
| 70 media_sink_service_(mock_sink_discovered_cb_.Get(), |
| 71 &profile_, |
| 72 &test_dial_registry_, |
| 73 &mock_description_service_, |
| 74 &mock_cache_service_) {} |
| 75 |
| 76 DialMediaSinkService* media_sink_service() { return &media_sink_service_; } |
| 77 |
| 78 protected: |
| 79 const content::TestBrowserThreadBundle thread_bundle_; |
| 80 TestingProfile profile_; |
| 81 |
| 82 base::MockCallback<MediaSinkService::OnSinksDiscoveredCallback> |
| 83 mock_sink_discovered_cb_; |
| 84 base::MockCallback< |
| 85 MockDeviceDescriptionService::DeviceDescriptionParseSuccessCallback> |
| 86 mock_success_cb_; |
| 87 base::MockCallback< |
| 88 MockDeviceDescriptionService::DeviceDescriptionParseErrorCallback> |
| 89 mock_error_cb_; |
| 90 |
| 91 TestDialRegistry test_dial_registry_; |
| 92 MockDeviceDescriptionService mock_description_service_; |
| 93 MockDialMediaSinkCacheService mock_cache_service_; |
| 94 |
| 95 TestDialMediaSinkService media_sink_service_; |
| 96 }; |
| 97 |
| 98 TEST_F(DialMediaSinkServiceTest, TestStart) { |
| 99 EXPECT_CALL(test_dial_registry_, RegisterObserver(&media_sink_service_)); |
| 100 media_sink_service()->Start(); |
| 101 |
| 102 DialRegistry::DeviceList deviceList; |
| 103 DialDeviceData first_device("first", GURL("http://127.0.0.1/dd.xml"), |
| 104 base::Time::Now()); |
| 105 DialDeviceData second_device("second", GURL("http://127.0.0.2/dd.xml"), |
| 106 base::Time::Now()); |
| 107 DialDeviceData third_device("third", GURL("http://127.0.0.3/dd.xml"), |
| 108 base::Time::Now()); |
| 109 deviceList.push_back(first_device); |
| 110 deviceList.push_back(second_device); |
| 111 deviceList.push_back(third_device); |
| 112 |
| 113 EXPECT_CALL(mock_description_service_, GetDeviceDescriptions(deviceList, _)); |
| 114 |
| 115 media_sink_service()->OnDialDeviceEvent(deviceList); |
| 116 EXPECT_TRUE(media_sink_service_.finish_timer_.IsRunning()); |
| 117 } |
| 118 |
| 119 TEST_F(DialMediaSinkServiceTest, TestFetchCompleted) { |
| 120 MediaSink sink1("sink1", "sink_name_1", MediaSink::IconType::CAST); |
| 121 DialSinkExtraData extra_data1("192.168.1.1", "model_name1", |
| 122 "https://example1.com"); |
| 123 |
| 124 MediaSink sink2("sink2", "sink_name_2", MediaSink::IconType::CAST); |
| 125 DialSinkExtraData extra_data2("192.168.1.2", "model_name2", |
| 126 "https://example2.com"); |
| 127 |
| 128 std::vector<MediaSinkInternal> sinks; |
| 129 sinks.push_back(MediaSinkInternal(sink1, extra_data1)); |
| 130 sinks.push_back(MediaSinkInternal(sink2, extra_data2)); |
| 131 |
| 132 EXPECT_CALL(mock_cache_service_, GetAliveSinks()) |
| 133 .WillRepeatedly(Return(sinks)); |
| 134 EXPECT_CALL(mock_sink_discovered_cb_, Run(sinks)); |
| 135 media_sink_service()->FetchCompleted(); |
| 136 |
| 137 EXPECT_CALL(mock_sink_discovered_cb_, Run(_)).Times(0); |
| 138 media_sink_service()->FetchCompleted(); |
| 139 } |
| 140 |
| 141 TEST_F(DialMediaSinkServiceTest, TestIsDifferent) { |
| 142 MediaSink sink1("sink1", "sink_name_1", MediaSink::IconType::CAST); |
| 143 DialSinkExtraData extra_data1("192.168.1.1", "model_name1", |
| 144 "https://example1.com"); |
| 145 |
| 146 MediaSink sink2("sink2", "sink_name_2", MediaSink::IconType::CAST); |
| 147 DialSinkExtraData extra_data2("192.168.1.2", "model_name2", |
| 148 "https://example2.com"); |
| 149 |
| 150 std::vector<MediaSinkInternal> old_sinks; |
| 151 old_sinks.push_back(MediaSinkInternal(sink1, extra_data1)); |
| 152 old_sinks.push_back(MediaSinkInternal(sink2, extra_data2)); |
| 153 |
| 154 // The same sinks. |
| 155 std::vector<MediaSinkInternal> new_sinks1(old_sinks); |
| 156 ASSERT_FALSE(media_sink_service()->IsDifferent(new_sinks1, old_sinks)); |
| 157 |
| 158 // Add a new sink to new_sinks. |
| 159 std::vector<MediaSinkInternal> new_sinks2(old_sinks); |
| 160 MediaSink sink3("sink3", "sink_name_3", MediaSink::IconType::CAST); |
| 161 DialSinkExtraData extra_data3("192.168.1.3", "model_name3", |
| 162 "https://example3.com"); |
| 163 |
| 164 new_sinks2.push_back(MediaSinkInternal(sink3, extra_data3)); |
| 165 ASSERT_TRUE(media_sink_service()->IsDifferent(new_sinks2, old_sinks)); |
| 166 |
| 167 // Remove a sink from new_sinks. |
| 168 std::vector<MediaSinkInternal> new_sinks3(old_sinks); |
| 169 new_sinks3.erase(new_sinks3.begin()); |
| 170 ASSERT_TRUE(media_sink_service()->IsDifferent(new_sinks3, old_sinks)); |
| 171 |
| 172 // updated a sink from new_sinks |
| 173 std::vector<MediaSinkInternal> new_sinks4(old_sinks); |
| 174 new_sinks4[0].set_name("sink_name_4"); |
| 175 ASSERT_TRUE(media_sink_service()->IsDifferent(new_sinks4, old_sinks)); |
| 176 |
| 177 // Combined add, remove and update |
| 178 std::vector<MediaSinkInternal> new_sinks5; |
| 179 new_sinks5.push_back(MediaSinkInternal(sink1, extra_data2)); |
| 180 new_sinks5.push_back(MediaSinkInternal(sink3, extra_data3)); |
| 181 ASSERT_TRUE(media_sink_service()->IsDifferent(new_sinks5, old_sinks)); |
| 182 } |
| 183 |
| 184 } // namespace media_router |
OLD | NEW |