OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 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 | 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 "modules/fetch/FetchFormDataConsumerHandle.h" | 5 #include "modules/fetch/FetchFormDataConsumerHandle.h" |
6 | 6 |
7 #include "core/dom/DOMTypedArray.h" | 7 #include "core/dom/DOMTypedArray.h" |
8 #include "core/html/FormData.h" | 8 #include "core/html/FormData.h" |
9 #include "core/loader/MockThreadableLoader.h" | 9 #include "core/loader/MockThreadableLoader.h" |
10 #include "core/loader/ThreadableLoaderClient.h" | 10 #include "core/loader/ThreadableLoaderClient.h" |
11 #include "core/testing/DummyPageHolder.h" | 11 #include "core/testing/DummyPageHolder.h" |
12 #include "modules/fetch/DataConsumerHandleTestUtil.h" | 12 #include "modules/fetch/DataConsumerHandleTestUtil.h" |
13 #include "platform/network/ResourceResponse.h" | 13 #include "platform/network/ResourceResponse.h" |
14 #include "platform/testing/UnitTestHelpers.h" | 14 #include "platform/testing/UnitTestHelpers.h" |
15 #include "platform/weborigin/KURL.h" | 15 #include "platform/weborigin/KURL.h" |
16 #include "testing/gmock/include/gmock/gmock.h" | 16 #include "testing/gmock/include/gmock/gmock.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
| 18 #include "wtf/PassOwnPtr.h" |
18 #include "wtf/PassRefPtr.h" | 19 #include "wtf/PassRefPtr.h" |
19 #include "wtf/RefPtr.h" | 20 #include "wtf/RefPtr.h" |
20 #include "wtf/Vector.h" | 21 #include "wtf/Vector.h" |
21 #include "wtf/text/TextEncoding.h" | 22 #include "wtf/text/TextEncoding.h" |
22 #include "wtf/text/WTFString.h" | 23 #include "wtf/text/WTFString.h" |
23 #include <string.h> | 24 #include <string.h> |
24 | 25 |
25 namespace blink { | 26 namespace blink { |
26 namespace { | 27 namespace { |
27 | 28 |
28 using Result = WebDataConsumerHandle::Result; | 29 using Result = WebDataConsumerHandle::Result; |
29 const Result kOk = WebDataConsumerHandle::Ok; | 30 const Result kOk = WebDataConsumerHandle::Ok; |
30 const Result kDone = WebDataConsumerHandle::Done; | 31 const Result kDone = WebDataConsumerHandle::Done; |
31 const Result kShouldWait = WebDataConsumerHandle::ShouldWait; | 32 const Result kShouldWait = WebDataConsumerHandle::ShouldWait; |
32 const WebDataConsumerHandle::Flags kNone = WebDataConsumerHandle::FlagNone; | 33 const WebDataConsumerHandle::Flags kNone = WebDataConsumerHandle::FlagNone; |
33 using HandleReader = DataConsumerHandleTestUtil::HandleReader; | 34 using HandleReader = DataConsumerHandleTestUtil::HandleReader; |
34 using HandleTwoPhaseReader = DataConsumerHandleTestUtil::HandleTwoPhaseReader; | 35 using HandleTwoPhaseReader = DataConsumerHandleTestUtil::HandleTwoPhaseReader; |
35 using HandleReadResult = DataConsumerHandleTestUtil::HandleReadResult; | 36 using HandleReadResult = DataConsumerHandleTestUtil::HandleReadResult; |
36 template <typename T> | 37 template <typename T> |
37 using HandleReaderRunner = DataConsumerHandleTestUtil::HandleReaderRunner<T>; | 38 using HandleReaderRunner = DataConsumerHandleTestUtil::HandleReaderRunner<T>; |
38 using ReplayingHandle = DataConsumerHandleTestUtil::ReplayingHandle; | 39 using ReplayingHandle = DataConsumerHandleTestUtil::ReplayingHandle; |
39 using Command = DataConsumerHandleTestUtil::Command; | 40 using Command = DataConsumerHandleTestUtil::Command; |
40 | 41 |
| 42 using ::testing::_; |
| 43 using ::testing::InvokeWithoutArgs; |
| 44 |
41 String toString(const Vector<char>& data) | 45 String toString(const Vector<char>& data) |
42 { | 46 { |
43 return String(data.data(), data.size()); | 47 return String(data.data(), data.size()); |
44 } | 48 } |
45 | 49 |
46 class LoaderFactory : public FetchBlobDataConsumerHandle::LoaderFactory { | 50 class LoaderFactory : public FetchBlobDataConsumerHandle::LoaderFactory { |
47 public: | 51 public: |
48 explicit LoaderFactory(PassOwnPtr<WebDataConsumerHandle> handle) : m_handle(
handle) {} | 52 explicit LoaderFactory(PassOwnPtr<WebDataConsumerHandle> handle) |
49 PassRefPtr<ThreadableLoader> create(ExecutionContext&, ThreadableLoaderClien
t* client, const ResourceRequest&, const ThreadableLoaderOptions&, const Resourc
eLoaderOptions&) override | 53 : m_client(nullptr) |
| 54 , m_handle(handle) {} |
| 55 PassRefPtr<ThreadableLoader> create(ExecutionContext&, ThreadableLoaderClien
t* client, const ThreadableLoaderOptions&, const ResourceLoaderOptions&) overrid
e |
50 { | 56 { |
| 57 m_client = client; |
| 58 |
51 RefPtr<MockThreadableLoader> loader = MockThreadableLoader::create(); | 59 RefPtr<MockThreadableLoader> loader = MockThreadableLoader::create(); |
| 60 EXPECT_CALL(*loader, start(_)).WillOnce(InvokeWithoutArgs(this, &LoaderF
actory::handleDidReceiveResponse)); |
52 EXPECT_CALL(*loader, cancel()).Times(1); | 61 EXPECT_CALL(*loader, cancel()).Times(1); |
53 client->didReceiveResponse(0, ResourceResponse(), m_handle.release()); | |
54 return loader.release(); | 62 return loader.release(); |
55 } | 63 } |
56 | 64 |
57 private: | 65 private: |
| 66 void handleDidReceiveResponse() |
| 67 { |
| 68 m_client->didReceiveResponse(0, ResourceResponse(), m_handle.release()); |
| 69 } |
| 70 |
| 71 ThreadableLoaderClient* m_client; |
58 OwnPtr<WebDataConsumerHandle> m_handle; | 72 OwnPtr<WebDataConsumerHandle> m_handle; |
59 }; | 73 }; |
60 | 74 |
61 class FetchFormDataConsumerHandleTest : public ::testing::Test { | 75 class FetchFormDataConsumerHandleTest : public ::testing::Test { |
62 public: | 76 public: |
63 FetchFormDataConsumerHandleTest() : m_page(DummyPageHolder::create(IntSize(1
, 1))) {} | 77 FetchFormDataConsumerHandleTest() : m_page(DummyPageHolder::create(IntSize(1
, 1))) {} |
64 | 78 |
65 protected: | 79 protected: |
66 Document* document() { return &m_page->document(); } | 80 Document* document() { return &m_page->document(); } |
67 | 81 |
(...skipping 355 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
423 EXPECT_EQ(kShouldWait, reader->beginRead(&buffer, kNone, &available)); | 437 EXPECT_EQ(kShouldWait, reader->beginRead(&buffer, kNone, &available)); |
424 testing::runPendingTasks(); | 438 testing::runPendingTasks(); |
425 EXPECT_EQ(kOk, reader->beginRead(&buffer, kNone, &available)); | 439 EXPECT_EQ(kOk, reader->beginRead(&buffer, kNone, &available)); |
426 EXPECT_FALSE(reader->drainAsFormData()); | 440 EXPECT_FALSE(reader->drainAsFormData()); |
427 reader->endRead(0); | 441 reader->endRead(0); |
428 EXPECT_FALSE(reader->drainAsFormData()); | 442 EXPECT_FALSE(reader->drainAsFormData()); |
429 } | 443 } |
430 | 444 |
431 } // namespace | 445 } // namespace |
432 } // namespace blink | 446 } // namespace blink |
OLD | NEW |