[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..138208bc24b9424f33215327730fdcce5a47b072
--- /dev/null
+++ b/chrome/browser/media/router/discovery/media_sink_internal_unittest.cc
@@ -0,0 +1,128 @@
+// 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"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace media_router {
+
+class MediaSinkInternalTest : public ::testing::Test {
+ public:
+  MediaSinkInternalTest()
+      : media_sink_(sink_id_, sink_name_, MediaSink::IconType::CAST) {}
+
+  DialSinkExtraData CreateDialSinkExtraData() {
+    return CreateDialSinkExtraData(model_name_, ip_address_, app_url_);
+  }
+
+  DialSinkExtraData CreateDialSinkExtraData(const std::string& model_name,
+                                            const std::string& ip_address,
+                                            const std::string& app_url) {
+    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;
+  }
+
+  CastSinkExtraData CreateCastSinkExtraData() {
+    return CreateCastSinkExtraData(model_name_, ip_address_, 2, 3);
+  }
+
+  CastSinkExtraData CreateCastSinkExtraData(const std::string& model_name,
+                                            const std::string& ip_address,
+                                            uint8_t capabilities,
+                                            int cast_channel_id) {
+    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;
+    return cast_extra_data;
+  }
+
+  MediaSink media_sink() { return media_sink_; }
+
+ private:
+  std::string sink_id_ = "sinkId123";
+  std::string sink_name_ = "The sink";
+  std::string ip_address_ = "192.168.1.2";
+  std::string model_name_ = "model name";
+  std::string app_url_ = "https://example.com";
+
+  MediaSink media_sink_;
+};
+
+TEST_F(MediaSinkInternalTest, TestIsValidSinkId) {
+  EXPECT_FALSE(MediaSinkInternal::IsValidSinkId(""));
+  EXPECT_TRUE(MediaSinkInternal::IsValidSinkId("rjuKv_yxhY4jg7QBIp0kbngLjR6A"));
+}
+
+TEST_F(MediaSinkInternalTest, TestConstructorAndAssignment) {
+  MediaSink sink = media_sink();
+  DialSinkExtraData dial_extra_data = CreateDialSinkExtraData();
+  CastSinkExtraData cast_extra_data = CreateCastSinkExtraData();
+
+  MediaSinkInternal generic_sink;
+  generic_sink.set_sink(sink);
+  MediaSinkInternal dial_sink(sink, dial_extra_data);
+  MediaSinkInternal cast_sink(sink, cast_extra_data);
+
+  MediaSinkInternal copied_generic_sink(generic_sink);
+  MediaSinkInternal copied_dial_sink(dial_sink);
+  MediaSinkInternal copied_cast_sink(cast_sink);
+
+  ASSERT_TRUE(generic_sink == copied_generic_sink);
+  ASSERT_TRUE(dial_sink == copied_dial_sink);
+  ASSERT_TRUE(cast_sink == copied_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);
+    }
+  }
+}
+
+TEST_F(MediaSinkInternalTest, TestSetExtraData) {
+  MediaSink sink = media_sink();
+  DialSinkExtraData dial_extra_data = CreateDialSinkExtraData();
+  CastSinkExtraData cast_extra_data = CreateCastSinkExtraData();
+
+  MediaSinkInternal dial_sink1;
+  dial_sink1.set_dial_data(dial_extra_data);
+  ASSERT_EQ(dial_extra_data, dial_sink1.dial_data());
+
+  MediaSinkInternal cast_sink1;
+  cast_sink1.set_cast_data(cast_extra_data);
+  ASSERT_EQ(cast_extra_data, cast_sink1.cast_data());
+
+  DialSinkExtraData dial_extra_data2 = CreateDialSinkExtraData(
+      "new_dial_model_name", "192.1.2.100", "https://example2.com");
+  CastSinkExtraData cast_extra_data2 =
+      CreateCastSinkExtraData("new_cast_model_name", "192.1.2.101", 4, 5);
+
+  MediaSinkInternal dial_sink2(sink, dial_extra_data);
+  dial_sink2.set_dial_data(dial_extra_data2);
+  ASSERT_EQ(dial_extra_data2, dial_sink2.dial_data());
+
+  MediaSinkInternal cast_sink2(sink, cast_extra_data);
+  cast_sink2.set_cast_data(cast_extra_data2);
+  ASSERT_EQ(cast_extra_data2, cast_sink2.cast_data());
+}
+
+}  // namespace media_router