[Media Router] Create MediaSinkServiceBase and make DialMediaSinkService inherit from it

chrome/browser/media/router/discovery/media_sink_service_base_unittest.cc

Index: chrome/browser/media/router/discovery/media_sink_service_base_unittest.cc
diff --git a/chrome/browser/media/router/discovery/media_sink_service_base_unittest.cc b/chrome/browser/media/router/discovery/media_sink_service_base_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..eac9ea494a9d2a82060a0916d1771c883f69449a
--- /dev/null
+++ b/chrome/browser/media/router/discovery/media_sink_service_base_unittest.cc
@@ -0,0 +1,143 @@
+// Copyright 2017 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/media/router/discovery/media_sink_service_base.h"
+#include "base/test/mock_callback.h"
+#include "base/timer/mock_timer.h"
+#include "chrome/browser/media/router/test_helper.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using ::testing::_;
+using ::testing::Return;
+
+namespace {
+
+media_router::DialSinkExtraData CreateDialSinkExtraData(
+    const std::string& model_name,
+    const std::string& ip_address,
+    const std::string& app_url) {
+  media_router::DialSinkExtraData dial_extra_data;
+  EXPECT_TRUE(dial_extra_data.ip_address.AssignFromIPLiteral(ip_address));
+  dial_extra_data.model_name = model_name;
+  dial_extra_data.app_url = GURL(app_url);
+  return dial_extra_data;
+}
+
+std::vector<media_router::MediaSinkInternal> CreateDialMediaSinks() {
+  media_router::MediaSink sink1("sink1", "sink_name_1",
+                                media_router::MediaSink::IconType::CAST);
+  media_router::DialSinkExtraData extra_data1 = CreateDialSinkExtraData(
+      "model_name1", "192.168.1.1", "https://example1.com");
+
+  media_router::MediaSink sink2("sink2", "sink_name_2",
+                                media_router::MediaSink::IconType::CAST);
+  media_router::DialSinkExtraData extra_data2 = CreateDialSinkExtraData(
+      "model_name2", "192.168.1.2", "https://example2.com");
+
+  std::vector<media_router::MediaSinkInternal> sinks;
+  sinks.push_back(media_router::MediaSinkInternal(sink1, extra_data1));
+  sinks.push_back(media_router::MediaSinkInternal(sink2, extra_data2));
+  return sinks;
+}
+
+}  // namespace
+
+namespace media_router {
+
+class TestMediaSinkServiceBase : public MediaSinkServiceBase {
+ public:
+  explicit TestMediaSinkServiceBase(const OnSinksDiscoveredCallback& callback)
+      : MediaSinkServiceBase(callback) {}
+
+  void Start() override {}
+  void Stop() override {}
+};
+
+class MediaSinkServiceBaseTest : public ::testing::Test {
+ public:
+  MediaSinkServiceBaseTest()
+      :  // thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP),
+        media_sink_service_(
+            new TestMediaSinkServiceBase(mock_sink_discovered_cb_.Get())) {}
+
+  void SetUp() override {
+    mock_timer_ =
+        new base::MockTimer(true /*retain_user_task*/, false /*is_repeating*/);
+    media_sink_service_->SetTimerForTest(base::WrapUnique(mock_timer_));
+  }
+
+  void TestFetchCompleted(const std::vector<MediaSinkInternal>& old_sinks,
+                          const std::vector<MediaSinkInternal>& new_sinks) {
+    media_sink_service_->mrp_sinks_ =
+        std::set<MediaSinkInternal>(old_sinks.begin(), old_sinks.end());
+    media_sink_service_->current_sinks_ =
+        std::set<MediaSinkInternal>(new_sinks.begin(), new_sinks.end());
+    EXPECT_CALL(mock_sink_discovered_cb_, Run(new_sinks));
+    media_sink_service_->OnFetchCompleted();
+  }
+
+ protected:
+  base::MockCallback<MediaSinkService::OnSinksDiscoveredCallback>
+      mock_sink_discovered_cb_;
+  base::MockTimer* mock_timer_;
+
+  std::unique_ptr<TestMediaSinkServiceBase> media_sink_service_;
+
+  DISALLOW_COPY_AND_ASSIGN(MediaSinkServiceBaseTest);
+};
+
+TEST_F(MediaSinkServiceBaseTest, TestFetchCompleted_SameSink) {
+  std::vector<MediaSinkInternal> old_sinks;
+  std::vector<MediaSinkInternal> new_sinks = CreateDialMediaSinks();
+  TestFetchCompleted(old_sinks, new_sinks);
+
+  // Same sink
+  EXPECT_CALL(mock_sink_discovered_cb_, Run(new_sinks)).Times(0);
+  media_sink_service_->OnFetchCompleted();
+}
+
+TEST_F(MediaSinkServiceBaseTest, TestFetchCompleted_OneNewSink) {
+  std::vector<MediaSinkInternal> old_sinks = CreateDialMediaSinks();
+  std::vector<MediaSinkInternal> new_sinks = CreateDialMediaSinks();
+  MediaSink sink3("sink3", "sink_name_3", MediaSink::IconType::CAST);
+  DialSinkExtraData extra_data3 = CreateDialSinkExtraData(
+      "model_name3", "192.168.1.3", "https://example3.com");
+  new_sinks.push_back(MediaSinkInternal(sink3, extra_data3));
+  TestFetchCompleted(old_sinks, new_sinks);
+}
+
+TEST_F(MediaSinkServiceBaseTest, TestFetchCompleted_RemovedOneSink) {
+  std::vector<MediaSinkInternal> old_sinks = CreateDialMediaSinks();
+  std::vector<MediaSinkInternal> new_sinks = CreateDialMediaSinks();
+  new_sinks.erase(new_sinks.begin());
+  TestFetchCompleted(old_sinks, new_sinks);
+}
+
+TEST_F(MediaSinkServiceBaseTest, TestFetchCompleted_UpdatedOneSink) {
+  std::vector<MediaSinkInternal> old_sinks = CreateDialMediaSinks();
+  std::vector<MediaSinkInternal> new_sinks = CreateDialMediaSinks();
+  new_sinks[0].set_name("sink_name_4");
+  TestFetchCompleted(old_sinks, new_sinks);
+}
+
+TEST_F(MediaSinkServiceBaseTest, TestFetchCompleted_Mixed) {
+  std::vector<MediaSinkInternal> old_sinks = CreateDialMediaSinks();
+
+  MediaSink sink1("sink1", "sink_name_1", MediaSink::IconType::CAST);
+  DialSinkExtraData extra_data2 = CreateDialSinkExtraData(
+      "model_name2", "192.168.1.2", "https://example2.com");
+
+  MediaSink sink3("sink3", "sink_name_3", MediaSink::IconType::CAST);
+  DialSinkExtraData extra_data3 = CreateDialSinkExtraData(
+      "model_name3", "192.168.1.3", "https://example3.com");
+
+  std::vector<MediaSinkInternal> new_sinks;
+  new_sinks.push_back(MediaSinkInternal(sink1, extra_data2));
+  new_sinks.push_back(MediaSinkInternal(sink3, extra_data3));
+
+  TestFetchCompleted(old_sinks, new_sinks);
+}
+
+}  // namespace media_router