Chromium Code Reviews| Index: Source/modules/fetch/FetchFormDataConsumerHandleTest.cpp |
| diff --git a/Source/modules/fetch/FetchFormDataConsumerHandleTest.cpp b/Source/modules/fetch/FetchFormDataConsumerHandleTest.cpp |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..91d0b2af88f62cd7fb6976cb3c9f56271153cf3c |
| --- /dev/null |
| +++ b/Source/modules/fetch/FetchFormDataConsumerHandleTest.cpp |
| @@ -0,0 +1,381 @@ |
| +// Copyright 2015 The Chromium Authors. All rights reserved. |
| +// Use of this source code is governed by a BSD-style license that can be |
| +// found in the LICENSE file. |
| + |
| +#include "config.h" |
| +#include "modules/fetch/FetchFormDataConsumerHandle.h" |
| + |
| +#include "core/dom/DOMTypedArray.h" |
| +#include "core/html/DOMFormData.h" |
| +#include "core/html/FormDataList.h" |
| +#include "core/loader/ThreadableLoader.h" |
| +#include "core/loader/ThreadableLoaderClient.h" |
| +#include "core/testing/DummyPageHolder.h" |
| +#include "modules/fetch/DataConsumerHandleTestUtil.h" |
| +#include "platform/network/ResourceResponse.h" |
| +#include "platform/testing/UnitTestHelpers.h" |
| +#include "platform/weborigin/KURL.h" |
| +#include "wtf/PassRefPtr.h" |
| +#include "wtf/RefPtr.h" |
| +#include "wtf/Vector.h" |
| +#include "wtf/text/TextEncoding.h" |
| +#include "wtf/text/WTFString.h" |
| + |
| +#include <gtest/gtest.h> |
| +#include <string.h> |
| + |
| +namespace blink { |
| +namespace { |
| + |
| +using Result = WebDataConsumerHandle::Result; |
| +const Result kOk = WebDataConsumerHandle::Ok; |
| +const Result kDone = WebDataConsumerHandle::Done; |
| +const Result kShouldWait = WebDataConsumerHandle::ShouldWait; |
| +const WebDataConsumerHandle::Flags kNone = WebDataConsumerHandle::FlagNone; |
| +using HandleReader = DataConsumerHandleTestUtil::HandleReader; |
| +using HandleTwoPhaseReader = DataConsumerHandleTestUtil::HandleTwoPhaseReader; |
| +using HandleReadResult = DataConsumerHandleTestUtil::HandleReadResult; |
| +template <typename T> |
| +using HandleReaderRunner = DataConsumerHandleTestUtil::HandleReaderRunner<T>; |
| +using ReplayingHandle = DataConsumerHandleTestUtil::ReplayingHandle; |
| +using Command = DataConsumerHandleTestUtil::Command; |
| + |
| +String toString(const Vector<char>& data) |
| +{ |
| + return String(data.data(), data.size()); |
| +} |
| + |
| +class NoopLoader final : public ThreadableLoader { |
| +public: |
| + static PassRefPtr<ThreadableLoader> create() { return adoptRef(new NoopLoader); } |
| + void overrideTimeout(unsigned long) override {} |
| + void cancel() override {} |
| +}; |
| + |
| +class LoaderFactory : public FetchBlobDataConsumerHandle::LoaderFactory { |
| +public: |
| + explicit LoaderFactory(PassOwnPtr<WebDataConsumerHandle> handle) : m_handle(handle) {} |
| + PassRefPtr<ThreadableLoader> create(ExecutionContext&, ThreadableLoaderClient* client, const ResourceRequest&, const ThreadableLoaderOptions&, const ResourceLoaderOptions&) override |
| + { |
| + RefPtr<ThreadableLoader> loader = NoopLoader::create(); |
| + client->didReceiveResponse(0, ResourceResponse(), m_handle.release()); |
| + return loader.release(); |
| + } |
| + |
| +private: |
| + OwnPtr<WebDataConsumerHandle> m_handle; |
| +}; |
| + |
| +class FetchFormDataConsumerHandleTest : public ::testing::Test { |
| +public: |
| + FetchFormDataConsumerHandleTest() : m_page(DummyPageHolder::create(IntSize(1, 1))) {} |
| + |
| +protected: |
| + Document* document() { return &m_page->document(); } |
| + |
| + OwnPtr<DummyPageHolder> m_page; |
| +}; |
| + |
| +PassRefPtr<FormData> complexFormData() |
| +{ |
| + RefPtr<FormData> data = FormData::create(); |
| + |
| + data->appendData("foo", 3); |
| + data->appendFileRange("/foo/bar/baz", 3, 4, 5); |
| + data->appendFileSystemURLRange(KURL(KURL(), "file:///foo/bar/baz"), 6, 7, 8); |
| + OwnPtr<BlobData> blobData = BlobData::create(); |
| + blobData->appendText("hello", false); |
| + auto size = blobData->length(); |
| + RefPtr<BlobDataHandle> blobDataHandle = BlobDataHandle::create(blobData.release(), size); |
| + data->appendBlob(blobDataHandle->uuid(), blobDataHandle); |
| + Vector<char> boundary; |
| + boundary.append("\0", 1); |
| + data->setBoundary(boundary); |
| + return data.release(); |
| +} |
| + |
| +void verifyFormData(FormData* data) |
|
hiroshige
2015/08/13 07:03:51
nit optional: a name like verifyComplexFormData()
yhirano
2015/08/17 04:47:09
Done.
|
| +{ |
| + const auto& elements = data->elements(); |
| + if (4 != elements.size()) { |
| + FAIL() << "data->elements().size() should be 4, but is " << data->elements().size() << "."; |
| + } |
| + EXPECT_EQ(FormDataElement::data, elements[0].m_type); |
| + EXPECT_EQ("foo", String(elements[0].m_data.data(), elements[0].m_data.size())); |
| + |
| + EXPECT_EQ(FormDataElement::encodedFile, elements[1].m_type); |
| + EXPECT_EQ("/foo/bar/baz", elements[1].m_filename); |
| + EXPECT_EQ(3, elements[1].m_fileStart); |
| + EXPECT_EQ(4, elements[1].m_fileLength); |
| + EXPECT_EQ(5, elements[1].m_expectedFileModificationTime); |
| + |
| + EXPECT_EQ(FormDataElement::encodedFileSystemURL, elements[2].m_type); |
| + EXPECT_EQ(KURL(KURL(), "file:///foo/bar/baz"), elements[2].m_fileSystemURL); |
| + EXPECT_EQ(6, elements[2].m_fileStart); |
| + EXPECT_EQ(7, elements[2].m_fileLength); |
| + EXPECT_EQ(8, elements[2].m_expectedFileModificationTime); |
| + |
| + EXPECT_EQ(FormDataElement::encodedBlob, elements[3].m_type); |
| + if (!elements[3].m_optionalBlobDataHandle) { |
| + FAIL() << "optional BlobDataHandle must be set."; |
| + } |
| + EXPECT_EQ(elements[3].m_blobUUID, elements[3].m_optionalBlobDataHandle->uuid()); |
| + EXPECT_EQ(5u, elements[3].m_optionalBlobDataHandle->size()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ReadFromString) |
| +{ |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String("hello, world")); |
| + HandleReaderRunner<HandleReader> runner(handle.release()); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + EXPECT_EQ("hello, world", toString(r->data())); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, TwoPhaseReadFromString) |
| +{ |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String("hello, world")); |
| + HandleReaderRunner<HandleTwoPhaseReader> runner(handle.release()); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + EXPECT_EQ("hello, world", toString(r->data())); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ReadFromStringNonLatain) |
|
hiroshige
2015/08/13 07:03:51
nit: s/Latain/Latin/
yhirano
2015/08/17 04:47:09
Done.
|
| +{ |
| + UChar cs[] = {0x3042, 0}; |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String(cs)); |
| + HandleReaderRunner<HandleReader> runner(handle.release()); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + EXPECT_EQ("\xe3\x81\x82", toString(r->data())); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ReadFromArrayBuffer) |
| +{ |
| + const unsigned char data[] = { 0x21, 0xfe, 0x00, 0x00, 0xff, 0xa3, 0x42, 0x30, 0x42, 0x99, 0x88 }; |
| + RefPtr<DOMArrayBuffer> buffer = DOMArrayBuffer::create(data, WTF_ARRAY_LENGTH(data)); |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(buffer); |
| + HandleReaderRunner<HandleReader> runner(handle.release()); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + Vector<char> expected; |
| + expected.append(data, WTF_ARRAY_LENGTH(data)); |
| + EXPECT_EQ(expected, r->data()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ReadFromArrayBufferView) |
| +{ |
| + const unsigned char data[] = { 0x21, 0xfe, 0x00, 0x00, 0xff, 0xa3, 0x42, 0x30, 0x42, 0x99, 0x88 }; |
| + const size_t offset = 1, size = 4; |
| + RefPtr<DOMArrayBuffer> buffer = DOMArrayBuffer::create(data, WTF_ARRAY_LENGTH(data)); |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(DOMUint8Array::create(buffer, offset, size)); |
| + HandleReaderRunner<HandleReader> runner(handle.release()); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + Vector<char> expected; |
| + expected.append(data + offset, size); |
| + EXPECT_EQ(expected, r->data()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ReadFromSimplFormData) |
| +{ |
| + RefPtr<FormData> data = FormData::create(); |
| + data->appendData("foo", 3); |
| + data->appendData("hoge", 4); |
| + |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(document(), data); |
| + HandleReaderRunner<HandleReader> runner(handle.release()); |
| + testing::runPendingTasks(); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + EXPECT_EQ("foohoge", toString(r->data())); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ReadFromComplexFormData) |
| +{ |
| + RefPtr<FormData> data = complexFormData(); |
| + OwnPtr<ReplayingHandle> src = ReplayingHandle::create(); |
| + src->add(Command(Command::Data, "bar")); |
| + src->add(Command(Command::Done)); |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::createForTest(document(), data, new LoaderFactory(src.release())); |
| + char c; |
| + size_t readSize; |
| + EXPECT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(&c, 1, kNone, &readSize)); |
| + |
| + HandleReaderRunner<HandleReader> runner(handle.release()); |
| + testing::runPendingTasks(); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + EXPECT_EQ("bar", toString(r->data())); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, TwoPhaseReadFromComplexFormData) |
| +{ |
| + RefPtr<FormData> data = complexFormData(); |
| + OwnPtr<ReplayingHandle> src = ReplayingHandle::create(); |
| + src->add(Command(Command::Data, "bar")); |
| + src->add(Command(Command::Done)); |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::createForTest(document(), data, new LoaderFactory(src.release())); |
| + char c; |
| + size_t readSize; |
| + EXPECT_EQ(kShouldWait, handle->obtainReader(nullptr)->read(&c, 1, kNone, &readSize)); |
| + |
| + HandleReaderRunner<HandleTwoPhaseReader> runner(handle.release()); |
| + testing::runPendingTasks(); |
| + OwnPtr<HandleReadResult> r = runner.wait(); |
| + EXPECT_EQ(kDone, r->result()); |
| + EXPECT_EQ("bar", toString(r->data())); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, DrainAsBlobDataHandle) |
| +{ |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String("hello, world")); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + EXPECT_FALSE(reader->drainAsBlobDataHandle(FetchDataConsumerHandle::Reader::AllowBlobWithInvalidSize)); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, DrainAsFormDataFromString) |
| +{ |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String("hello, world")); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + RefPtr<FormData> formData = reader->drainAsFormData(); |
| + ASSERT_TRUE(formData); |
| + EXPECT_TRUE(formData->isSafeToSendToAnotherThread()); |
| + EXPECT_EQ("hello, world", formData->flattenToString()); |
| + |
| + const void* buffer = nullptr; |
| + size_t size; |
| + EXPECT_EQ(kDone, reader->read(nullptr, 0, kNone, &size)); |
| + EXPECT_EQ(kDone, reader->beginRead(&buffer, kNone, &size)); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, DrainAsFormDataFromArrayBuffer) |
| +{ |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(DOMArrayBuffer::create("foo", 3)); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + RefPtr<FormData> formData = reader->drainAsFormData(); |
| + ASSERT_TRUE(formData); |
| + EXPECT_TRUE(formData->isSafeToSendToAnotherThread()); |
| + EXPECT_EQ("foo", formData->flattenToString()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, DrainAsFormDataFromSimpleFormData) |
| +{ |
| + DOMFormData* data = DOMFormData::create(UTF8Encoding()); |
| + data->append("name1", "value1"); |
| + data->append("name2", "value2"); |
| + RefPtr<FormData> inputFormData = data->createMultiPartFormData(); |
| + |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(document(), inputFormData); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + RefPtr<FormData> outputFormData = reader->drainAsFormData(); |
| + ASSERT_TRUE(outputFormData); |
| + EXPECT_TRUE(outputFormData->isSafeToSendToAnotherThread()); |
| + EXPECT_NE(outputFormData.get(), inputFormData.get()); |
| + EXPECT_EQ(inputFormData->flattenToString(), outputFormData->flattenToString()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, DrainAsFormDataFromComplexFormData) |
| +{ |
| + RefPtr<FormData> inputFormData = complexFormData(); |
| + |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(document(), inputFormData); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + RefPtr<FormData> outputFormData = reader->drainAsFormData(); |
| + ASSERT_TRUE(outputFormData); |
| + EXPECT_TRUE(outputFormData->isSafeToSendToAnotherThread()); |
| + EXPECT_NE(outputFormData.get(), inputFormData.get()); |
| + verifyFormData(outputFormData.get()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ZeroByteReadDoesNotAffectDraining) |
| +{ |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String("hello, world")); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + size_t readSize; |
| + EXPECT_EQ(kOk, reader->read(nullptr, 0, kNone, &readSize)); |
| + RefPtr<FormData> formData = reader->drainAsFormData(); |
| + ASSERT_TRUE(formData); |
| + EXPECT_TRUE(formData->isSafeToSendToAnotherThread()); |
| + EXPECT_EQ("hello, world", formData->flattenToString()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, OneByteReadAffectsDraining) |
| +{ |
| + char c; |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String("hello, world")); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + size_t readSize; |
| + EXPECT_EQ(kOk, reader->read(&c, 1, kNone, &readSize)); |
| + EXPECT_EQ(1u, readSize); |
| + EXPECT_EQ('h', c); |
| + EXPECT_FALSE(reader->drainAsFormData()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, BeginReadAffectsDraining) |
| +{ |
| + const void* buffer = nullptr; |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::create(String("hello, world")); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + size_t available; |
| + EXPECT_EQ(kOk, reader->beginRead(&buffer, kNone, &available)); |
| + ASSERT_TRUE(buffer); |
| + EXPECT_EQ("hello, world", String(static_cast<const char*>(buffer), available)); |
| + EXPECT_FALSE(reader->drainAsFormData()); |
| + reader->endRead(available); |
|
hiroshige
2015/08/13 07:03:50
Should we call reader->endRead(0) and EXPECT_FALSE
yhirano
2015/08/17 04:47:09
Done.
|
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, ZeroByteReadDoesNotAffectDrainingForComplexFormData) |
| +{ |
| + OwnPtr<ReplayingHandle> src = ReplayingHandle::create(); |
| + src->add(Command(Command::Data, "bar")); |
| + src->add(Command(Command::Done)); |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::createForTest(document(), complexFormData(), new LoaderFactory(src.release())); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + size_t readSize; |
| + EXPECT_EQ(kShouldWait, reader->read(nullptr, 0, kNone, &readSize)); |
| + testing::runPendingTasks(); |
| + EXPECT_EQ(kOk, reader->read(nullptr, 0, kNone, &readSize)); |
| + RefPtr<FormData> formData = reader->drainAsFormData(); |
| + ASSERT_TRUE(formData); |
| + EXPECT_TRUE(formData->isSafeToSendToAnotherThread()); |
| + verifyFormData(formData.get()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, OneByteReadAffectsDrainingForComplexFormData) |
| +{ |
| + OwnPtr<ReplayingHandle> src = ReplayingHandle::create(); |
| + src->add(Command(Command::Data, "bar")); |
| + src->add(Command(Command::Done)); |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::createForTest(document(), complexFormData(), new LoaderFactory(src.release())); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + char c; |
| + size_t readSize; |
| + EXPECT_EQ(kShouldWait, reader->read(&c, 1, kNone, &readSize)); |
| + testing::runPendingTasks(); |
| + EXPECT_EQ(kOk, reader->read(&c, 1, kNone, &readSize)); |
| + EXPECT_EQ(1u, readSize); |
| + EXPECT_EQ('b', c); |
| + EXPECT_FALSE(reader->drainAsFormData()); |
| +} |
| + |
| +TEST_F(FetchFormDataConsumerHandleTest, BeginReadAffectsDrainingForComplexFormData) |
| +{ |
| + OwnPtr<ReplayingHandle> src = ReplayingHandle::create(); |
| + src->add(Command(Command::Data, "bar")); |
| + src->add(Command(Command::Done)); |
| + const void* buffer = nullptr; |
| + OwnPtr<FetchDataConsumerHandle> handle = FetchFormDataConsumerHandle::createForTest(document(), complexFormData(), new LoaderFactory(src.release())); |
| + OwnPtr<FetchDataConsumerHandle::Reader> reader = handle->obtainReader(nullptr); |
| + size_t available; |
| + EXPECT_EQ(kShouldWait, reader->beginRead(&buffer, kNone, &available)); |
| + testing::runPendingTasks(); |
| + EXPECT_EQ(kOk, reader->beginRead(&buffer, kNone, &available)); |
| + EXPECT_FALSE(reader->drainAsFormData()); |
| + reader->endRead(0); |
| + EXPECT_FALSE(reader->drainAsFormData()); |
| +} |
| + |
| +} // namespace |
| +} // namespace blink |