[Media Router] Add origins to sink query results.

chrome/browser/media/router/media_sinks_observer_unittest.cc

Index: chrome/browser/media/router/media_sinks_observer_unittest.cc
diff --git a/chrome/browser/media/router/media_sinks_observer_unittest.cc b/chrome/browser/media/router/media_sinks_observer_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..60ab2ced04ab4bb95f1c78d638b38aaa94498f5c
--- /dev/null
+++ b/chrome/browser/media/router/media_sinks_observer_unittest.cc
@@ -0,0 +1,36 @@
+// Copyright 2016 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/media_source_helper.h"
+#include "chrome/browser/media/router/mock_media_router.h"
+#include "chrome/browser/media/router/test_helper.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace media_router {
+
+TEST(MediaSinksObserverTest, OriginMatching) {
+  MockMediaRouter router;
+  MediaSource source(MediaSourceForPresentationUrl("https://presentation.com"));
+  GURL origin("https://origin.com");
+  std::vector<GURL> origin_list;
+  origin_list.push_back(origin);
+  std::vector<MediaSink> sink_list;
+  sink_list.push_back(MediaSink("sinkId", "Sink", MediaSink::IconType::CAST));
+  MockMediaSinksObserver observer(&router, source, origin);
+
+  EXPECT_CALL(observer, OnSinksReceived(SequenceEquals(sink_list)));
+  observer.OnSinksUpdated(sink_list, origin_list);
+
+  EXPECT_CALL(observer, OnSinksReceived(SequenceEquals(sink_list)));
+  observer.OnSinksUpdated(sink_list, std::vector<GURL>());
+
+  GURL origin2("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