[Media Router] Add MediaSink subtypes

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

Index: chrome/browser/media/router/discovery/media_sink_internal_unittest.cc
diff --git a/chrome/browser/media/router/discovery/media_sink_internal_unittest.cc b/chrome/browser/media/router/discovery/media_sink_internal_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b8d2bb78f90a011d48e2e1bc590e45fe291f48bd
--- /dev/null
+++ b/chrome/browser/media/router/discovery/media_sink_internal_unittest.cc
@@ -0,0 +1,66 @@
+// 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_internal.h"
+#include "testing/gmock/include/gmock/gmock.h"
+
+namespace media_router {
+
+TEST(MediaSinkInternalTest, TestIsValidSinkId) {
+  EXPECT_FALSE(MediaSinkInternal::IsValidSinkId(""));
+  EXPECT_TRUE(MediaSinkInternal::IsValidSinkId("rjuKv_yxhY4jg7QBIp0kbngLjR6A"));
+}
+
+TEST(MediaSinkInternalTest, TestConstructorAndAssignment) {
+  MediaSink::Id sink_id("sinkId123");
+  std::string sink_name("The sink");
+  MediaSink::IconType icon_type(MediaSink::IconType::CAST);
+  std::string ip_address("192.168.1.2");
+  std::string model_name("model name");
+  GURL app_url("https://example.com");
+  MediaSink sink(sink_id, sink_name, icon_type);
+
+  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 = app_url;
+
+  CastSinkExtraData cast_extra_data;
+  EXPECT_TRUE(cast_extra_data.ip_address.AssignFromIPLiteral(ip_address));
+  cast_extra_data.model_name = model_name;
+  cast_extra_data.capabilities = 2;
+  cast_extra_data.cast_channel_id = 3;
+
+  MediaSinkInternal generic_sink(sink);
+  MediaSinkInternal dial_sink(sink, dial_extra_data);
+  MediaSinkInternal cast_sink(sink, cast_extra_data);
+
+  MediaSinkInternal copy_generic_sink(generic_sink);
+  MediaSinkInternal copy_dial_sink(dial_sink);
+  MediaSinkInternal copy_cast_sink(cast_sink);
+
+  ASSERT_TRUE(generic_sink == copy_generic_sink);
+  ASSERT_TRUE(dial_sink == copy_dial_sink);
+  ASSERT_TRUE(cast_sink == copy_cast_sink);
+
+  MediaSinkInternal assigned_empty_sink;
+  MediaSinkInternal assigned_generic_sink = generic_sink;
+  MediaSinkInternal assigned_dial_sink = dial_sink;
+  MediaSinkInternal assigned_cast_sink = cast_sink;
+
+  std::vector<MediaSinkInternal> assigned_sinks(
+      {assigned_empty_sink, assigned_generic_sink, assigned_dial_sink,
+       assigned_cast_sink});
+  std::vector<MediaSinkInternal> original_sinks(
+      {generic_sink, dial_sink, cast_sink});
+
+  for (auto& actual_sink : assigned_sinks) {
+    for (const auto& original_sink : original_sinks) {
+      actual_sink = original_sink;
+      EXPECT_EQ(original_sink, actual_sink);
+    }
+  }
+}
+
+}  // namespace media_router