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

Side by Side Diff: third_party/WebKit/Source/modules/fetch/FetchBlobDataConsumerHandleTest.cpp

Issue 2147633002: Remove nonstandard 'endings' option for Blob/File constructor (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebased Created 3 years, 11 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
(Empty)
1 // Copyright 2015 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 "modules/fetch/FetchBlobDataConsumerHandle.h"
6
7 #include "core/dom/ExecutionContext.h"
8 #include "core/fetch/ResourceLoaderOptions.h"
9 #include "core/loader/MockThreadableLoader.h"
10 #include "core/loader/ThreadableLoaderClient.h"
11 #include "core/testing/DummyPageHolder.h"
12 #include "modules/fetch/DataConsumerHandleTestUtil.h"
13 #include "platform/blob/BlobData.h"
14 #include "platform/blob/BlobURL.h"
15 #include "platform/network/ResourceError.h"
16 #include "platform/network/ResourceRequest.h"
17 #include "platform/network/ResourceResponse.h"
18 #include "platform/testing/UnitTestHelpers.h"
19 #include "testing/gmock/include/gmock/gmock.h"
20 #include "testing/gtest/include/gtest/gtest.h"
21 #include "wtf/PassRefPtr.h"
22 #include "wtf/PtrUtil.h"
23 #include "wtf/RefPtr.h"
24 #include <memory>
25 #include <string.h>
26
27 namespace blink {
28 namespace {
29
30 using Result = WebDataConsumerHandle::Result;
31 const Result kShouldWait = WebDataConsumerHandle::ShouldWait;
32 const Result kUnexpectedError = WebDataConsumerHandle::UnexpectedError;
33 const Result kDone = WebDataConsumerHandle::Done;
34 using Flags = WebDataConsumerHandle::Flags;
35 const Flags kNone = WebDataConsumerHandle::FlagNone;
36 using Thread = DataConsumerHandleTestUtil::Thread;
37 using HandleReader = DataConsumerHandleTestUtil::HandleReader;
38 using HandleTwoPhaseReader = DataConsumerHandleTestUtil::HandleTwoPhaseReader;
39 using HandleReadResult = DataConsumerHandleTestUtil::HandleReadResult;
40 using ReplayingHandle = DataConsumerHandleTestUtil::ReplayingHandle;
41 using Command = DataConsumerHandleTestUtil::Command;
42 template <typename T>
43 using HandleReaderRunner = DataConsumerHandleTestUtil::HandleReaderRunner<T>;
44
45 using ::testing::_;
46 using ::testing::DoAll;
47 using ::testing::InSequence;
48 using ::testing::Ref;
49 using ::testing::Return;
50 using ::testing::SaveArg;
51 using ::testing::StrictMock;
52 using Checkpoint = StrictMock<::testing::MockFunction<void(int)>>;
53
54 class MockLoaderFactory : public FetchBlobDataConsumerHandle::LoaderFactory {
55 public:
56 MOCK_METHOD4(create,
57 ThreadableLoader*(ExecutionContext&,
58 ThreadableLoaderClient*,
59 const ThreadableLoaderOptions&,
60 const ResourceLoaderOptions&));
61 };
62
63 PassRefPtr<BlobDataHandle> createBlobDataHandle(const char* s) {
64 std::unique_ptr<BlobData> data = BlobData::create();
65 data->appendText(s);
66 auto size = data->length();
67 return BlobDataHandle::create(std::move(data), size);
68 }
69
70 String toString(const Vector<char>& data) {
71 return String(data.data(), data.size());
72 }
73
74 class FetchBlobDataConsumerHandleTest : public ::testing::Test {
75 public:
76 FetchBlobDataConsumerHandleTest()
77 : m_dummyPageHolder(DummyPageHolder::create(IntSize(1, 1))) {}
78 ~FetchBlobDataConsumerHandleTest() override {
79 m_dummyPageHolder = nullptr;
80 // We need this to collect garbage-collected mocks.
81 ThreadState::current()->collectAllGarbage();
82 }
83
84 Document& document() { return m_dummyPageHolder->document(); }
85
86 private:
87 std::unique_ptr<DummyPageHolder> m_dummyPageHolder;
88 };
89
90 TEST_F(FetchBlobDataConsumerHandleTest, CreateLoader) {
91 auto factory = new StrictMock<MockLoaderFactory>;
92 Checkpoint checkpoint;
93
94 ResourceRequest request;
95 ThreadableLoaderOptions options;
96 ResourceLoaderOptions resourceLoaderOptions;
97
98 Persistent<MockThreadableLoader> loader = MockThreadableLoader::create();
99
100 InSequence s;
101 EXPECT_CALL(checkpoint, Call(1));
102 EXPECT_CALL(*factory, create(Ref(document()), _, _, _))
103 .WillOnce(DoAll(SaveArg<2>(&options), SaveArg<3>(&resourceLoaderOptions),
104 Return(loader.get())));
105 EXPECT_CALL(*loader, start(_)).WillOnce(SaveArg<0>(&request));
106 EXPECT_CALL(checkpoint, Call(2));
107 EXPECT_CALL(*loader, cancel());
108
109 RefPtr<BlobDataHandle> blobDataHandle =
110 createBlobDataHandle("Once upon a time");
111 std::unique_ptr<WebDataConsumerHandle> handle =
112 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
113 testing::runPendingTasks();
114
115 char buffer[1];
116 size_t size = 0;
117 ASSERT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(
118 buffer, sizeof(buffer), kNone, &size));
119 checkpoint.Call(1);
120 testing::runPendingTasks();
121 checkpoint.Call(2);
122
123 EXPECT_TRUE(request.url().getString().startsWith("blob:"));
124 EXPECT_TRUE(request.useStreamOnResponse());
125
126 EXPECT_EQ(ConsiderPreflight, options.preflightPolicy);
127 EXPECT_EQ(DenyCrossOriginRequests, options.crossOriginRequestPolicy);
128 EXPECT_EQ(DoNotEnforceContentSecurityPolicy,
129 options.contentSecurityPolicyEnforcement);
130
131 EXPECT_EQ(DoNotBufferData, resourceLoaderOptions.dataBufferingPolicy);
132 EXPECT_EQ(DoNotAllowStoredCredentials,
133 resourceLoaderOptions.allowCredentials);
134 EXPECT_EQ(ClientDidNotRequestCredentials,
135 resourceLoaderOptions.credentialsRequested);
136 EXPECT_EQ(CheckContentSecurityPolicy,
137 resourceLoaderOptions.contentSecurityPolicyOption);
138 EXPECT_EQ(DocumentContext, resourceLoaderOptions.requestInitiatorContext);
139 EXPECT_EQ(RequestAsynchronously, resourceLoaderOptions.synchronousPolicy);
140 EXPECT_EQ(NotCORSEnabled, resourceLoaderOptions.corsEnabled);
141 }
142
143 TEST_F(FetchBlobDataConsumerHandleTest, ZeroByteReadDoesNotCreateLoader) {
144 auto factory = new StrictMock<MockLoaderFactory>;
145 Checkpoint checkpoint;
146
147 InSequence s;
148 EXPECT_CALL(checkpoint, Call(1));
149 EXPECT_CALL(checkpoint, Call(2));
150
151 RefPtr<BlobDataHandle> blobDataHandle =
152 createBlobDataHandle("Once upon a time");
153 std::unique_ptr<WebDataConsumerHandle> handle =
154 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
155 testing::runPendingTasks();
156
157 size_t size = 0;
158 ASSERT_EQ(kShouldWait,
159 handle->obtainReader(nullptr)->read(nullptr, 0, kNone, &size));
160 checkpoint.Call(1);
161 testing::runPendingTasks();
162 checkpoint.Call(2);
163 }
164
165 TEST_F(FetchBlobDataConsumerHandleTest, CancelLoaderWhenStopped) {
166 auto factory = new StrictMock<MockLoaderFactory>;
167 Checkpoint checkpoint;
168
169 Persistent<MockThreadableLoader> loader = MockThreadableLoader::create();
170
171 InSequence s;
172 EXPECT_CALL(checkpoint, Call(1));
173 EXPECT_CALL(*factory, create(Ref(document()), _, _, _))
174 .WillOnce(Return(loader.get()));
175 EXPECT_CALL(*loader, start(_));
176 EXPECT_CALL(checkpoint, Call(2));
177 EXPECT_CALL(*loader, cancel());
178 EXPECT_CALL(checkpoint, Call(3));
179
180 RefPtr<BlobDataHandle> blobDataHandle =
181 createBlobDataHandle("Once upon a time");
182 std::unique_ptr<WebDataConsumerHandle> handle =
183 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
184 testing::runPendingTasks();
185
186 char buffer[1];
187 size_t size = 0;
188 ASSERT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(
189 buffer, sizeof(buffer), kNone, &size));
190 checkpoint.Call(1);
191 testing::runPendingTasks();
192 checkpoint.Call(2);
193 document().stopActiveDOMObjects();
194 checkpoint.Call(3);
195 }
196
197 TEST_F(FetchBlobDataConsumerHandleTest, CancelLoaderWhenDestinationDetached) {
198 auto factory = new StrictMock<MockLoaderFactory>;
199 Checkpoint checkpoint;
200
201 Persistent<MockThreadableLoader> loader = MockThreadableLoader::create();
202
203 InSequence s;
204 EXPECT_CALL(checkpoint, Call(1));
205 EXPECT_CALL(*factory, create(Ref(document()), _, _, _))
206 .WillOnce(Return(loader.get()));
207 EXPECT_CALL(*loader, start(_));
208 EXPECT_CALL(checkpoint, Call(2));
209 EXPECT_CALL(checkpoint, Call(3));
210 EXPECT_CALL(*loader, cancel());
211 EXPECT_CALL(checkpoint, Call(4));
212
213 RefPtr<BlobDataHandle> blobDataHandle =
214 createBlobDataHandle("Once upon a time");
215 std::unique_ptr<WebDataConsumerHandle> handle =
216 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
217 std::unique_ptr<WebDataConsumerHandle::Reader> reader =
218 handle->obtainReader(nullptr);
219 testing::runPendingTasks();
220
221 char buffer[1];
222 size_t size = 0;
223 ASSERT_EQ(kShouldWait, reader->read(buffer, sizeof(buffer), kNone, &size));
224 checkpoint.Call(1);
225 testing::runPendingTasks();
226 checkpoint.Call(2);
227 handle = nullptr;
228 reader = nullptr;
229 checkpoint.Call(3);
230 ThreadState::current()->collectAllGarbage();
231 checkpoint.Call(4);
232 }
233
234 TEST_F(FetchBlobDataConsumerHandleTest, ReadTest) {
235 auto factory = new StrictMock<MockLoaderFactory>;
236 Checkpoint checkpoint;
237
238 Persistent<MockThreadableLoader> loader = MockThreadableLoader::create();
239 ThreadableLoaderClient* client = nullptr;
240
241 InSequence s;
242 EXPECT_CALL(checkpoint, Call(1));
243 EXPECT_CALL(*factory, create(Ref(document()), _, _, _))
244 .WillOnce(DoAll(SaveArg<1>(&client), Return(loader.get())));
245 EXPECT_CALL(*loader, start(_));
246 EXPECT_CALL(checkpoint, Call(2));
247 EXPECT_CALL(*loader, cancel());
248
249 RefPtr<BlobDataHandle> blobDataHandle =
250 createBlobDataHandle("Once upon a time");
251 std::unique_ptr<WebDataConsumerHandle> handle =
252 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
253
254 std::unique_ptr<ReplayingHandle> src = ReplayingHandle::create();
255 src->add(Command(Command::Wait));
256 src->add(Command(Command::Data, "hello, "));
257 src->add(Command(Command::Data, "world"));
258 src->add(Command(Command::Wait));
259 src->add(Command(Command::Done));
260
261 char buffer[1];
262 size_t size = 0;
263 ASSERT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(
264 buffer, sizeof(buffer), kNone, &size));
265 checkpoint.Call(1);
266 testing::runPendingTasks();
267 checkpoint.Call(2);
268 client->didReceiveResponse(0, ResourceResponse(), std::move(src));
269 HandleReaderRunner<HandleReader> runner(std::move(handle));
270 std::unique_ptr<HandleReadResult> r = runner.wait();
271 EXPECT_EQ(kDone, r->result());
272 EXPECT_EQ("hello, world", toString(r->data()));
273 }
274
275 TEST_F(FetchBlobDataConsumerHandleTest, TwoPhaseReadTest) {
276 auto factory = new StrictMock<MockLoaderFactory>;
277 Checkpoint checkpoint;
278
279 Persistent<MockThreadableLoader> loader = MockThreadableLoader::create();
280 ThreadableLoaderClient* client = nullptr;
281
282 InSequence s;
283 EXPECT_CALL(checkpoint, Call(1));
284 EXPECT_CALL(*factory, create(Ref(document()), _, _, _))
285 .WillOnce(DoAll(SaveArg<1>(&client), Return(loader.get())));
286 EXPECT_CALL(*loader, start(_));
287 EXPECT_CALL(checkpoint, Call(2));
288 EXPECT_CALL(*loader, cancel());
289
290 RefPtr<BlobDataHandle> blobDataHandle =
291 createBlobDataHandle("Once upon a time");
292 std::unique_ptr<WebDataConsumerHandle> handle =
293 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
294
295 std::unique_ptr<ReplayingHandle> src = ReplayingHandle::create();
296 src->add(Command(Command::Wait));
297 src->add(Command(Command::Data, "hello, "));
298 src->add(Command(Command::Data, "world"));
299 src->add(Command(Command::Wait));
300 src->add(Command(Command::Done));
301
302 char buffer[1];
303 size_t size = 0;
304 ASSERT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(
305 buffer, sizeof(buffer), kNone, &size));
306 checkpoint.Call(1);
307 testing::runPendingTasks();
308 checkpoint.Call(2);
309 client->didReceiveResponse(0, ResourceResponse(), std::move(src));
310 HandleReaderRunner<HandleTwoPhaseReader> runner(std::move(handle));
311 std::unique_ptr<HandleReadResult> r = runner.wait();
312 EXPECT_EQ(kDone, r->result());
313 EXPECT_EQ("hello, world", toString(r->data()));
314 }
315
316 TEST_F(FetchBlobDataConsumerHandleTest, LoadErrorTest) {
317 auto factory = new StrictMock<MockLoaderFactory>;
318 Checkpoint checkpoint;
319
320 Persistent<MockThreadableLoader> loader = MockThreadableLoader::create();
321 ThreadableLoaderClient* client = nullptr;
322
323 InSequence s;
324 EXPECT_CALL(checkpoint, Call(1));
325 EXPECT_CALL(*factory, create(Ref(document()), _, _, _))
326 .WillOnce(DoAll(SaveArg<1>(&client), Return(loader.get())));
327 EXPECT_CALL(*loader, start(_));
328 EXPECT_CALL(checkpoint, Call(2));
329
330 RefPtr<BlobDataHandle> blobDataHandle =
331 createBlobDataHandle("Once upon a time");
332 std::unique_ptr<WebDataConsumerHandle> handle =
333 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
334
335 char buffer[1];
336 size_t size = 0;
337 ASSERT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(
338 buffer, sizeof(buffer), kNone, &size));
339 checkpoint.Call(1);
340 testing::runPendingTasks();
341 checkpoint.Call(2);
342 client->didFail(ResourceError());
343 HandleReaderRunner<HandleReader> runner(std::move(handle));
344 std::unique_ptr<HandleReadResult> r = runner.wait();
345 EXPECT_EQ(kUnexpectedError, r->result());
346 }
347
348 TEST_F(FetchBlobDataConsumerHandleTest, BodyLoadErrorTest) {
349 auto factory = new StrictMock<MockLoaderFactory>;
350 Checkpoint checkpoint;
351
352 Persistent<MockThreadableLoader> loader = MockThreadableLoader::create();
353 ThreadableLoaderClient* client = nullptr;
354
355 InSequence s;
356 EXPECT_CALL(checkpoint, Call(1));
357 EXPECT_CALL(*factory, create(Ref(document()), _, _, _))
358 .WillOnce(DoAll(SaveArg<1>(&client), Return(loader.get())));
359 EXPECT_CALL(*loader, start(_));
360 EXPECT_CALL(checkpoint, Call(2));
361 EXPECT_CALL(*loader, cancel());
362
363 RefPtr<BlobDataHandle> blobDataHandle =
364 createBlobDataHandle("Once upon a time");
365 std::unique_ptr<WebDataConsumerHandle> handle =
366 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
367
368 std::unique_ptr<ReplayingHandle> src = ReplayingHandle::create();
369 src->add(Command(Command::Wait));
370 src->add(Command(Command::Data, "hello, "));
371 src->add(Command(Command::Error));
372
373 char buffer[1];
374 size_t size = 0;
375 ASSERT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(
376 buffer, sizeof(buffer), kNone, &size));
377 checkpoint.Call(1);
378 testing::runPendingTasks();
379 checkpoint.Call(2);
380 client->didReceiveResponse(0, ResourceResponse(), std::move(src));
381 HandleReaderRunner<HandleReader> runner(std::move(handle));
382 std::unique_ptr<HandleReadResult> r = runner.wait();
383 EXPECT_EQ(kUnexpectedError, r->result());
384 }
385
386 TEST_F(FetchBlobDataConsumerHandleTest, DrainAsBlobDataHandle) {
387 auto factory = new StrictMock<MockLoaderFactory>;
388
389 RefPtr<BlobDataHandle> blobDataHandle =
390 createBlobDataHandle("Once upon a time");
391 std::unique_ptr<FetchDataConsumerHandle> handle =
392 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
393
394 size_t size = 0;
395 EXPECT_EQ(blobDataHandle,
396 handle->obtainFetchDataReader(nullptr)->drainAsBlobDataHandle());
397 EXPECT_FALSE(handle->obtainFetchDataReader(nullptr)->drainAsFormData());
398
399 EXPECT_EQ(kDone,
400 handle->obtainReader(nullptr)->read(nullptr, 0, kNone, &size));
401 }
402
403 TEST_F(FetchBlobDataConsumerHandleTest, DrainAsFormData) {
404 auto factory = new StrictMock<MockLoaderFactory>;
405
406 RefPtr<BlobDataHandle> blobDataHandle =
407 createBlobDataHandle("Once upon a time");
408 std::unique_ptr<FetchDataConsumerHandle> handle =
409 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
410
411 RefPtr<EncodedFormData> formData =
412 handle->obtainFetchDataReader(nullptr)->drainAsFormData();
413 ASSERT_TRUE(formData);
414 EXPECT_TRUE(formData->isSafeToSendToAnotherThread());
415 ASSERT_EQ(1u, formData->elements().size());
416 EXPECT_EQ(FormDataElement::encodedBlob, formData->elements()[0].m_type);
417 EXPECT_EQ(blobDataHandle->uuid(), formData->elements()[0].m_blobUUID);
418 EXPECT_EQ(blobDataHandle, formData->elements()[0].m_optionalBlobDataHandle);
419
420 EXPECT_FALSE(handle->obtainFetchDataReader(nullptr)->drainAsBlobDataHandle());
421 size_t size;
422 EXPECT_EQ(kDone, handle->obtainFetchDataReader(nullptr)->read(nullptr, 0,
423 kNone, &size));
424 }
425
426 TEST_F(FetchBlobDataConsumerHandleTest, ZeroByteReadDoesNotAffectDraining) {
427 auto factory = new StrictMock<MockLoaderFactory>;
428
429 RefPtr<BlobDataHandle> blobDataHandle =
430 createBlobDataHandle("Once upon a time");
431 std::unique_ptr<FetchDataConsumerHandle> handle =
432 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
433 std::unique_ptr<FetchDataConsumerHandle::Reader> reader =
434 handle->obtainFetchDataReader(nullptr);
435
436 size_t readSize;
437 EXPECT_EQ(kShouldWait, reader->read(nullptr, 0, kNone, &readSize));
438 EXPECT_EQ(blobDataHandle, reader->drainAsBlobDataHandle());
439 }
440
441 TEST_F(FetchBlobDataConsumerHandleTest, OneByteReadAffectsDraining) {
442 auto factory = new StrictMock<MockLoaderFactory>;
443
444 RefPtr<BlobDataHandle> blobDataHandle =
445 createBlobDataHandle("Once upon a time");
446 std::unique_ptr<FetchDataConsumerHandle> handle =
447 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
448 std::unique_ptr<FetchDataConsumerHandle::Reader> reader =
449 handle->obtainFetchDataReader(nullptr);
450
451 size_t readSize;
452 char c;
453 EXPECT_EQ(kShouldWait, reader->read(&c, 1, kNone, &readSize));
454 EXPECT_FALSE(reader->drainAsFormData());
455 }
456
457 TEST_F(FetchBlobDataConsumerHandleTest, BeginReadAffectsDraining) {
458 auto factory = new StrictMock<MockLoaderFactory>;
459
460 RefPtr<BlobDataHandle> blobDataHandle =
461 createBlobDataHandle("Once upon a time");
462 std::unique_ptr<FetchDataConsumerHandle> handle =
463 FetchBlobDataConsumerHandle::create(&document(), blobDataHandle, factory);
464 std::unique_ptr<FetchDataConsumerHandle::Reader> reader =
465 handle->obtainFetchDataReader(nullptr);
466
467 const void* buffer;
468 size_t available;
469 EXPECT_EQ(kShouldWait, reader->beginRead(&buffer, kNone, &available));
470 EXPECT_FALSE(reader->drainAsBlobDataHandle());
471 }
472
473 } // namespace
474 } // namespace blink
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698