[Fetch API] Implement Request.formData and Response.formData.

third_party/WebKit/Source/modules/fetch/FetchDataLoaderTest.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 "modules/fetch/FetchDataLoader.h"
 
+#include "core/fileapi/Blob.h"
+#include "core/html/FormData.h"
 #include "modules/fetch/BytesConsumerForDataConsumerHandle.h"
 #include "modules/fetch/BytesConsumerTestUtil.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include <memory>
 
 namespace blink {
 
 namespace {
 
 using ::testing::ByMove;
 using ::testing::InSequence;
 using ::testing::Return;
 using ::testing::DoAll;
 using ::testing::StrictMock;
 using ::testing::_;
 using ::testing::SaveArg;
 using ::testing::SetArgPointee;
 using Checkpoint = StrictMock<::testing::MockFunction<void(int)>>;
 using MockFetchDataLoaderClient =
     BytesConsumerTestUtil::MockFetchDataLoaderClient;
 using MockBytesConsumer = BytesConsumerTestUtil::MockBytesConsumer;
 using Result = BytesConsumer::Result;
 
 constexpr char kQuickBrownFox[] = "Quick brown fox";
 constexpr size_t kQuickBrownFoxLength = 15;
 constexpr size_t kQuickBrownFoxLengthWithTerminatingNull = 16;
+constexpr char kQuickBrownFoxFormData[] =
+    "--boundary\r\n"
+    "Content-Disposition: form-data; name=blob; filename=blob\r\n"
+    "Content-Type: text/plain; charset=iso-8859-1\r\n"
+    "\r\n"
+    "Quick brown fox\r\n"
+    "--boundary\r\n"
+    "Content-Disposition: form-data; name=blob\xC2\xA0without\xC2\xA0type; "
+    "filename=\"blob\xC2\xA0without\xC2\xA0type.txt\"\r\n"
+    "\r\n"
+    "Quick brown fox\r\n"
+    "--boundary\r\n"
+    "Content-Disposition: form-data; name=string\r\n"
+    "\r\n"
+    "Quick brown fox\r\n"
+    "--boundary\r\n"
+    "Content-Disposition: form-data; name=string-with-type\r\n"
+    "Content-Type: text/plain; charset=invalid\r\n"
+    "\r\n"
+    "Quick brown fox\r\n"
+    "--boundary--\r\n";
+constexpr size_t kQuickBrownFoxFormDataLength =
+    WTF_ARRAY_LENGTH(kQuickBrownFoxFormData) - 1u;
 
 TEST(FetchDataLoaderTest, LoadAsBlob) {
   Checkpoint checkpoint;
   BytesConsumer::Client* client = nullptr;
   MockBytesConsumer* consumer = MockBytesConsumer::create();
 
   FetchDataLoader* fetchDataLoader =
       FetchDataLoader::createLoaderAsBlobHandle("text/test");
   MockFetchDataLoaderClient* fetchDataLoaderClient =
       MockFetchDataLoaderClient::create();
@@ skipping 299 matching lines @@
   EXPECT_CALL(*consumer, cancel());
   EXPECT_CALL(checkpoint, Call(3));
 
   checkpoint.Call(1);
   fetchDataLoader->start(consumer, fetchDataLoaderClient);
   checkpoint.Call(2);
   fetchDataLoader->cancel();
   checkpoint.Call(3);
 }
 
+TEST(FetchDataLoaderTest, LoadAsFormData) {
+  Checkpoint checkpoint;
+  BytesConsumer::Client* client = nullptr;
+  MockBytesConsumer* consumer = MockBytesConsumer::create();
+
+  FetchDataLoader* fetchDataLoader =
+      FetchDataLoader::createLoaderAsFormData("boundary");
+  MockFetchDataLoaderClient* fetchDataLoaderClient =
+      MockFetchDataLoaderClient::create();
+  FormData* formData = nullptr;
+
+  InSequence s;
+  EXPECT_CALL(checkpoint, Call(1));
+  EXPECT_CALL(*consumer, setClient(_)).WillOnce(SaveArg<0>(&client));
+  EXPECT_CALL(*consumer, beginRead(_, _))
+      .WillOnce(DoAll(SetArgPointee<0>(nullptr), SetArgPointee<1>(0),
+                      Return(Result::ShouldWait)));
+  EXPECT_CALL(checkpoint, Call(2));
+  EXPECT_CALL(*consumer, beginRead(_, _))
+      .WillOnce(DoAll(SetArgPointee<0>(kQuickBrownFoxFormData),
+                      SetArgPointee<1>(kQuickBrownFoxFormDataLength),
+                      Return(Result::Ok)));
+  EXPECT_CALL(*consumer, endRead(kQuickBrownFoxFormDataLength))
+      .WillOnce(Return(Result::Ok));
+  EXPECT_CALL(*consumer, beginRead(_, _)).WillOnce(Return(Result::Done));
+  EXPECT_CALL(*fetchDataLoaderClient, didFetchDataLoadedFormDataMock(_))
+      .WillOnce(SaveArg<0>(&formData));
+  EXPECT_CALL(checkpoint, Call(3));
+  EXPECT_CALL(*consumer, cancel());
+  EXPECT_CALL(checkpoint, Call(4));
+
+  checkpoint.Call(1);
+  fetchDataLoader->start(consumer, fetchDataLoaderClient);
+  checkpoint.Call(2);
+  ASSERT_TRUE(client);
+  client->onStateChange();
+  checkpoint.Call(3);
+  fetchDataLoader->cancel();
+  checkpoint.Call(4);
+
+  ASSERT_TRUE(formData);
+  ASSERT_EQ(4u, formData->entries().size());
+
+  EXPECT_EQ("blob", formData->entries()[0]->name());
+  EXPECT_EQ("blob", formData->entries()[0]->filename());
+  ASSERT_TRUE(formData->entries()[0]->isFile());
+  EXPECT_EQ(kQuickBrownFoxLength, formData->entries()[0]->blob()->size());
+  EXPECT_EQ("text/plain; charset=iso-8859-1",
+            formData->entries()[0]->blob()->type());
+
+  EXPECT_EQ("blob\xC2\xA0without\xC2\xA0type", formData->entries()[1]->name());
+  EXPECT_EQ("blob\xC2\xA0without\xC2\xA0type.txt",
+            formData->entries()[1]->filename().utf8());
+  ASSERT_TRUE(formData->entries()[1]->isFile());
+  EXPECT_EQ(kQuickBrownFoxLength, formData->entries()[1]->blob()->size());
+  EXPECT_EQ("text/plain", formData->entries()[1]->blob()->type());
+
+  EXPECT_EQ("string", formData->entries()[2]->name());
+  EXPECT_TRUE(formData->entries()[2]->filename().isNull());
+  ASSERT_TRUE(formData->entries()[2]->isString());
+  EXPECT_EQ(kQuickBrownFox, formData->entries()[2]->value());
+
+  EXPECT_EQ("string-with-type", formData->entries()[3]->name());
+  EXPECT_TRUE(formData->entries()[3]->filename().isNull());
+  ASSERT_TRUE(formData->entries()[3]->isString());
+  EXPECT_EQ(kQuickBrownFox, formData->entries()[3]->value());
+}
+
+TEST(FetchDataLoaderTest, LoadAsFormDataPartialInput) {
+  Checkpoint checkpoint;
+  BytesConsumer::Client* client = nullptr;
+  MockBytesConsumer* consumer = MockBytesConsumer::create();
+
+  FetchDataLoader* fetchDataLoader =
+      FetchDataLoader::createLoaderAsFormData("boundary");
+  MockFetchDataLoaderClient* fetchDataLoaderClient =
+      MockFetchDataLoaderClient::create();
+
+  InSequence s;
+  EXPECT_CALL(checkpoint, Call(1));
+  EXPECT_CALL(*consumer, setClient(_)).WillOnce(SaveArg<0>(&client));
+  EXPECT_CALL(*consumer, beginRead(_, _))
+      .WillOnce(DoAll(SetArgPointee<0>(nullptr), SetArgPointee<1>(0),
+                      Return(Result::ShouldWait)));
+  EXPECT_CALL(checkpoint, Call(2));
+  EXPECT_CALL(*consumer, beginRead(_, _))
+      .WillOnce(DoAll(SetArgPointee<0>(kQuickBrownFoxFormData),
+                      SetArgPointee<1>(kQuickBrownFoxFormDataLength - 3u),
+                      Return(Result::Ok)));
+  EXPECT_CALL(*consumer, endRead(kQuickBrownFoxFormDataLength - 3u))
+      .WillOnce(Return(Result::Ok));
+  EXPECT_CALL(*consumer, beginRead(_, _)).WillOnce(Return(Result::Done));
+  EXPECT_CALL(*fetchDataLoaderClient, didFetchDataLoadFailed());
+  EXPECT_CALL(checkpoint, Call(3));
+  EXPECT_CALL(*consumer, cancel());
+  EXPECT_CALL(checkpoint, Call(4));
+
+  checkpoint.Call(1);
+  fetchDataLoader->start(consumer, fetchDataLoaderClient);
+  checkpoint.Call(2);
+  ASSERT_TRUE(client);
+  client->onStateChange();
+  checkpoint.Call(3);
+  fetchDataLoader->cancel();
+  checkpoint.Call(4);
+}
+
+TEST(FetchDataLoaderTest, LoadAsFormDataFailed) {
+  Checkpoint checkpoint;
+  BytesConsumer::Client* client = nullptr;
+  MockBytesConsumer* consumer = MockBytesConsumer::create();
+
+  FetchDataLoader* fetchDataLoader =
+      FetchDataLoader::createLoaderAsFormData("boundary");
+  MockFetchDataLoaderClient* fetchDataLoaderClient =
+      MockFetchDataLoaderClient::create();
+
+  InSequence s;
+  EXPECT_CALL(checkpoint, Call(1));
+  EXPECT_CALL(*consumer, setClient(_)).WillOnce(SaveArg<0>(&client));
+  EXPECT_CALL(*consumer, beginRead(_, _))
+      .WillOnce(DoAll(SetArgPointee<0>(nullptr), SetArgPointee<1>(0),
+                      Return(Result::ShouldWait)));
+  EXPECT_CALL(checkpoint, Call(2));
+  EXPECT_CALL(*consumer, beginRead(_, _))
+      .WillOnce(DoAll(SetArgPointee<0>(kQuickBrownFoxFormData),
+                      SetArgPointee<1>(kQuickBrownFoxFormDataLength),
+                      Return(Result::Ok)));
+  EXPECT_CALL(*consumer, endRead(kQuickBrownFoxFormDataLength))
+      .WillOnce(Return(Result::Ok));
+  EXPECT_CALL(*consumer, beginRead(_, _)).WillOnce(Return(Result::Error));
+  EXPECT_CALL(*fetchDataLoaderClient, didFetchDataLoadFailed());
+  EXPECT_CALL(checkpoint, Call(3));
+  EXPECT_CALL(*consumer, cancel());
+  EXPECT_CALL(checkpoint, Call(4));
+
+  checkpoint.Call(1);
+  fetchDataLoader->start(consumer, fetchDataLoaderClient);
+  checkpoint.Call(2);
+  ASSERT_TRUE(client);
+  client->onStateChange();
+  checkpoint.Call(3);
+  fetchDataLoader->cancel();
+  checkpoint.Call(4);
+}
+
+TEST(FetchDataLoaderTest, LoadAsFormDataCancel) {
+  Checkpoint checkpoint;
+  BytesConsumer::Client* client = nullptr;
+  MockBytesConsumer* consumer = MockBytesConsumer::create();
+
+  FetchDataLoader* fetchDataLoader =
+      FetchDataLoader::createLoaderAsFormData("boundary");
+  MockFetchDataLoaderClient* fetchDataLoaderClient =
+      MockFetchDataLoaderClient::create();
+
+  InSequence s;
+  EXPECT_CALL(checkpoint, Call(1));
+  EXPECT_CALL(*consumer, setClient(_)).WillOnce(SaveArg<0>(&client));
+  EXPECT_CALL(*consumer, beginRead(_, _))
+      .WillOnce(DoAll(SetArgPointee<0>(nullptr), SetArgPointee<1>(0),
+                      Return(Result::ShouldWait)));
+  EXPECT_CALL(checkpoint, Call(2));
+  EXPECT_CALL(*consumer, cancel());
+  EXPECT_CALL(checkpoint, Call(3));
+
+  checkpoint.Call(1);
+  fetchDataLoader->start(consumer, fetchDataLoaderClient);
+  checkpoint.Call(2);
+  fetchDataLoader->cancel();
+  checkpoint.Call(3);
+}
+
 TEST(FetchDataLoaderTest, LoadAsString) {
   Checkpoint checkpoint;
   BytesConsumer::Client* client = nullptr;
   MockBytesConsumer* consumer = MockBytesConsumer::create();
 
   FetchDataLoader* fetchDataLoader = FetchDataLoader::createLoaderAsString();
   MockFetchDataLoaderClient* fetchDataLoaderClient =
       MockFetchDataLoaderClient::create();
 
   InSequence s;
@@ skipping 126 matching lines @@
   checkpoint.Call(1);
   fetchDataLoader->start(consumer, fetchDataLoaderClient);
   checkpoint.Call(2);
   fetchDataLoader->cancel();
   checkpoint.Call(3);
 }
 
 }  // namespace
 
 }  // namespace blink