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..923fec1b85c5f390f0f20f94cbb8c29a93c1d1db |
--- /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::GENERAL, 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 |