Move multipart resource handling to core/fetch (2/2)

third_party/WebKit/Source/core/fetch/MultipartImageResourceParserTest.cpp

 // 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 "core/fetch/MultipartImageResourceParser.h"
 
 #include "platform/network/ResourceResponse.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebURL.h"
 #include "public/platform/WebURLResponse.h"
@@ skipping 11 matching lines @@
 {
     if (data.isEmpty())
         return String("");
     return String(data.data(), data.size());
 }
 
 class MockClient final : public NoBaseWillBeGarbageCollectedFinalized<MockClient>, public MultipartImageResourceParser::Client {
     WILL_BE_USING_GARBAGE_COLLECTED_MIXIN(MockClient);
 
 public:
-    void onePartInMultipartReceived(const ResourceResponse& response) override
+    void onePartInMultipartReceived(const ResourceResponse& response, bool isFirstPart) override
     {
+        if (isFirstPart != m_responses.isEmpty())
+            FAIL() << "m_responses.size() = " << m_responses.size() << ", isFirstPart = " << isFirstPart;
+
         m_responses.append(response);
         m_data.append(Vector<char>());
     }
     void multipartDataReceived(const char* bytes, size_t size) override
     {
         m_data.last().append(bytes, size);
     }
 
     Vector<ResourceResponse> m_responses;
     Vector<Vector<char>> m_data;
@@ skipping 391 matching lines @@
     MultipartImageResourceParser* parser = new MultipartImageResourceParser(response, boundary, client);
 
     const char data[] = "--bound\r\n\r\n--bound\r\n\r\nfoofoo--bound--";
     parser->appendData(data, strlen(data));
     ASSERT_EQ(2u, client->m_responses.size());
     ASSERT_EQ(2u, client->m_data.size());
     EXPECT_EQ("", toString(client->m_data[0]));
     EXPECT_EQ("foofoo", toString(client->m_data[1]));
 }
 
-TEST(MultipartResponseTest, MultipartPayloadSet)
+TEST(MultipartResponseTest, IsFirstPartSet)
 {
     ResourceResponse response;
     response.setMimeType("multipart/x-mixed-replace");
     MockClient* client = new MockClient;
     Vector<char> boundary;
     boundary.append("bound", 5);
 
     MultipartImageResourceParser* parser = new MultipartImageResourceParser(response, boundary, client);
 
+    // isFirstPart is checked at MockClient::didReceiveResponse.
     const char data[] =
         "--bound\n"
         "Content-type: text/plain\n\n"
         "response data\n"
         "--bound\n";
     parser->appendData(data, strlen(data));
     ASSERT_EQ(1u, client->m_responses.size());
     ASSERT_EQ(1u, client->m_data.size());
     EXPECT_EQ("response data", toString(client->m_data[0]));
-    EXPECT_FALSE(client->m_responses[0].isMultipartPayload());
 
     const char data2[] =
         "Content-type: text/plain\n\n"
         "response data2\n"
         "--bound\n";
     parser->appendData(data2, strlen(data2));
     ASSERT_EQ(2u, client->m_responses.size());
     ASSERT_EQ(2u, client->m_data.size());
     EXPECT_EQ("response data2", toString(client->m_data[1]));
-    EXPECT_TRUE(client->m_responses[1].isMultipartPayload());
 }
 
 } // namespace
 
 } // namespace blink