| Index: third_party/WebKit/Source/modules/fetch/FetchDataLoaderTest.cpp
|
| diff --git a/third_party/WebKit/Source/modules/fetch/FetchDataLoaderTest.cpp b/third_party/WebKit/Source/modules/fetch/FetchDataLoaderTest.cpp
|
| index ea13c9fab03ce2d04e052138b7754d719a50428d..dfe1df50e0929d778d008e79730eae2c4fe68e01 100644
|
| --- a/third_party/WebKit/Source/modules/fetch/FetchDataLoaderTest.cpp
|
| +++ b/third_party/WebKit/Source/modules/fetch/FetchDataLoaderTest.cpp
|
| @@ -4,6 +4,8 @@
|
|
|
| #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"
|
| @@ -31,6 +33,29 @@ 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;
|
| @@ -350,6 +375,179 @@ TEST(FetchDataLoaderTest, LoadAsArrayBufferCancel) {
|
| 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;
|
|
|