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_cache_servi
ce.h" |
| 6 #include "chrome/browser/media/router/discovery/dial/mock_device_description_ser
vice.h" |
| 7 #include "chrome/test/base/testing_profile.h" |
| 8 #include "content/public/test/test_browser_thread_bundle.h" |
| 9 #include "testing/gmock/include/gmock/gmock.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" |
| 11 |
| 12 using ::testing::_; |
| 13 |
| 14 namespace media_router { |
| 15 |
| 16 class DialMediaSinkCacheServiceTest : public ::testing::Test { |
| 17 public: |
| 18 void TearDown() override { cache_service_.cached_sinks_.clear(); } |
| 19 |
| 20 void AddTestDialMediaSinks() { |
| 21 MediaSink sink1("sink1", "sink_name_1", MediaSink::IconType::CAST); |
| 22 DialSinkExtraData extra_data1("192.168.1.1", "model_name1", |
| 23 "https://example1.com"); |
| 24 MediaSinkInternal dial_sink1(sink1, extra_data1); |
| 25 GURL gurl1 = GURL("https://device1"); |
| 26 base::Time last_update_time1 = |
| 27 base::Time::Now() - 3 * base::TimeDelta::FromSeconds(60); |
| 28 AddTestDialMediaSink(dial_sink1, gurl1, last_update_time1); |
| 29 |
| 30 MediaSink sink2("sink2", "sink_name_2", MediaSink::IconType::CAST); |
| 31 DialSinkExtraData extra_data2("192.168.1.2", "model_name2", |
| 32 "https://example2.com"); |
| 33 MediaSinkInternal dial_sink2(sink2, extra_data2); |
| 34 GURL gurl2 = GURL("https://device2"); |
| 35 base::Time last_update_time2 = base::Time::Now(); |
| 36 AddTestDialMediaSink(dial_sink2, gurl2, last_update_time2); |
| 37 } |
| 38 |
| 39 void AddTestDialMediaSink(const MediaSinkInternal& media_sink, |
| 40 const GURL& description_url, |
| 41 base::Time last_update_time) { |
| 42 CachedDialMediaSink dial_sink; |
| 43 dial_sink.sink = media_sink; |
| 44 dial_sink.last_update_time = last_update_time; |
| 45 cache_service_.cached_sinks_.insert( |
| 46 std::make_pair(media_sink.sink().id(), dial_sink)); |
| 47 } |
| 48 |
| 49 DialMediaSinkCacheService* cache_service() { return &cache_service_; } |
| 50 |
| 51 protected: |
| 52 const content::TestBrowserThreadBundle thread_bundle_; |
| 53 TestingProfile profile_; |
| 54 DialMediaSinkCacheService cache_service_; |
| 55 }; |
| 56 |
| 57 TEST_F(DialMediaSinkCacheServiceTest, GetAliveSinks) { |
| 58 auto sinks = cache_service()->GetAliveSinks(); |
| 59 ASSERT_TRUE(sinks.empty()); |
| 60 |
| 61 AddTestDialMediaSinks(); |
| 62 |
| 63 sinks = cache_service()->GetAliveSinks(); |
| 64 EXPECT_EQ(size_t(1), sinks.size()); |
| 65 } |
| 66 |
| 67 TEST_F(DialMediaSinkCacheServiceTest, TestAddOrUpdateSink) { |
| 68 std::string sink_id = "sink_id"; |
| 69 std::string sink_name = "sink_name"; |
| 70 std::string sink_name_updated = "sink_name_updated"; |
| 71 |
| 72 DialDeviceData device_data; |
| 73 device_data.set_device_description_url(GURL("http://192.168.1.1")); |
| 74 |
| 75 ParsedDialDeviceDescription desc1; |
| 76 desc1.unique_id = sink_id; |
| 77 desc1.friendly_name = sink_name; |
| 78 cache_service()->MayAddOrUpdateSink(device_data, desc1); |
| 79 |
| 80 auto sinks = cache_service()->GetAliveSinks(); |
| 81 EXPECT_EQ(size_t(1), sinks.size()); |
| 82 EXPECT_EQ(sink_id, sinks[0].sink().id()); |
| 83 EXPECT_EQ(sink_name, sinks[0].sink().name()); |
| 84 |
| 85 device_data.set_device_description_url(GURL("http://192.168.1.2")); |
| 86 |
| 87 ParsedDialDeviceDescription desc2; |
| 88 desc2.unique_id = sink_id; |
| 89 desc2.friendly_name = sink_name_updated; |
| 90 cache_service()->MayAddOrUpdateSink(device_data, desc2); |
| 91 sinks = cache_service()->GetAliveSinks(); |
| 92 EXPECT_EQ(size_t(1), sinks.size()); |
| 93 EXPECT_EQ(sink_id, sinks[0].sink().id()); |
| 94 EXPECT_EQ(sink_name_updated, sinks[0].sink().name()); |
| 95 |
| 96 device_data.set_device_description_url(GURL("http://192.168.1.2")); |
| 97 |
| 98 ParsedDialDeviceDescription desc3; |
| 99 desc3.unique_id = sink_id; |
| 100 desc3.friendly_name = sink_name_updated; |
| 101 cache_service()->MayAddOrUpdateSink(device_data, desc3); |
| 102 sinks = cache_service()->GetAliveSinks(); |
| 103 EXPECT_EQ(size_t(1), sinks.size()); |
| 104 } |
| 105 |
| 106 } // namespace media_router |
OLD | NEW |