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/browser_presentation_connection_proxy.h" | |
6 | |
7 #include "base/run_loop.h" | |
8 #include "base/test/mock_callback.h" | |
9 #include "chrome/browser/media/router/media_source.h" | |
10 #include "chrome/browser/media/router/media_source_helper.h" | |
11 #include "chrome/browser/media/router/mock_media_router.h" | |
12 #include "content/public/test/test_browser_thread_bundle.h" | |
13 #include "testing/gmock/include/gmock/gmock.h" | |
14 | |
15 using ::testing::_; | |
16 | |
17 namespace media_router { | |
18 | |
19 constexpr char kMediaRouteId[] = "MockRouteId"; | |
20 | |
21 class MockOnMessageCallback { | |
imcheng
2017/01/31 22:22:48
This can be removed now
zhaobin
2017/02/01 04:16:09
Done.
| |
22 public: | |
23 MOCK_METHOD1(OnMessage, void(bool)); | |
24 }; | |
25 | |
26 class MockPresentationConnectionProxy | |
27 : public NON_EXPORTED_BASE(blink::mojom::PresentationConnection) { | |
28 public: | |
29 void OnMessage(blink::mojom::ConnectionMessagePtr message, | |
30 const OnMessageCallback& on_message_callback) override { | |
31 OnMessageRaw(message.get(), on_message_callback); | |
32 } | |
33 | |
34 MOCK_METHOD2(OnMessageRaw, | |
35 void(const blink::mojom::ConnectionMessage*, | |
36 const OnMessageCallback&)); | |
37 MOCK_METHOD1(DidChangeState, | |
38 void(blink::mojom::PresentationConnectionState state)); | |
39 }; | |
40 | |
41 class BrowserPresentationConnectionProxyTest : public ::testing::Test { | |
42 public: | |
43 BrowserPresentationConnectionProxyTest() = default; | |
44 | |
45 void SetUp() override { | |
46 mock_controller_connection_proxy_ = | |
47 base::MakeUnique<MockPresentationConnectionProxy>(); | |
48 blink::mojom::PresentationConnectionPtr controller_connection_ptr; | |
49 mojo::Binding<blink::mojom::PresentationConnection> binding( | |
50 mock_controller_connection_proxy_.get(), | |
51 mojo::MakeRequest(&controller_connection_ptr)); | |
52 EXPECT_CALL( | |
53 *mock_controller_connection_proxy_, | |
54 DidChangeState(blink::mojom::PresentationConnectionState::CONNECTED)); | |
55 | |
56 blink::mojom::PresentationConnectionPtr receiver_connection_ptr; | |
57 | |
58 base::RunLoop run_loop; | |
59 browser_connection_proxy_ = | |
60 base::MakeUnique<BrowserPresentationConnectionProxy>( | |
61 &mock_router_, "MockRouteId", | |
62 mojo::MakeRequest(&receiver_connection_ptr), | |
63 std::move(controller_connection_ptr)); | |
64 run_loop.RunUntilIdle(); | |
65 } | |
66 | |
67 void TearDown() override { | |
68 browser_connection_proxy_.reset(); | |
69 mock_controller_connection_proxy_.reset(); | |
70 } | |
71 | |
72 BrowserPresentationConnectionProxy* browser_connection_proxy() { | |
73 return browser_connection_proxy_.get(); | |
74 } | |
75 | |
76 MockMediaRouter* mock_router() { return &mock_router_; } | |
77 | |
78 private: | |
79 std::unique_ptr<MockPresentationConnectionProxy> | |
80 mock_controller_connection_proxy_; | |
81 std::unique_ptr<BrowserPresentationConnectionProxy> browser_connection_proxy_; | |
82 MockMediaRouter mock_router_; | |
83 content::TestBrowserThreadBundle thread_bundle_; | |
84 }; | |
85 | |
86 TEST_F(BrowserPresentationConnectionProxyTest, TestOnMessageTextMessage) { | |
87 std::string message = "test message"; | |
88 blink::mojom::ConnectionMessagePtr session_message = | |
89 blink::mojom::ConnectionMessage::New(); | |
90 session_message->type = blink::mojom::PresentationMessageType::TEXT; | |
91 session_message->message = message; | |
92 | |
93 base::MockCallback<base::Callback<void(bool success)>> | |
imcheng
2017/01/31 22:22:48
sucess param not needed.
zhaobin
2017/02/01 04:16:09
Done.
| |
94 mock_on_message_callback; | |
95 EXPECT_CALL(*mock_router(), SendRouteMessage(kMediaRouteId, message, _)); | |
96 | |
97 browser_connection_proxy()->OnMessage(std::move(session_message), | |
98 mock_on_message_callback.Get()); | |
99 } | |
100 | |
101 TEST_F(BrowserPresentationConnectionProxyTest, TestOnMessageBinaryMessage) { | |
102 std::vector<uint8_t> expected_data; | |
103 expected_data.push_back(42); | |
104 expected_data.push_back(36); | |
105 | |
106 blink::mojom::ConnectionMessagePtr session_message = | |
107 blink::mojom::ConnectionMessage::New(); | |
108 session_message->type = blink::mojom::PresentationMessageType::BINARY; | |
109 session_message->data = expected_data; | |
110 | |
111 base::MockCallback<base::Callback<void(bool success)>> | |
imcheng
2017/01/31 22:22:48
sucess param not needed.
zhaobin
2017/02/01 04:16:09
Done.
| |
112 mock_on_message_callback; | |
113 EXPECT_CALL(*mock_router(), SendRouteBinaryMessageInternal(_, _, _)) | |
114 .WillOnce(::testing::Invoke([this, &expected_data]( | |
115 const MediaRoute::Id& route_id, std::vector<uint8_t>* data, | |
116 const BrowserPresentationConnectionProxy::OnMessageCallback& | |
117 callback) { EXPECT_EQ(expected_data, *data); })); | |
118 | |
119 browser_connection_proxy()->OnMessage(std::move(session_message), | |
120 mock_on_message_callback.Get()); | |
121 } | |
122 | |
123 } // namespace media_router | |
OLD | NEW |