OLD | NEW |
---|---|
(Empty) | |
1 // Copyright 2016 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 <vector> | |
6 | |
7 #include "base/bind.h" | |
8 #include "chrome/browser/media/router/offscreen_presentation_manager.h" | |
9 #include "chrome/browser/media/router/test_helper.h" | |
10 #include "testing/gmock/include/gmock/gmock.h" | |
11 #include "testing/gtest/include/gtest/gtest.h" | |
12 | |
13 using testing::_; | |
14 | |
15 namespace media_router { | |
16 | |
17 namespace { | |
18 const char kPresentationId[] = "presentationId"; | |
19 const char kPresentationId2[] = "presentationId2"; | |
20 } // namespace | |
21 | |
22 class MockOffscreenPresentationClient | |
23 : public content::OffscreenPresentationClient { | |
24 public: | |
25 MOCK_METHOD1(OnReceiverConnectionAvailable, | |
26 void(const content::PresentationSessionInfo&)); | |
27 MOCK_METHOD1(AddOffscreenPresentationObserver, | |
28 void(OffscreenPresentationClient*)); | |
29 MOCK_METHOD1(RemoveOffscreenPresentationObserver, | |
30 void(OffscreenPresentationClient*)); | |
31 }; | |
32 | |
33 class OffscreenPresentationManagerTest : public ::testing::Test { | |
34 public: | |
35 OffscreenPresentationManager* manager() { return &manager_; } | |
36 | |
37 void verifyPresentationsSize(size_t expected); | |
38 void verifyControllerSize(size_t expected, | |
39 const OffscreenPresentation* presentation); | |
40 | |
41 private: | |
42 OffscreenPresentationManager manager_; | |
43 }; | |
44 | |
45 void OffscreenPresentationManagerTest::verifyPresentationsSize( | |
imcheng
2016/09/19 23:01:17
These don't need to be their own functions since t
zhaobin
2016/09/23 17:18:18
Create these functions to access private variable
| |
46 size_t expected) { | |
47 EXPECT_EQ(expected, manager_.offscreen_presentations_.size()); | |
48 } | |
49 | |
50 void OffscreenPresentationManagerTest::verifyControllerSize( | |
51 size_t expected, | |
52 const OffscreenPresentation* presentation) { | |
53 EXPECT_EQ(expected, presentation->controllers_.size()); | |
54 } | |
55 | |
56 TEST_F(OffscreenPresentationManagerTest, | |
57 RegisterControllerCreateNewPresentation) { | |
58 MockOffscreenPresentationClient controller; | |
59 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
60 &controller); | |
61 verifyPresentationsSize(1); | |
62 } | |
63 | |
64 TEST_F(OffscreenPresentationManagerTest, RegisterOneReceiverDoNothing) { | |
65 MockOffscreenPresentationClient receiver; | |
66 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
67 verifyPresentationsSize(0); | |
68 } | |
69 | |
70 TEST_F(OffscreenPresentationManagerTest, UnregisterControllerNoCrash) { | |
71 MockOffscreenPresentationClient controller; | |
72 manager()->UnregisterOffscreenPresentationController(kPresentationId, | |
73 &controller); | |
74 verifyPresentationsSize(0); | |
75 } | |
76 | |
77 TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverNoCrash) { | |
78 manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); | |
79 verifyPresentationsSize(0); | |
80 } | |
81 | |
82 TEST_F(OffscreenPresentationManagerTest, | |
83 RegisterControllerCreatePresentationWithTwoControllers) { | |
84 MockOffscreenPresentationClient controller1; | |
85 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
86 &controller1); | |
87 MockOffscreenPresentationClient controller2; | |
88 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
89 &controller2); | |
90 verifyPresentationsSize(1); | |
91 } | |
92 | |
93 TEST_F(OffscreenPresentationManagerTest, | |
94 RegisterControllerCreateTwoPresentations) { | |
95 MockOffscreenPresentationClient controller1; | |
96 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
97 &controller1); | |
98 MockOffscreenPresentationClient controller2; | |
99 manager()->RegisterOffscreenPresentationController(kPresentationId2, | |
100 &controller2); | |
101 verifyPresentationsSize(2); | |
102 } | |
103 | |
104 TEST_F(OffscreenPresentationManagerTest, | |
105 RegisterControllerAndReceiverFireOnConnectionAvailable) { | |
106 MockOffscreenPresentationClient controller; | |
107 MockOffscreenPresentationClient receiver; | |
108 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
109 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
110 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); | |
111 | |
112 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
113 &controller); | |
114 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
115 } | |
116 | |
117 TEST_F(OffscreenPresentationManagerTest, RegisterUnregisterController) { | |
118 MockOffscreenPresentationClient controller; | |
119 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
120 &controller); | |
121 manager()->UnregisterOffscreenPresentationController(kPresentationId, | |
122 &controller); | |
123 verifyPresentationsSize(0); | |
124 } | |
125 | |
126 TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverOnly) { | |
127 MockOffscreenPresentationClient controller; | |
128 MockOffscreenPresentationClient receiver; | |
129 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
130 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
131 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); | |
132 | |
133 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
134 &controller); | |
135 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
136 | |
137 EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)); | |
138 EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)); | |
139 manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); | |
140 | |
141 verifyPresentationsSize(0); | |
142 } | |
143 | |
144 TEST_F(OffscreenPresentationManagerTest, UnregisterControllerOnly) { | |
145 MockOffscreenPresentationClient controller; | |
146 MockOffscreenPresentationClient receiver; | |
147 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
148 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
149 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); | |
150 | |
151 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
152 &controller); | |
153 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
154 | |
155 EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)); | |
156 EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)); | |
157 manager()->UnregisterOffscreenPresentationController(kPresentationId, | |
158 &controller); | |
159 verifyPresentationsSize(0); | |
160 } | |
161 | |
162 TEST_F(OffscreenPresentationManagerTest, UnregisterControllerReceiver) { | |
163 MockOffscreenPresentationClient controller; | |
164 MockOffscreenPresentationClient receiver; | |
165 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
166 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
167 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); | |
168 | |
169 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
170 &controller); | |
171 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
172 | |
173 EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)) | |
174 .Times(1); | |
175 EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)) | |
176 .Times(1); | |
177 manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); | |
178 manager()->UnregisterOffscreenPresentationController(kPresentationId, | |
179 &controller); | |
180 | |
181 verifyPresentationsSize(0); | |
182 } | |
183 | |
184 TEST_F(OffscreenPresentationManagerTest, UnregisterReceiverController) { | |
185 MockOffscreenPresentationClient controller; | |
186 MockOffscreenPresentationClient receiver; | |
187 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
188 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
189 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)); | |
190 | |
191 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
192 &controller); | |
193 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
194 | |
195 EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)) | |
196 .Times(1); | |
197 EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(&controller)) | |
198 .Times(1); | |
199 manager()->UnregisterOffscreenPresentationController(kPresentationId, | |
200 &controller); | |
201 manager()->UnregisterOffscreenPresentationReceiver(kPresentationId); | |
202 | |
203 verifyPresentationsSize(0); | |
204 } | |
205 | |
206 TEST_F(OffscreenPresentationManagerTest, | |
207 RegisterTwoControllersThenReceiverFireOnConnectionAvailableTwice) { | |
208 MockOffscreenPresentationClient controller; | |
209 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
210 &controller); | |
211 MockOffscreenPresentationClient controller2; | |
212 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
213 &controller2); | |
214 | |
215 MockOffscreenPresentationClient receiver; | |
216 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
217 EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver)); | |
218 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
219 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller2)); | |
220 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(2); | |
221 | |
222 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
223 } | |
224 | |
225 TEST_F( | |
226 OffscreenPresentationManagerTest, | |
227 RegisterControllerThenReceiverThenConontrollerFireOnConnectionAvailableTwice ) { | |
228 MockOffscreenPresentationClient controller; | |
229 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
230 &controller); | |
231 MockOffscreenPresentationClient receiver; | |
232 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
233 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
234 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(2); | |
235 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
236 MockOffscreenPresentationClient controller2; | |
237 EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver)); | |
238 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller2)); | |
239 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
240 &controller2); | |
241 } | |
242 | |
243 TEST_F(OffscreenPresentationManagerTest, | |
244 OnePresentationUnregisterFirstController) { | |
245 MockOffscreenPresentationClient controller; | |
246 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
247 &controller); | |
248 MockOffscreenPresentationClient controller2; | |
249 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
250 &controller2); | |
251 MockOffscreenPresentationClient receiver; | |
252 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)); | |
253 EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver)); | |
254 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)); | |
255 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller2)); | |
256 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(2); | |
257 | |
258 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
259 | |
260 EXPECT_CALL(controller, RemoveOffscreenPresentationObserver(&receiver)); | |
261 EXPECT_CALL(controller2, RemoveOffscreenPresentationObserver(&receiver)); | |
262 EXPECT_CALL(receiver, RemoveOffscreenPresentationObserver(_)).Times(2); | |
263 | |
264 manager()->UnregisterOffscreenPresentationController(kPresentationId, | |
265 &controller); | |
266 manager()->UnregisterOffscreenPresentationController(kPresentationId, | |
267 &controller2); | |
268 | |
269 verifyPresentationsSize(0); | |
270 } | |
271 | |
272 TEST_F(OffscreenPresentationManagerTest, TwoPresentations) { | |
273 MockOffscreenPresentationClient controller; | |
274 manager()->RegisterOffscreenPresentationController(kPresentationId, | |
275 &controller); | |
276 MockOffscreenPresentationClient receiver; | |
277 EXPECT_CALL(controller, AddOffscreenPresentationObserver(&receiver)).Times(1); | |
278 EXPECT_CALL(receiver, AddOffscreenPresentationObserver(&controller)).Times(1); | |
279 EXPECT_CALL(receiver, OnReceiverConnectionAvailable(_)).Times(1); | |
280 | |
281 manager()->RegisterOffscreenPresentationReceiver(kPresentationId, &receiver); | |
282 | |
283 MockOffscreenPresentationClient controller2; | |
284 manager()->RegisterOffscreenPresentationController(kPresentationId2, | |
285 &controller2); | |
286 MockOffscreenPresentationClient receiver2; | |
287 EXPECT_CALL(controller2, AddOffscreenPresentationObserver(&receiver2)) | |
288 .Times(1); | |
289 EXPECT_CALL(receiver2, AddOffscreenPresentationObserver(&controller2)) | |
290 .Times(1); | |
291 EXPECT_CALL(receiver2, OnReceiverConnectionAvailable(_)).Times(1); | |
292 | |
293 manager()->RegisterOffscreenPresentationReceiver(kPresentationId2, | |
294 &receiver2); | |
295 | |
296 verifyPresentationsSize(2); | |
297 | |
298 EXPECT_CALL(receiver2, RemoveOffscreenPresentationObserver(&controller2)) | |
299 .Times(1); | |
300 EXPECT_CALL(controller2, RemoveOffscreenPresentationObserver(&receiver2)) | |
301 .Times(1); | |
302 manager()->UnregisterOffscreenPresentationController(kPresentationId2, | |
303 &controller2); | |
304 | |
305 verifyPresentationsSize(1); | |
306 } | |
307 | |
308 } // namespace media_router | |
OLD | NEW |