Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(175)

Side by Side Diff: chrome/browser/media/router/offscreen_presentation_manager_unittest.cc

Issue 2962623002: [Media Router] Clean up OffscreenPresentationManager params. (Closed)
Patch Set: Addressed Bin's comments Created 3 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
1 // Copyright 2016 The Chromium Authors. All rights reserved. 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 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 <vector> 5 #include <vector>
6 6
7 #include "base/bind.h" 7 #include "base/bind.h"
8 #include "base/stl_util.h" 8 #include "base/stl_util.h"
9 #include "chrome/browser/media/router/offscreen_presentation_manager.h" 9 #include "chrome/browser/media/router/offscreen_presentation_manager.h"
10 #include "chrome/browser/media/router/test_helper.h" 10 #include "chrome/browser/media/router/test_helper.h"
(...skipping 22 matching lines...) Expand all
33 33
34 MOCK_METHOD2(OnReceiverConnectionAvailableRaw, 34 MOCK_METHOD2(OnReceiverConnectionAvailableRaw,
35 void(const content::PresentationInfo&, 35 void(const content::PresentationInfo&,
36 blink::mojom::PresentationConnection*)); 36 blink::mojom::PresentationConnection*));
37 }; 37 };
38 38
39 class OffscreenPresentationManagerTest : public ::testing::Test { 39 class OffscreenPresentationManagerTest : public ::testing::Test {
40 public: 40 public:
41 OffscreenPresentationManagerTest() 41 OffscreenPresentationManagerTest()
42 : render_frame_host_id_(1, 1), 42 : render_frame_host_id_(1, 1),
43 presentation_url_(kPresentationUrl), 43 presentation_info_(GURL(kPresentationUrl), kPresentationId),
44 route_("route_1", 44 route_("route_1",
45 MediaSource("source_1"), 45 MediaSource("source_1"),
46 "sink_1", 46 "sink_1",
47 "", 47 "",
48 false, 48 false,
49 "", 49 "",
50 false) {} 50 false) {}
51 51
52 OffscreenPresentationManager* manager() { return &manager_; } 52 OffscreenPresentationManager* manager() { return &manager_; }
53 53
54 void VerifyPresentationsSize(size_t expected) { 54 void VerifyPresentationsSize(size_t expected) {
55 EXPECT_EQ(expected, manager_.offscreen_presentations_.size()); 55 EXPECT_EQ(expected, manager_.offscreen_presentations_.size());
56 } 56 }
57 57
58 void VerifyControllerSize(size_t expected, 58 void VerifyControllerSize(size_t expected,
59 const std::string& presentationId) { 59 const std::string& presentationId) {
60 EXPECT_TRUE( 60 EXPECT_TRUE(
61 base::ContainsKey(manager_.offscreen_presentations_, presentationId)); 61 base::ContainsKey(manager_.offscreen_presentations_, presentationId));
62 EXPECT_EQ(expected, manager_.offscreen_presentations_[presentationId] 62 EXPECT_EQ(expected, manager_.offscreen_presentations_[presentationId]
63 ->pending_controllers_.size()); 63 ->pending_controllers_.size());
64 } 64 }
65 65
66 void RegisterController(const std::string& presentation_id, 66 void RegisterController(const std::string& presentation_id,
67 content::PresentationConnectionPtr controller) { 67 content::PresentationConnectionPtr controller) {
68 content::PresentationConnectionRequest receiver_conn_request; 68 RegisterController(
69 manager()->RegisterOffscreenPresentationController( 69 content::PresentationInfo(GURL(kPresentationUrl), presentation_id),
70 presentation_id, presentation_url_, render_frame_host_id_, 70 render_frame_host_id_, std::move(controller));
71 std::move(controller), std::move(receiver_conn_request), route_);
72 } 71 }
73 72
74 void RegisterController(const RenderFrameHostId& render_frame_id, 73 void RegisterController(const RenderFrameHostId& render_frame_id,
75 content::PresentationConnectionPtr controller) { 74 content::PresentationConnectionPtr controller) {
76 content::PresentationConnectionRequest receiver_conn_request; 75 RegisterController(presentation_info_, render_frame_id,
77 manager()->RegisterOffscreenPresentationController( 76 std::move(controller));
78 kPresentationId, presentation_url_, render_frame_id,
79 std::move(controller), std::move(receiver_conn_request), route_);
80 } 77 }
81 78
82 void RegisterController(content::PresentationConnectionPtr controller) { 79 void RegisterController(content::PresentationConnectionPtr controller) {
80 RegisterController(presentation_info_, render_frame_host_id_,
81 std::move(controller));
82 }
83
84 void RegisterController(const content::PresentationInfo& presentation_info,
85 const RenderFrameHostId& render_frame_id,
86 content::PresentationConnectionPtr controller) {
83 content::PresentationConnectionRequest receiver_conn_request; 87 content::PresentationConnectionRequest receiver_conn_request;
84 manager()->RegisterOffscreenPresentationController( 88 manager()->RegisterOffscreenPresentationController(
85 kPresentationId, presentation_url_, render_frame_host_id_, 89 presentation_info, render_frame_id, std::move(controller),
86 std::move(controller), std::move(receiver_conn_request), route_); 90 std::move(receiver_conn_request), route_);
87 } 91 }
88 92
89 void RegisterReceiver( 93 void RegisterReceiver(
90 MockReceiverConnectionAvailableCallback& receiver_callback) { 94 MockReceiverConnectionAvailableCallback& receiver_callback) {
91 RegisterReceiver(kPresentationId, receiver_callback); 95 RegisterReceiver(kPresentationId, receiver_callback);
92 } 96 }
93 97
94 void RegisterReceiver( 98 void RegisterReceiver(
95 const std::string& presentation_id, 99 const std::string& presentation_id,
96 MockReceiverConnectionAvailableCallback& receiver_callback) { 100 MockReceiverConnectionAvailableCallback& receiver_callback) {
97 manager()->OnOffscreenPresentationReceiverCreated( 101 manager()->OnOffscreenPresentationReceiverCreated(
98 presentation_id, presentation_url_, 102 content::PresentationInfo(GURL(kPresentationUrl), presentation_id),
99 base::Bind(&MockReceiverConnectionAvailableCallback:: 103 base::Bind(&MockReceiverConnectionAvailableCallback::
100 OnReceiverConnectionAvailable, 104 OnReceiverConnectionAvailable,
101 base::Unretained(&receiver_callback))); 105 base::Unretained(&receiver_callback)));
102 } 106 }
103 107
104 void UnregisterController(const RenderFrameHostId& render_frame_id) { 108 void UnregisterController(const RenderFrameHostId& render_frame_id) {
105 manager()->UnregisterOffscreenPresentationController(kPresentationId, 109 manager()->UnregisterOffscreenPresentationController(kPresentationId,
106 render_frame_id); 110 render_frame_id);
107 } 111 }
108 112
109 void UnregisterController() { 113 void UnregisterController() {
110 manager()->UnregisterOffscreenPresentationController(kPresentationId, 114 manager()->UnregisterOffscreenPresentationController(kPresentationId,
111 render_frame_host_id_); 115 render_frame_host_id_);
112 } 116 }
113 117
114 void UnregisterReceiver() { 118 void UnregisterReceiver() {
115 manager()->OnOffscreenPresentationReceiverTerminated(kPresentationId); 119 manager()->OnOffscreenPresentationReceiverTerminated(kPresentationId);
116 } 120 }
117 121
118 private: 122 private:
119 const RenderFrameHostId render_frame_host_id_; 123 const RenderFrameHostId render_frame_host_id_;
120 GURL presentation_url_; 124 const content::PresentationInfo presentation_info_;
121 OffscreenPresentationManager manager_; 125 OffscreenPresentationManager manager_;
122 MediaRoute route_; 126 MediaRoute route_;
123 }; 127 };
124 128
125 TEST_F(OffscreenPresentationManagerTest, RegisterUnregisterController) { 129 TEST_F(OffscreenPresentationManagerTest, RegisterUnregisterController) {
126 content::PresentationConnectionPtr controller; 130 content::PresentationConnectionPtr controller;
127 RegisterController(std::move(controller)); 131 RegisterController(std::move(controller));
128 VerifyPresentationsSize(1); 132 VerifyPresentationsSize(1);
129 UnregisterController(); 133 UnregisterController();
130 VerifyPresentationsSize(0); 134 VerifyPresentationsSize(0);
(...skipping 190 matching lines...) Expand 10 before | Expand all | Expand 10 after
321 EXPECT_FALSE(manager()->GetRoute(kPresentationId)); 325 EXPECT_FALSE(manager()->GetRoute(kPresentationId));
322 content::PresentationConnectionPtr controller; 326 content::PresentationConnectionPtr controller;
323 RegisterController(std::move(controller)); 327 RegisterController(std::move(controller));
324 328
325 auto* actual_route = manager()->GetRoute(kPresentationId); 329 auto* actual_route = manager()->GetRoute(kPresentationId);
326 EXPECT_TRUE(actual_route); 330 EXPECT_TRUE(actual_route);
327 EXPECT_TRUE(route.Equals(*actual_route)); 331 EXPECT_TRUE(route.Equals(*actual_route));
328 } 332 }
329 333
330 } // namespace media_router 334 } // namespace media_router
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698