| OLD | NEW |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/media/router/discovery/mdns/cast_media_sink_service.h" | 5 #include "chrome/browser/media/router/discovery/mdns/cast_media_sink_service.h" |
| 6 #include "base/strings/stringprintf.h" | 6 #include "base/strings/stringprintf.h" |
| 7 #include "base/test/mock_callback.h" | 7 #include "base/test/mock_callback.h" |
| 8 #include "base/timer/mock_timer.h" | 8 #include "base/timer/mock_timer.h" |
| 9 #include "chrome/browser/media/router/discovery/mdns/mock_dns_sd_registry.h" | 9 #include "chrome/browser/media/router/discovery/mdns/mock_dns_sd_registry.h" |
| 10 #include "chrome/browser/media/router/test_helper.h" | 10 #include "chrome/browser/media/router/test_helper.h" |
| (...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 61 if (!audio_only) | 61 if (!audio_only) |
| 62 capabilities |= cast_channel::CastDeviceCapability::VIDEO_OUT; | 62 capabilities |= cast_channel::CastDeviceCapability::VIDEO_OUT; |
| 63 EXPECT_EQ(capabilities, cast_sink.cast_data().capabilities); | 63 EXPECT_EQ(capabilities, cast_sink.cast_data().capabilities); |
| 64 EXPECT_EQ(channel_id, cast_sink.cast_data().cast_channel_id); | 64 EXPECT_EQ(channel_id, cast_sink.cast_data().cast_channel_id); |
| 65 } | 65 } |
| 66 | 66 |
| 67 } // namespace | 67 } // namespace |
| 68 | 68 |
| 69 namespace media_router { | 69 namespace media_router { |
| 70 | 70 |
| 71 class MockCastSocketService : public cast_channel::CastSocketService { | |
| 72 public: | |
| 73 MOCK_METHOD4(OpenSocket, | |
| 74 int(const net::IPEndPoint& ip_endpoint, | |
| 75 net::NetLog* net_log, | |
| 76 const cast_channel::CastSocket::OnOpenCallback& open_cb, | |
| 77 cast_channel::CastSocket::Observer* observer)); | |
| 78 MOCK_CONST_METHOD1(GetSocket, cast_channel::CastSocket*(int channel_id)); | |
| 79 | |
| 80 private: | |
| 81 ~MockCastSocketService() {} | |
| 82 }; | |
| 83 | |
| 84 class CastMediaSinkServiceTest : public ::testing::Test { | 71 class CastMediaSinkServiceTest : public ::testing::Test { |
| 85 public: | 72 public: |
| 86 CastMediaSinkServiceTest() | 73 CastMediaSinkServiceTest() |
| 87 : mock_cast_socket_service_(new MockCastSocketService()), | 74 : mock_cast_socket_service_(new cast_channel::MockCastSocketService()), |
| 88 media_sink_service_( | 75 media_sink_service_( |
| 89 new CastMediaSinkService(mock_sink_discovered_cb_.Get(), | 76 new CastMediaSinkService(mock_sink_discovered_cb_.Get(), |
| 90 mock_cast_socket_service_.get())), | 77 mock_cast_socket_service_.get())), |
| 91 test_dns_sd_registry_(media_sink_service_.get()) {} | 78 test_dns_sd_registry_(media_sink_service_.get()) {} |
| 92 | 79 |
| 93 void SetUp() override { | 80 void SetUp() override { |
| 94 auto mock_timer = base::MakeUnique<base::MockTimer>( | 81 auto mock_timer = base::MakeUnique<base::MockTimer>( |
| 95 true /*retain_user_task*/, false /*is_repeating*/); | 82 true /*retain_user_task*/, false /*is_repeating*/); |
| 96 mock_timer_ = mock_timer.get(); | 83 mock_timer_ = mock_timer.get(); |
| 97 media_sink_service_->SetTimerForTest(std::move(mock_timer)); | 84 media_sink_service_->SetTimerForTest(std::move(mock_timer)); |
| 98 } | 85 } |
| 99 | 86 |
| 100 protected: | 87 protected: |
| 101 const content::TestBrowserThreadBundle thread_bundle_; | 88 const content::TestBrowserThreadBundle thread_bundle_; |
| 102 TestingProfile profile_; | 89 TestingProfile profile_; |
| 103 | 90 |
| 104 base::MockCallback<MediaSinkService::OnSinksDiscoveredCallback> | 91 base::MockCallback<MediaSinkService::OnSinksDiscoveredCallback> |
| 105 mock_sink_discovered_cb_; | 92 mock_sink_discovered_cb_; |
| 106 scoped_refptr<MockCastSocketService> mock_cast_socket_service_; | 93 std::unique_ptr<cast_channel::MockCastSocketService> |
| 94 mock_cast_socket_service_; |
| 107 scoped_refptr<CastMediaSinkService> media_sink_service_; | 95 scoped_refptr<CastMediaSinkService> media_sink_service_; |
| 108 MockDnsSdRegistry test_dns_sd_registry_; | 96 MockDnsSdRegistry test_dns_sd_registry_; |
| 109 base::MockTimer* mock_timer_; | 97 base::MockTimer* mock_timer_; |
| 110 | 98 |
| 111 DISALLOW_COPY_AND_ASSIGN(CastMediaSinkServiceTest); | 99 DISALLOW_COPY_AND_ASSIGN(CastMediaSinkServiceTest); |
| 112 }; | 100 }; |
| 113 | 101 |
| 114 TEST_F(CastMediaSinkServiceTest, TestReStartAfterStop) { | 102 TEST_F(CastMediaSinkServiceTest, TestReStartAfterStop) { |
| 115 EXPECT_CALL(test_dns_sd_registry_, AddObserver(media_sink_service_.get())) | 103 EXPECT_CALL(test_dns_sd_registry_, AddObserver(media_sink_service_.get())) |
| 116 .Times(2); | 104 .Times(2); |
| (...skipping 183 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 300 EXPECT_EQ(size_t(1), sinks.size()); | 288 EXPECT_EQ(size_t(1), sinks.size()); |
| 301 VerifyMediaSinkInternal(sinks[0], service2, 2, false); | 289 VerifyMediaSinkInternal(sinks[0], service2, 2, false); |
| 302 | 290 |
| 303 EXPECT_FALSE(mock_timer_->IsRunning()); | 291 EXPECT_FALSE(mock_timer_->IsRunning()); |
| 304 // Channel 1 is opened and timer is restarted. | 292 // Channel 1 is opened and timer is restarted. |
| 305 media_sink_service_->OnChannelOpenedOnUIThread(service1, 1, false); | 293 media_sink_service_->OnChannelOpenedOnUIThread(service1, 1, false); |
| 306 EXPECT_TRUE(mock_timer_->IsRunning()); | 294 EXPECT_TRUE(mock_timer_->IsRunning()); |
| 307 } | 295 } |
| 308 | 296 |
| 309 } // namespace media_router | 297 } // namespace media_router |
| OLD | NEW |