[Media Router] Support dual discovery

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

 // 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"
@@ skipping 28 matching lines @@
   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 MockMediaSinkServiceObserver : public MediaSinkService::Observer {
+ public:
+  MOCK_METHOD1(OnMediaSinkAdded, void(const MediaSinkInternal& sink));
+  MOCK_METHOD0(OnMediaSinksRemoved, void());
+};
+
 class TestMediaSinkServiceBase : public MediaSinkServiceBase {
  public:
   explicit TestMediaSinkServiceBase(const OnSinksDiscoveredCallback& callback)
       : MediaSinkServiceBase(callback) {}
 
   void Start() override {}
   void Stop() override {}
 };
 
 class MediaSinkServiceBaseTest : public ::testing::Test {
@@ skipping 74 matching lines @@
   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);
 }
 
+TEST_F(MediaSinkServiceBaseTest, TestAddRemoveSinks) {
+  MockMediaSinkServiceObserver mock_observer;
+
+  auto sinks = CreateDialMediaSinks();
+  EXPECT_CALL(mock_observer, OnMediaSinkAdded(sinks[0]));
+  EXPECT_CALL(mock_observer, OnMediaSinkAdded(sinks[1]));
+
+  media_sink_service_->AddObserver(&mock_observer);
+  media_sink_service_->AddSink(sinks[0]);
+  EXPECT_EQ(size_t(1), media_sink_service_->GetCurrentSinksSize());
+  media_sink_service_->AddSink(sinks[0]);
+  EXPECT_EQ(size_t(1), media_sink_service_->GetCurrentSinksSize());
+  media_sink_service_->AddSink(sinks[1]);
+  EXPECT_EQ(size_t(2), media_sink_service_->GetCurrentSinksSize());
+
+  EXPECT_CALL(mock_observer, OnMediaSinksRemoved());
+  media_sink_service_->RemoveSinks();
+  EXPECT_EQ(size_t(0), media_sink_service_->GetCurrentSinksSize());
+}
+
 }  // namespace media_router