OLD | NEW |
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 "content/child/url_response_body_consumer.h" | 5 #include "content/child/url_response_body_consumer.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/callback_forward.h" | 8 #include "base/callback_forward.h" |
9 #include "base/macros.h" | 9 #include "base/macros.h" |
10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
(...skipping 142 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
153 | 153 |
154 TEST_F(URLResponseBodyConsumerTest, ReceiveData) { | 154 TEST_F(URLResponseBodyConsumerTest, ReceiveData) { |
155 TestRequestPeer::Context context; | 155 TestRequestPeer::Context context; |
156 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); | 156 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); |
157 int request_id = SetUpRequestPeer(std::move(request), &context); | 157 int request_id = SetUpRequestPeer(std::move(request), &context); |
158 mojo::DataPipe data_pipe(CreateDataPipeOptions()); | 158 mojo::DataPipe data_pipe(CreateDataPipeOptions()); |
159 | 159 |
160 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( | 160 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( |
161 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), | 161 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), |
162 message_loop_.task_runner())); | 162 message_loop_.task_runner())); |
163 consumer->Start(); | |
164 | 163 |
165 mojo::ScopedDataPipeProducerHandle writer = | 164 mojo::ScopedDataPipeProducerHandle writer = |
166 std::move(data_pipe.producer_handle); | 165 std::move(data_pipe.producer_handle); |
167 std::string buffer = "hello"; | 166 std::string buffer = "hello"; |
168 uint32_t size = buffer.size(); | 167 uint32_t size = buffer.size(); |
169 MojoResult result = | 168 MojoResult result = |
170 mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone); | 169 mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone); |
171 ASSERT_EQ(MOJO_RESULT_OK, result); | 170 ASSERT_EQ(MOJO_RESULT_OK, result); |
172 ASSERT_EQ(buffer.size(), size); | 171 ASSERT_EQ(buffer.size(), size); |
173 | 172 |
174 Run(&context); | 173 Run(&context); |
175 | 174 |
176 EXPECT_FALSE(context.complete); | 175 EXPECT_FALSE(context.complete); |
177 EXPECT_EQ("hello", context.data); | 176 EXPECT_EQ("hello", context.data); |
178 } | 177 } |
179 | 178 |
180 TEST_F(URLResponseBodyConsumerTest, OnCompleteThenClose) { | 179 TEST_F(URLResponseBodyConsumerTest, OnCompleteThenClose) { |
181 TestRequestPeer::Context context; | 180 TestRequestPeer::Context context; |
182 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); | 181 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); |
183 int request_id = SetUpRequestPeer(std::move(request), &context); | 182 int request_id = SetUpRequestPeer(std::move(request), &context); |
184 mojo::DataPipe data_pipe(CreateDataPipeOptions()); | 183 mojo::DataPipe data_pipe(CreateDataPipeOptions()); |
185 | 184 |
186 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( | 185 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( |
187 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), | 186 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), |
188 message_loop_.task_runner())); | 187 message_loop_.task_runner())); |
189 consumer->Start(); | |
190 | 188 |
191 consumer->OnComplete(ResourceRequestCompletionStatus()); | 189 consumer->OnComplete(ResourceRequestCompletionStatus()); |
192 mojo::ScopedDataPipeProducerHandle writer = | 190 mojo::ScopedDataPipeProducerHandle writer = |
193 std::move(data_pipe.producer_handle); | 191 std::move(data_pipe.producer_handle); |
194 std::string buffer = "hello"; | 192 std::string buffer = "hello"; |
195 uint32_t size = buffer.size(); | 193 uint32_t size = buffer.size(); |
196 MojoResult result = | 194 MojoResult result = |
197 mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone); | 195 mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone); |
198 ASSERT_EQ(MOJO_RESULT_OK, result); | 196 ASSERT_EQ(MOJO_RESULT_OK, result); |
199 ASSERT_EQ(buffer.size(), size); | 197 ASSERT_EQ(buffer.size(), size); |
(...skipping 15 matching lines...) Expand all Loading... |
215 TEST_F(URLResponseBodyConsumerTest, OnCompleteThenCloseWithAsyncRelease) { | 213 TEST_F(URLResponseBodyConsumerTest, OnCompleteThenCloseWithAsyncRelease) { |
216 TestRequestPeer::Context context; | 214 TestRequestPeer::Context context; |
217 context.release_data_asynchronously = true; | 215 context.release_data_asynchronously = true; |
218 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); | 216 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); |
219 int request_id = SetUpRequestPeer(std::move(request), &context); | 217 int request_id = SetUpRequestPeer(std::move(request), &context); |
220 mojo::DataPipe data_pipe(CreateDataPipeOptions()); | 218 mojo::DataPipe data_pipe(CreateDataPipeOptions()); |
221 | 219 |
222 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( | 220 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( |
223 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), | 221 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), |
224 message_loop_.task_runner())); | 222 message_loop_.task_runner())); |
225 consumer->Start(); | |
226 | 223 |
227 consumer->OnComplete(ResourceRequestCompletionStatus()); | 224 consumer->OnComplete(ResourceRequestCompletionStatus()); |
228 mojo::ScopedDataPipeProducerHandle writer = | 225 mojo::ScopedDataPipeProducerHandle writer = |
229 std::move(data_pipe.producer_handle); | 226 std::move(data_pipe.producer_handle); |
230 std::string buffer = "hello"; | 227 std::string buffer = "hello"; |
231 uint32_t size = buffer.size(); | 228 uint32_t size = buffer.size(); |
232 MojoResult result = | 229 MojoResult result = |
233 mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone); | 230 mojo::WriteDataRaw(writer.get(), buffer.c_str(), &size, kNone); |
234 ASSERT_EQ(MOJO_RESULT_OK, result); | 231 ASSERT_EQ(MOJO_RESULT_OK, result); |
235 ASSERT_EQ(buffer.size(), size); | 232 ASSERT_EQ(buffer.size(), size); |
(...skipping 12 matching lines...) Expand all Loading... |
248 | 245 |
249 TEST_F(URLResponseBodyConsumerTest, CloseThenOnComplete) { | 246 TEST_F(URLResponseBodyConsumerTest, CloseThenOnComplete) { |
250 TestRequestPeer::Context context; | 247 TestRequestPeer::Context context; |
251 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); | 248 std::unique_ptr<ResourceRequest> request(CreateResourceRequest()); |
252 int request_id = SetUpRequestPeer(std::move(request), &context); | 249 int request_id = SetUpRequestPeer(std::move(request), &context); |
253 mojo::DataPipe data_pipe(CreateDataPipeOptions()); | 250 mojo::DataPipe data_pipe(CreateDataPipeOptions()); |
254 | 251 |
255 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( | 252 scoped_refptr<URLResponseBodyConsumer> consumer(new URLResponseBodyConsumer( |
256 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), | 253 request_id, dispatcher_.get(), std::move(data_pipe.consumer_handle), |
257 message_loop_.task_runner())); | 254 message_loop_.task_runner())); |
258 consumer->Start(); | |
259 | 255 |
260 ResourceRequestCompletionStatus status; | 256 ResourceRequestCompletionStatus status; |
261 status.error_code = net::ERR_FAILED; | 257 status.error_code = net::ERR_FAILED; |
262 data_pipe.producer_handle.reset(); | 258 data_pipe.producer_handle.reset(); |
263 consumer->OnComplete(status); | 259 consumer->OnComplete(status); |
264 | 260 |
265 Run(&context); | 261 Run(&context); |
266 | 262 |
267 EXPECT_TRUE(context.complete); | 263 EXPECT_TRUE(context.complete); |
268 EXPECT_EQ(net::ERR_FAILED, context.error_code); | 264 EXPECT_EQ(net::ERR_FAILED, context.error_code); |
269 EXPECT_EQ("", context.data); | 265 EXPECT_EQ("", context.data); |
270 } | 266 } |
271 | 267 |
272 } // namespace | 268 } // namespace |
273 | 269 |
274 } // namespace content | 270 } // namespace content |
OLD | NEW |