OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/memory/ref_counted.h" | 5 #include "base/memory/ref_counted.h" |
6 #include "base/message_loop.h" | 6 #include "base/message_loop.h" |
7 #include "ppapi/c/pp_errors.h" | 7 #include "ppapi/c/pp_errors.h" |
8 #include "ppapi/c/ppb_websocket.h" | 8 #include "ppapi/c/ppb_websocket.h" |
9 #include "ppapi/c/ppb_var.h" | 9 #include "ppapi/c/ppb_var.h" |
| 10 #include "ppapi/proxy/locking_resource_releaser.h" |
10 #include "ppapi/proxy/websocket_resource.h" | 11 #include "ppapi/proxy/websocket_resource.h" |
11 #include "ppapi/proxy/ppapi_messages.h" | 12 #include "ppapi/proxy/ppapi_messages.h" |
12 #include "ppapi/proxy/ppapi_proxy_test.h" | 13 #include "ppapi/proxy/ppapi_proxy_test.h" |
| 14 #include "ppapi/shared_impl/ppapi_globals.h" |
13 #include "ppapi/shared_impl/ppb_var_shared.h" | 15 #include "ppapi/shared_impl/ppb_var_shared.h" |
| 16 #include "ppapi/shared_impl/proxy_lock.h" |
| 17 #include "ppapi/shared_impl/resource_tracker.h" |
14 #include "ppapi/shared_impl/scoped_pp_resource.h" | 18 #include "ppapi/shared_impl/scoped_pp_resource.h" |
15 #include "ppapi/shared_impl/scoped_pp_var.h" | 19 #include "ppapi/shared_impl/scoped_pp_var.h" |
16 #include "ppapi/shared_impl/tracked_callback.h" | 20 #include "ppapi/shared_impl/tracked_callback.h" |
17 #include "ppapi/shared_impl/var.h" | 21 #include "ppapi/shared_impl/var.h" |
18 #include "ppapi/thunk/thunk.h" | 22 #include "ppapi/thunk/thunk.h" |
19 | 23 |
20 namespace ppapi { | 24 namespace ppapi { |
21 namespace proxy { | 25 namespace proxy { |
22 | 26 |
23 namespace { | 27 namespace { |
(...skipping 28 matching lines...) Expand all Loading... |
52 TEST_F(WebSocketResourceTest, Connect) { | 56 TEST_F(WebSocketResourceTest, Connect) { |
53 const PPB_WebSocket_1_0* websocket_iface = | 57 const PPB_WebSocket_1_0* websocket_iface = |
54 thunk::GetPPB_WebSocket_1_0_Thunk(); | 58 thunk::GetPPB_WebSocket_1_0_Thunk(); |
55 | 59 |
56 std::string url("ws://ws.google.com"); | 60 std::string url("ws://ws.google.com"); |
57 std::string protocol0("x-foo"); | 61 std::string protocol0("x-foo"); |
58 std::string protocol1("x-bar"); | 62 std::string protocol1("x-bar"); |
59 PP_Var url_var = MakeStringVar(url); | 63 PP_Var url_var = MakeStringVar(url); |
60 PP_Var protocols[] = { MakeStringVar(protocol0), MakeStringVar(protocol1) }; | 64 PP_Var protocols[] = { MakeStringVar(protocol0), MakeStringVar(protocol1) }; |
61 | 65 |
62 ScopedPPResource res(ScopedPPResource::PassRef(), | 66 LockingResourceReleaser res(websocket_iface->Create(pp_instance())); |
63 websocket_iface->Create(pp_instance())); | |
64 | 67 |
65 int32_t result = | 68 int32_t result = websocket_iface->Connect(res.get(), url_var, protocols, 2, |
66 websocket_iface->Connect(res, url_var, protocols, 2, MakeCallback()); | 69 MakeCallback()); |
67 ASSERT_EQ(PP_OK_COMPLETIONPENDING, result); | 70 ASSERT_EQ(PP_OK_COMPLETIONPENDING, result); |
68 | 71 |
69 // Should be sent a "Connect" message. | 72 // Should be sent a "Connect" message. |
70 ResourceMessageCallParams params; | 73 ResourceMessageCallParams params; |
71 IPC::Message msg; | 74 IPC::Message msg; |
72 ASSERT_TRUE(sink().GetFirstResourceCallMatching( | 75 ASSERT_TRUE(sink().GetFirstResourceCallMatching( |
73 PpapiHostMsg_WebSocket_Connect::ID, ¶ms, &msg)); | 76 PpapiHostMsg_WebSocket_Connect::ID, ¶ms, &msg)); |
74 PpapiHostMsg_WebSocket_Connect::Schema::Param p; | 77 PpapiHostMsg_WebSocket_Connect::Schema::Param p; |
75 PpapiHostMsg_WebSocket_Connect::Read(&msg, &p); | 78 PpapiHostMsg_WebSocket_Connect::Read(&msg, &p); |
76 EXPECT_EQ(url, p.a); | 79 EXPECT_EQ(url, p.a); |
(...skipping 10 matching lines...) Expand all Loading... |
87 | 90 |
88 EXPECT_EQ(PP_OK, g_callback_result); | 91 EXPECT_EQ(PP_OK, g_callback_result); |
89 EXPECT_EQ(true, g_callback_called); | 92 EXPECT_EQ(true, g_callback_called); |
90 } | 93 } |
91 | 94 |
92 // Does a test for unsolicited replies. | 95 // Does a test for unsolicited replies. |
93 TEST_F(WebSocketResourceTest, UnsolicitedReplies) { | 96 TEST_F(WebSocketResourceTest, UnsolicitedReplies) { |
94 const PPB_WebSocket_1_0* websocket_iface = | 97 const PPB_WebSocket_1_0* websocket_iface = |
95 thunk::GetPPB_WebSocket_1_0_Thunk(); | 98 thunk::GetPPB_WebSocket_1_0_Thunk(); |
96 | 99 |
97 ScopedPPResource res(ScopedPPResource::PassRef(), | 100 LockingResourceReleaser res(websocket_iface->Create(pp_instance())); |
98 websocket_iface->Create(pp_instance())); | |
99 | 101 |
100 // Check if BufferedAmountReply is handled. | 102 // Check if BufferedAmountReply is handled. |
101 ResourceMessageReplyParams reply_params(res, 0); | 103 ResourceMessageReplyParams reply_params(res.get(), 0); |
102 reply_params.set_result(PP_OK); | 104 reply_params.set_result(PP_OK); |
103 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( | 105 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( |
104 PpapiPluginMsg_ResourceReply( | 106 PpapiPluginMsg_ResourceReply( |
105 reply_params, | 107 reply_params, |
106 PpapiPluginMsg_WebSocket_BufferedAmountReply(19760227u)))); | 108 PpapiPluginMsg_WebSocket_BufferedAmountReply(19760227u)))); |
107 | 109 |
108 uint64_t amount = websocket_iface->GetBufferedAmount(res); | 110 uint64_t amount = websocket_iface->GetBufferedAmount(res.get()); |
109 EXPECT_EQ(19760227u, amount); | 111 EXPECT_EQ(19760227u, amount); |
110 | 112 |
111 // Check if StateReply is handled. | 113 // Check if StateReply is handled. |
112 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( | 114 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( |
113 PpapiPluginMsg_ResourceReply( | 115 PpapiPluginMsg_ResourceReply( |
114 reply_params, | 116 reply_params, |
115 PpapiPluginMsg_WebSocket_StateReply( | 117 PpapiPluginMsg_WebSocket_StateReply( |
116 static_cast<int32_t>(PP_WEBSOCKETREADYSTATE_CLOSING))))); | 118 static_cast<int32_t>(PP_WEBSOCKETREADYSTATE_CLOSING))))); |
117 | 119 |
118 PP_WebSocketReadyState state = websocket_iface->GetReadyState(res); | 120 PP_WebSocketReadyState state = websocket_iface->GetReadyState(res.get()); |
119 EXPECT_EQ(PP_WEBSOCKETREADYSTATE_CLOSING, state); | 121 EXPECT_EQ(PP_WEBSOCKETREADYSTATE_CLOSING, state); |
120 } | 122 } |
121 | 123 |
122 TEST_F(WebSocketResourceTest, MessageError) { | 124 TEST_F(WebSocketResourceTest, MessageError) { |
123 const PPB_WebSocket_1_0* websocket_iface = | 125 const PPB_WebSocket_1_0* websocket_iface = |
124 thunk::GetPPB_WebSocket_1_0_Thunk(); | 126 thunk::GetPPB_WebSocket_1_0_Thunk(); |
125 | 127 |
126 std::string url("ws://ws.google.com"); | 128 std::string url("ws://ws.google.com"); |
127 PP_Var url_var = MakeStringVar(url); | 129 PP_Var url_var = MakeStringVar(url); |
128 | 130 |
129 ScopedPPResource res(ScopedPPResource::PassRef(), | 131 LockingResourceReleaser res(websocket_iface->Create(pp_instance())); |
130 websocket_iface->Create(pp_instance())); | |
131 | 132 |
132 // Establish the connection virtually. | 133 // Establish the connection virtually. |
133 int32_t result = | 134 int32_t result = |
134 websocket_iface->Connect(res, url_var, NULL, 0, MakeCallback()); | 135 websocket_iface->Connect(res.get(), url_var, NULL, 0, MakeCallback()); |
135 ASSERT_EQ(PP_OK_COMPLETIONPENDING, result); | 136 ASSERT_EQ(PP_OK_COMPLETIONPENDING, result); |
136 | 137 |
137 ResourceMessageCallParams params; | 138 ResourceMessageCallParams params; |
138 IPC::Message msg; | 139 IPC::Message msg; |
139 ASSERT_TRUE(sink().GetFirstResourceCallMatching( | 140 ASSERT_TRUE(sink().GetFirstResourceCallMatching( |
140 PpapiHostMsg_WebSocket_Connect::ID, ¶ms, &msg)); | 141 PpapiHostMsg_WebSocket_Connect::ID, ¶ms, &msg)); |
141 | 142 |
142 ResourceMessageReplyParams connect_reply_params(params.pp_resource(), | 143 ResourceMessageReplyParams connect_reply_params(params.pp_resource(), |
143 params.sequence()); | 144 params.sequence()); |
144 connect_reply_params.set_result(PP_OK); | 145 connect_reply_params.set_result(PP_OK); |
145 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( | 146 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( |
146 PpapiPluginMsg_ResourceReply(connect_reply_params, | 147 PpapiPluginMsg_ResourceReply(connect_reply_params, |
147 PpapiPluginMsg_WebSocket_ConnectReply(url, std::string())))); | 148 PpapiPluginMsg_WebSocket_ConnectReply(url, std::string())))); |
148 | 149 |
149 EXPECT_EQ(PP_OK, g_callback_result); | 150 EXPECT_EQ(PP_OK, g_callback_result); |
150 EXPECT_TRUE(g_callback_called); | 151 EXPECT_TRUE(g_callback_called); |
151 | 152 |
152 PP_Var message; | 153 PP_Var message; |
153 result = websocket_iface->ReceiveMessage(res, &message, MakeCallback()); | 154 result = websocket_iface->ReceiveMessage(res.get(), &message, MakeCallback()); |
154 EXPECT_FALSE(g_callback_called); | 155 EXPECT_FALSE(g_callback_called); |
155 | 156 |
156 // Synthesize a WebSocket_ErrorReply message. | 157 // Synthesize a WebSocket_ErrorReply message. |
157 ResourceMessageReplyParams error_reply_params(res, 0); | 158 ResourceMessageReplyParams error_reply_params(res.get(), 0); |
158 error_reply_params.set_result(PP_OK); | 159 error_reply_params.set_result(PP_OK); |
159 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( | 160 ASSERT_TRUE(plugin_dispatcher()->OnMessageReceived( |
160 PpapiPluginMsg_ResourceReply(error_reply_params, | 161 PpapiPluginMsg_ResourceReply(error_reply_params, |
161 PpapiPluginMsg_WebSocket_ErrorReply()))); | 162 PpapiPluginMsg_WebSocket_ErrorReply()))); |
162 | 163 |
163 EXPECT_EQ(PP_ERROR_FAILED, g_callback_result); | 164 EXPECT_EQ(PP_ERROR_FAILED, g_callback_result); |
164 EXPECT_TRUE(g_callback_called); | 165 EXPECT_TRUE(g_callback_called); |
165 } | 166 } |
166 | 167 |
167 } // namespace proxy | 168 } // namespace proxy |
168 } // namespace ppapi | 169 } // namespace ppapi |
OLD | NEW |