[Media Router] Add MediaSink subtypes

chrome/browser/media/router/mojo/media_router_struct_traits_unittest.cc

Index: chrome/browser/media/router/mojo/media_router_struct_traits_unittest.cc
diff --git a/chrome/browser/media/router/mojo/media_router_struct_traits_unittest.cc b/chrome/browser/media/router/mojo/media_router_struct_traits_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..8febb5349ffa0a5d66ffbb750f0d3f9ce995d131
--- /dev/null
+++ b/chrome/browser/media/router/mojo/media_router_struct_traits_unittest.cc
@@ -0,0 +1,119 @@
+// 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/mojo/media_router_struct_traits.h"
+
+#include "base/message_loop/message_loop.h"
+#include "chrome/browser/media/router/cast_media_sink.h"
+#include "chrome/browser/media/router/dial_media_sink.h"
+#include "chrome/browser/media/router/media_sink.h"
+#include "chrome/browser/media/router/mojo/media_router.mojom.h"
+#include "chrome/browser/media/router/mojo/media_router_traits_test_service.mojom.h"
+#include "mojo/public/cpp/bindings/binding_set.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace media_router {
+
+class MediaRouterStructTraitsTest
+    : public testing::Test,
+      public media_router::mojom::MediaRouterTraitsTestService {
+ public:
+  MediaRouterStructTraitsTest() {}
+
+ protected:
+  mojom::MediaRouterTraitsTestServicePtr GetTraitsTestProxy() {
+    return traits_test_bindings_.CreateInterfacePtrAndBind(this);
+  }
+
+ private:
+  // MediaRouterTraitsTestService Impl
+  void EchoMediaSink(std::unique_ptr<MediaSink> sink,
+                     const EchoMediaSinkCallback& callback) override {
+    callback.Run(std::move(sink));
+  }
+
+  base::MessageLoop loop_;
+  mojo::BindingSet<MediaRouterTraitsTestService> traits_test_bindings_;
+
+  DISALLOW_COPY_AND_ASSIGN(MediaRouterStructTraitsTest);
+};
+
+TEST_F(MediaRouterStructTraitsTest, DialMediaSink) {
+  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");
+  GURL app_url("https://example.com");
+
+  DialMediaSink* dial_sink = new DialMediaSink(sink_id, sink_name, icon_type);
+  dial_sink->set_ip_address(ip_address);
+  dial_sink->set_app_url(app_url);
+
+  mojom::MediaRouterTraitsTestServicePtr proxy = GetTraitsTestProxy();
+  std::unique_ptr<MediaSink> output;
+  proxy->EchoMediaSink(base::WrapUnique(dial_sink), &output);
+
+  EXPECT_EQ(MediaSink::SinkType::DIAL, output->type());
+  const DialMediaSink* out_sink = DialMediaSink::Cast(output.get());
+  EXPECT_EQ(sink_id, out_sink->id());
+  EXPECT_EQ(sink_name, out_sink->name());
+  EXPECT_EQ(icon_type, out_sink->icon_type());
+  EXPECT_EQ(ip_address, out_sink->ip_address());
+  EXPECT_EQ(app_url, out_sink->app_url());
+}
+
+TEST_F(MediaRouterStructTraitsTest, CastMediaSink) {
+  MediaSink::Id sink_id("sinkId123");
+  std::string sink_name("The sink");
+  MediaSink::IconType icon_type(MediaSink::IconType::CAST);
+  int capabilities = 20;
+  std::string ip_address("192.168.1.2");
+  int cast_channel_id = 3;
+
+  CastMediaSink* cast_sink = new CastMediaSink(sink_id, sink_name, icon_type);
+  cast_sink->set_capabilities(capabilities);
+  cast_sink->set_ip_address(ip_address);
+  cast_sink->set_cast_channel_id(cast_channel_id);
+
+  mojom::MediaRouterTraitsTestServicePtr proxy = GetTraitsTestProxy();
+  std::unique_ptr<MediaSink> output;
+  proxy->EchoMediaSink(base::WrapUnique(cast_sink), &output);
+
+  EXPECT_EQ(MediaSink::SinkType::CAST, output->type());
+  const CastMediaSink* out_sink = CastMediaSink::Cast(output.get());
+  EXPECT_EQ(sink_id, out_sink->id());
+  EXPECT_EQ(sink_name, out_sink->name());
+  EXPECT_EQ(icon_type, out_sink->icon_type());
+  EXPECT_EQ(capabilities, out_sink->capabilities());
+  EXPECT_EQ(ip_address, out_sink->ip_address());
+  EXPECT_EQ(cast_channel_id, out_sink->cast_channel_id());
+}
+
+TEST_F(MediaRouterStructTraitsTest, GeneralMediaSink) {
+  MediaSink::Id sink_id("sinkId123");
+  std::string sink_name("The sink");
+  MediaSink::IconType icon_type(MediaSink::IconType::CAST);
+  std::string description("description");
+  std::string domain("domain");
+  std::string model_name("model name");
+
+  MediaSink* media_sink = new MediaSink(sink_id, sink_name, icon_type);
+  media_sink->set_description(description);
+  media_sink->set_domain(domain);
+  media_sink->set_model_name(model_name);
+
+  mojom::MediaRouterTraitsTestServicePtr proxy = GetTraitsTestProxy();
+  std::unique_ptr<MediaSink> output;
+  proxy->EchoMediaSink(base::WrapUnique(media_sink), &output);
+
+  EXPECT_EQ(MediaSink::SinkType::GENERIC, output->type());
+  EXPECT_EQ(sink_name, output->name());
+  EXPECT_EQ(sink_id, output->id());
+  EXPECT_EQ(description, output->description());
+  EXPECT_EQ(domain, output->domain());
+  EXPECT_EQ(icon_type, output->icon_type());
+  EXPECT_EQ(model_name, output->model_name());
+}
+
+}  // namespace media_router