components/media_router/browser/media_sinks_observer_unittest.cc

// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/media_router/browser/media_sinks_observer.h"

#include "components/media_router/browser/test/mock_media_router.h"
#include "components/media_router/browser/test/test_helper.h"
#include "components/media_router/common/media_source.h"
#include "components/media_router/common/test/test_helper.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace media_router {

TEST(MediaSinksObserverTest, OriginMatching) {
  content::BrowserTaskEnvironment task_environment;
  MockMediaRouter router;
  MediaSource source(
      MediaSource::ForPresentationUrl(GURL("https://presentation.com")));
  url::Origin origin = url::Origin::Create(GURL("https://origin.com"));
  std::vector<url::Origin> origin_list({origin});
  std::vector<MediaSink> sink_list;
  sink_list.push_back(CreateCastSink("sinkId", "Sink"));
  MockMediaSinksObserver observer(&router, source, origin);

  EXPECT_CALL(observer, OnSinksReceived(sink_list));
  observer.OnSinksUpdated(sink_list, origin_list);

  EXPECT_CALL(observer, OnSinksReceived(sink_list));
  observer.OnSinksUpdated(sink_list, std::vector<url::Origin>());

  url::Origin origin2 =
      url::Origin::Create(GURL("https://differentOrigin.com"));
  origin_list.clear();
  origin_list.push_back(origin2);
  EXPECT_CALL(observer, OnSinksReceived(testing::IsEmpty()));
  observer.OnSinksUpdated(sink_list, origin_list);
}

}  // namespace media_router