OLD | NEW |
---|---|
(Empty) | |
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 | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/media/router/discovery/dial/dial_media_sink_service_del egate.h" | |
6 #include "base/run_loop.h" | |
7 #include "base/test/mock_callback.h" | |
8 #include "chrome/browser/media/router/discovery/dial/dial_media_sink_service.h" | |
9 #include "chrome/browser/media/router/test_helper.h" | |
10 #include "chrome/test/base/testing_profile.h" | |
11 #include "content/public/test/test_browser_thread_bundle.h" | |
12 #include "testing/gmock/include/gmock/gmock.h" | |
13 #include "testing/gtest/include/gtest/gtest.h" | |
14 | |
15 using ::testing::_; | |
16 using ::testing::Return; | |
17 | |
18 namespace media_router { | |
19 | |
20 class MockDialMediaSinkService : public DialMediaSinkService { | |
21 public: | |
22 MockDialMediaSinkService(const MediaSinkService::OnSinksDiscoveredCallback& | |
23 sink_discovery_callback, | |
24 net::URLRequestContextGetter* request_context) | |
25 : DialMediaSinkService(sink_discovery_callback, request_context) {} | |
26 ~MockDialMediaSinkService() override {} | |
27 | |
28 MOCK_METHOD0(Start, void()); | |
29 MOCK_METHOD0(Stop, void()); | |
30 | |
31 void OnSinksDiscovered() { | |
32 sink_discovery_callback_.Run(std::vector<MediaSinkInternal>()); | |
33 } | |
34 }; | |
35 | |
36 class DialMediaSinkServiceDelegateTest : public ::testing::Test { | |
37 public: | |
38 DialMediaSinkServiceDelegateTest() | |
39 : delegate_(new DialMediaSinkServiceDelegate()) { | |
40 std::unique_ptr<MockDialMediaSinkService, | |
41 content::BrowserThread::DeleteOnIOThread> | |
42 mock_dial_media_sink_service; | |
mark a. foltz
2017/05/12 21:37:38
= MakeUnique<MockDialMediaSinkService>(...)
zhaobin
2017/05/12 23:03:38
MakeUnique<> cannot take content::BrowserThread::D
| |
43 mock_dial_media_sink_service.reset(new MockDialMediaSinkService( | |
44 base::Bind(&DialMediaSinkServiceDelegate::OnSinksDiscoveredOnIOThread, | |
45 base::Unretained(delegate_.get())), | |
46 profile_.GetRequestContext())); | |
47 mock_service_ = mock_dial_media_sink_service.get(); | |
48 delegate_->SetDialMediaSinkServiceForTest( | |
49 std::move(mock_dial_media_sink_service)); | |
50 } | |
51 | |
52 protected: | |
53 const content::TestBrowserThreadBundle thread_bundle_; | |
54 TestingProfile profile_; | |
55 | |
56 base::MockCallback<MediaSinkService::OnSinksDiscoveredCallback> | |
57 mock_sink_discovered_cb_; | |
58 | |
59 MockDialMediaSinkService* mock_service_; | |
60 scoped_refptr<DialMediaSinkServiceDelegate> delegate_; | |
61 | |
62 DISALLOW_COPY_AND_ASSIGN(DialMediaSinkServiceDelegateTest); | |
63 }; | |
64 | |
65 TEST_F(DialMediaSinkServiceDelegateTest, TestStart) { | |
66 EXPECT_CALL(*mock_service_, Start()); | |
67 | |
68 delegate_->Start(mock_sink_discovered_cb_.Get(), | |
69 profile_.GetRequestContext()); | |
70 base::RunLoop().RunUntilIdle(); | |
mark a. foltz
2017/05/12 21:37:38
Nit: This can be put in TearDown().
zhaobin
2017/05/12 23:03:38
Done.
| |
71 } | |
72 | |
73 TEST_F(DialMediaSinkServiceDelegateTest, TestStop) { | |
74 EXPECT_CALL(*mock_service_, Stop()); | |
75 | |
76 delegate_->Stop(); | |
77 base::RunLoop().RunUntilIdle(); | |
78 } | |
79 | |
80 TEST_F(DialMediaSinkServiceDelegateTest, TestOnSinksDiscovered) { | |
81 EXPECT_CALL(*mock_service_, Start()); | |
82 EXPECT_CALL(mock_sink_discovered_cb_, Run(_)); | |
mark a. foltz
2017/05/12 21:37:38
Nit: .Times(1)
zhaobin
2017/05/12 23:03:39
Done.
| |
83 | |
84 delegate_->Start(mock_sink_discovered_cb_.Get(), | |
85 profile_.GetRequestContext()); | |
86 mock_service_->OnSinksDiscovered(); | |
87 base::RunLoop().RunUntilIdle(); | |
88 } | |
89 | |
90 TEST_F(DialMediaSinkServiceDelegateTest, TestOnSinksDiscoveredAfterStop) { | |
91 EXPECT_CALL(*mock_service_, Start()); | |
92 EXPECT_CALL(*mock_service_, Stop()); | |
93 EXPECT_CALL(mock_sink_discovered_cb_, Run(_)).Times(0); | |
94 | |
95 delegate_->Start(mock_sink_discovered_cb_.Get(), | |
96 profile_.GetRequestContext()); | |
97 delegate_->Stop(); | |
98 mock_service_->OnSinksDiscovered(); | |
99 base::RunLoop().RunUntilIdle(); | |
100 } | |
101 | |
102 } // namespace media_router | |
OLD | NEW |