Chromium Code Reviews| Index: third_party/WebKit/Source/modules/fetch/MultipartParserTest.cpp |
| diff --git a/third_party/WebKit/Source/modules/fetch/MultipartParserTest.cpp b/third_party/WebKit/Source/modules/fetch/MultipartParserTest.cpp |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..2b2cb271b52144b9cdc811c94700285c49f31f20 |
| --- /dev/null |
| +++ b/third_party/WebKit/Source/modules/fetch/MultipartParserTest.cpp |
| @@ -0,0 +1,232 @@ |
| +// Copyright 2016 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/MultipartParser.h" |
| + |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +#include <algorithm> |
| + |
| +namespace blink { |
| + |
| +namespace { |
| + |
| +class MockClient final : public GarbageCollectedFinalized<MockClient>, public MultipartParser::Client { |
| + USING_GARBAGE_COLLECTED_MIXIN(MockClient); |
| +public: |
| + struct Part { |
| + Part() = default; |
| + explicit Part(const ResourceResponse& response) : response(response), dataFullyReceived(false) {} |
| + ResourceResponse response; |
| + Vector<char> data; |
| + bool dataFullyReceived; |
| + }; |
| + void partHeaderFieldsInMultipartReceived(const ResourceResponse& response) override |
| + { |
| + m_parts.append(response); |
| + } |
| + void partDataInMultipartReceived(const char* bytes, size_t size) override |
| + { |
| + m_parts.last().data.append(bytes, size); |
| + } |
| + void partDataInMultipartFullyReceived() override |
| + { |
| + m_parts.last().dataFullyReceived = true; |
| + } |
| + size_t numberOfParts() const |
| + { |
| + return m_parts.size(); |
| + } |
| + const Part& part(size_t partIndex) const |
| + { |
| + EXPECT_LT(partIndex, numberOfParts()); |
| + return partIndex < numberOfParts() ? m_parts[partIndex] : m_emptyPart; |
| + } |
| +private: |
| + Part m_emptyPart; |
| + Vector<Part> m_parts; |
| +}; |
| + |
| +const char* kBytes = |
|
yhirano
2016/09/21 09:02:58
[optional] How about using |constexpr char kBytes[
e_hakkinen
2016/09/22 22:27:16
Done.
|
| + "preamble" |
| + "\r\n--boundary\r\n\r\n" |
| + "\r\n--boundary\r\ncontent-type: application/xhtml+xml\r\n\r\n1" |
| + "\r\n--boundary\t\r\ncontent-type: text/html\r\n\r\n2\r\n--\r\n--bound--\r\n--\r\n2\r\n" |
| + "\r\n--boundary \r\ncontent-type: text/plain\r\n\r\n333" |
| + "\r\n--boundary--\t \r\n" |
| + "epilogue"; |
| + |
| +TEST(MultipartParserTest, AppendDataInChunks) |
| +{ |
| + const size_t sizes[] = {1u, 2u, strlen(kBytes)}; |
| + |
| + Vector<char> boundary; |
| + boundary.append("boundary", 8u); |
| + for (size_t size : sizes) { |
| + MockClient* client = new MockClient; |
| + MultipartParser* parser = new MultipartParser(boundary, client); |
| + |
| + for (size_t i = 0u; kBytes[i]; i += size) |
|
yhirano
2016/09/21 09:02:58
|i| can be greater than or equal to sizeof(kBytes)
e_hakkinen
2016/09/22 22:27:17
Done.
|
| + EXPECT_TRUE(parser->appendData(kBytes + i, std::min(size, strlen(kBytes + i)))); |
| + EXPECT_TRUE(parser->finish()) << " size=" << size; |
| + EXPECT_EQ(4u, client->numberOfParts()) << " size=" << size; |
| + EXPECT_EQ(0u, client->part(0).data.size()); |
| + EXPECT_TRUE(client->part(0).dataFullyReceived); |
| + EXPECT_EQ(String("1"), client->part(1).data); |
|
yhirano
2016/09/21 09:02:58
Can you verify the content type for each part?
e_hakkinen
2016/09/22 22:27:16
Sure.
|
| + EXPECT_TRUE(client->part(1).dataFullyReceived); |
| + EXPECT_EQ(String("2\r\n--\r\n--bound--\r\n--\r\n2\r\n"), client->part(2).data); |
| + EXPECT_TRUE(client->part(2).dataFullyReceived); |
| + EXPECT_EQ(String("333"), client->part(3).data); |
| + EXPECT_TRUE(client->part(3).dataFullyReceived); |
| + } |
| +} |
| + |
| +TEST(MultipartParserTest, Epilogue) |
| +{ |
| + const size_t ends[] = { |
|
yhirano
2016/09/21 09:02:58
constexpr
e_hakkinen
2016/09/22 22:27:16
Done.
|
| + 0u, // Non-empty epilogue in the end. |
| + 8u, // Empty epilogue in the end. |
| + 9u, // Partial CRLF after close delimiter in the end. |
| + 10u, // No CRLF after close delimiter in the end. |
| + 12u, // No transport padding nor CRLF after close delimiter in the end. |
| + 13u, // Partial close delimiter in the end. |
| + 14u, // No close delimiter but a delimiter in the end. |
| + 15u // Partial delimiter in the end. |
| + }; |
| + |
| + Vector<char> boundary; |
| + boundary.append("boundary", 8u); |
| + for (size_t end : ends) { |
| + MockClient* client = new MockClient; |
| + MultipartParser* parser = new MultipartParser(boundary, client); |
| + |
| + EXPECT_TRUE(parser->appendData(kBytes, strlen(kBytes) - end)); |
| + EXPECT_EQ(end <= 12u, parser->finish()) << " end=" << end; |
| + EXPECT_EQ(4u, client->numberOfParts()) << " end=" << end; |
| + EXPECT_EQ(0u, client->part(0).data.size()); |
| + EXPECT_TRUE(client->part(0).dataFullyReceived); |
| + EXPECT_EQ(String("1"), client->part(1).data); |
| + EXPECT_TRUE(client->part(1).dataFullyReceived); |
| + EXPECT_EQ(String("2\r\n--\r\n--bound--\r\n--\r\n2\r\n"), client->part(2).data); |
| + EXPECT_TRUE(client->part(2).dataFullyReceived); |
| + switch (end) { |
| + case 14u: |
| + EXPECT_EQ(String("333"), client->part(3).data); |
| + EXPECT_TRUE(client->part(3).dataFullyReceived); |
| + break; |
| + case 15u: |
| + EXPECT_EQ(String("333\r\n--boundar"), client->part(3).data); |
| + EXPECT_FALSE(client->part(3).dataFullyReceived); |
| + break; |
| + default: |
| + EXPECT_EQ(String("333"), client->part(3).data); |
|
yhirano
2016/09/21 09:02:58
Are this case and the |14| case same?
e_hakkinen
2016/09/22 22:27:16
Yep, they are now. When I tested parser->finish()
|
| + EXPECT_TRUE(client->part(3).dataFullyReceived); |
| + } |
| + } |
| +} |
| + |
| +TEST(MultipartParserTest, NoEndBoundary) |
| +{ |
| + const char* bytes = "--boundary\r\ncontent-type: application/xhtml+xml\r\n\r\n1"; |
|
yhirano
2016/09/21 09:02:58
How about using constexpr char[]?
e_hakkinen
2016/09/22 22:27:16
Done.
|
| + |
| + Vector<char> boundary; |
| + boundary.append("boundary", 8u); |
| + MockClient* client = new MockClient; |
| + MultipartParser* parser = new MultipartParser(boundary, client); |
| + |
| + EXPECT_TRUE(parser->appendData(bytes, strlen(bytes))); |
| + EXPECT_FALSE(parser->finish()); // No close delimiter. |
| + EXPECT_EQ(1u, client->numberOfParts()); |
| + EXPECT_EQ(String("1"), client->part(0).data); |
| + EXPECT_FALSE(client->part(0).dataFullyReceived); |
| +} |
| + |
| +TEST(MultipartParserTest, NoStartBoundary) |
| +{ |
| + const char* bytes = "content-type: application/xhtml+xml\r\n\r\n1\r\n--boundary--\r\n"; |
| + |
| + Vector<char> boundary; |
| + boundary.append("boundary", 8u); |
| + MockClient* client = new MockClient; |
| + MultipartParser* parser = new MultipartParser(boundary, client); |
| + |
| + EXPECT_FALSE(parser->appendData(bytes, strlen(bytes))); // Close delimiter before delimiter. |
| + EXPECT_FALSE(parser->finish()); // No parts. |
| + EXPECT_EQ(0u, client->numberOfParts()); |
| +} |
| + |
| +TEST(MultipartParserTest, NoStartNorEndBoundary) |
| +{ |
| + const char* bytes = "content-type: application/xhtml+xml\r\n\r\n1"; |
|
yhirano
2016/09/21 09:02:58
ditto
e_hakkinen
2016/09/22 22:27:17
Done.
|
| + |
| + Vector<char> boundary; |
| + boundary.append("boundary", 8u); |
| + MockClient* client = new MockClient; |
| + MultipartParser* parser = new MultipartParser(boundary, client); |
| + |
| + EXPECT_TRUE(parser->appendData(bytes, strlen(bytes))); // Valid preamble. |
| + EXPECT_FALSE(parser->finish()); // No parts. |
| + EXPECT_EQ(0u, client->numberOfParts()); |
| +} |
| + |
| +const size_t kStarts[] = { |
|
yhirano
2016/09/21 09:02:58
constexpr
e_hakkinen
2016/09/22 22:27:16
Done.
|
| + 0u, // Non-empty preamble in the beginning. |
| + 8u, // Empty preamble in the beginning. |
| + 9u, // Truncated delimiter in the beginning. |
| + 10u, // No preamble in the beginning. |
| + 11u // Truncated dash boundary in the beginning. |
| +}; |
| + |
| +TEST(MultipartParserTest, Preamble) |
| +{ |
| + Vector<char> boundary; |
| + boundary.append("boundary", 8u); |
| + for (size_t start : kStarts) { |
| + MockClient* client = new MockClient; |
| + MultipartParser* parser = new MultipartParser(boundary, client); |
| + |
| + EXPECT_TRUE(parser->appendData(kBytes + start, strlen(kBytes + start))); |
| + EXPECT_TRUE(parser->finish()); |
| + switch (start) { |
| + case 9u: |
| + case 11u: |
| + EXPECT_EQ(3u, client->numberOfParts()) << " start=" << start; |
| + EXPECT_EQ(String("1"), client->part(0).data); |
| + EXPECT_TRUE(client->part(0).dataFullyReceived); |
| + EXPECT_EQ(String("2\r\n--\r\n--bound--\r\n--\r\n2\r\n"), client->part(1).data); |
| + EXPECT_TRUE(client->part(1).dataFullyReceived); |
| + EXPECT_EQ(String("333"), client->part(2).data); |
| + EXPECT_TRUE(client->part(2).dataFullyReceived); |
| + break; |
| + default: |
| + EXPECT_EQ(4u, client->numberOfParts()) << " start=" << start; |
| + EXPECT_EQ(0u, client->part(0).data.size()); |
| + EXPECT_TRUE(client->part(0).dataFullyReceived); |
| + EXPECT_EQ(String("1"), client->part(1).data); |
| + EXPECT_TRUE(client->part(1).dataFullyReceived); |
| + EXPECT_EQ(String("2\r\n--\r\n--bound--\r\n--\r\n2\r\n"), client->part(2).data); |
| + EXPECT_TRUE(client->part(2).dataFullyReceived); |
| + EXPECT_EQ(String("333"), client->part(3).data); |
| + EXPECT_TRUE(client->part(3).dataFullyReceived); |
| + } |
| + } |
| +} |
| + |
| +TEST(MultipartParserTest, PreambleWithMalformedBoundary) |
| +{ |
| + Vector<char> boundary; |
| + boundary.append("--boundary", 10u); |
| + for (size_t start : kStarts) { |
| + MockClient* client = new MockClient; |
| + MultipartParser* parser = new MultipartParser(boundary, client); |
| + |
| + EXPECT_TRUE(parser->appendData(kBytes + start, strlen(kBytes + start))); // Valid preamble. |
| + EXPECT_FALSE(parser->finish()); // No parts. |
| + EXPECT_EQ(0u, client->numberOfParts()); |
| + } |
| +} |
| + |
| +} // namespace |
| + |
| +} // namespace blink |