Introduce FetchFormDataConsumerHandle.

Source/modules/fetch/FetchFormDataConsumerHandleTest.cpp

+// 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/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/FormData.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);
+    return data.release();
+}
+
+void verifyFormData(FormData* data)
+{
+    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)
+{
+    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)
+{
+    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, 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->hasOneRef());
+    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->hasOneRef());
+    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->hasOneRef());
+    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->hasOneRef());
+    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->hasOneRef());
+    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);
+}
+
+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->hasOneRef());
+    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