Merge page serializers [7/12]

Source/web/tests/MHTMLTest.cpp

 /*
  * Copyright (C) 2014 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 16 matching lines @@
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 
 #include "core/dom/Document.h"
 #include "core/frame/LocalFrame.h"
 #include "core/frame/Location.h"
 #include "core/page/Page.h"
+#include "platform/SerializedResource.h"
+#include "platform/SharedBuffer.h"
+#include "platform/mhtml/MHTMLArchive.h"
 #include "platform/testing/URLTestHelpers.h"
 #include "platform/weborigin/KURL.h"
 #include "public/platform/Platform.h"
 #include "public/platform/WebString.h"
 #include "public/platform/WebURL.h"
 #include "public/platform/WebURLRequest.h"
 #include "public/platform/WebURLResponse.h"
 #include "public/platform/WebUnitTestSupport.h"
 #include "public/web/WebDocument.h"
 #include "public/web/WebFrame.h"
 #include "public/web/WebView.h"
 #include "web/tests/FrameTestHelpers.h"
 #include <gtest/gtest.h>
 
+using namespace blink;
+
+using blink::URLTestHelpers::toKURL;
+
 namespace {
 
+class LineReader {
+public:
+    LineReader(const std::string& text) : m_text(text), m_index(0) { }
+    bool getNextLine(std::string* line)
+    {
+        line->clear();
+        if (m_index >= m_text.length())
+            return false;
+
+        size_t endOfLineIndex = m_text.find("\r\n", m_index);
+        if (endOfLineIndex == std::string::npos) {
+            *line = m_text.substr(m_index);
+            m_index = m_text.length();
+        } else {

[hajimehoshi, 2015/06/10 09:02:03]

nit: early return is better

[Tiger (Sony Mobile), 2015/06/10 10:30:36]

Done.

+            *line = m_text.substr(m_index, endOfLineIndex - m_index);
+            m_index = endOfLineIndex + 2;
+        }
+        return true;
+    }
+
+private:
+    std::string m_text;
+    size_t m_index;
+};
+
 using blink::URLTestHelpers::toKURL;
 using namespace blink;
 
 class MHTMLTest : public testing::Test {
 public:
     MHTMLTest()
     {
+        m_filePath = Platform::current()->unitTestSupport()->webKitRootDir();
+        m_filePath.append("/Source/web/tests/data/mhtml/");
     }
 
 protected:
     virtual void SetUp()
     {
         m_helper.initialize();
     }
 
     virtual void TearDown()
     {
         Platform::current()->unitTestSupport()->unregisterAllMockedURLs();
     }
 
     void registerMockedURLLoad(const std::string& url, const WebString& fileName)
     {
         URLTestHelpers::registerMockedURLLoad(toKURL(url), fileName, WebString::fromUTF8("mhtml/"), WebString::fromUTF8("text/html"));
     }
 
     void loadURLInTopFrame(const WebURL& url)
     {
         FrameTestHelpers::loadFrame(m_helper.webView()->mainFrame(), url.string().utf8().data());
     }
 
     Page* page() const { return m_helper.webViewImpl()->page(); }
 
+
+    void addResource(const char* url, const char* mime, PassRefPtr<SharedBuffer> data)
+    {
+        SerializedResource resource(toKURL(url), mime, data);
+        m_resources.append(resource);
+    }
+
+    void addResource(const char* url, const char* mime, const char* fileName)
+    {
+        addResource(url, mime, readFile(fileName));
+    }
+
+    void addTestResources()
+    {
+        addResource("http://www.test.com", "text/html", "css_test_page.html");
+        addResource("http://www.test.com/link_styles.css", "text/css", "link_styles.css");
+        addResource("http://www.test.com/import_style_from_link.css", "text/css", "import_style_from_link.css");
+        addResource("http://www.test.com/import_styles.css", "text/css", "import_styles.css");
+        addResource("http://www.test.com/red_background.png", "image/png", "red_background.png");
+        addResource("http://www.test.com/orange_background.png", "image/png", "orange_background.png");
+        addResource("http://www.test.com/yellow_background.png", "image/png", "yellow_background.png");
+        addResource("http://www.test.com/green_background.png", "image/png", "green_background.png");
+        addResource("http://www.test.com/blue_background.png", "image/png", "blue_background.png");
+        addResource("http://www.test.com/purple_background.png", "image/png", "purple_background.png");
+        addResource("http://www.test.com/ul-dot.png", "image/png", "ul-dot.png");
+        addResource("http://www.test.com/ol-dot.png", "image/png", "ol-dot.png");
+    }
+
+    PassRefPtr<SharedBuffer> serialize(const char *title, const char *mime,  MHTMLArchive::EncodingPolicy encodingPolicy)
+    {
+        return MHTMLArchive::generateMHTMLData(m_resources, encodingPolicy, title, mime);
+    }
+
+
+
 private:
+    PassRefPtr<SharedBuffer> readFile(const char* fileName)
+    {
+        String filePath = m_filePath + fileName;
+        return Platform::current()->unitTestSupport()->readFromFile(filePath);
+    }
+
+    String m_filePath;
+    Vector<SerializedResource> m_resources;
     FrameTestHelpers::WebViewHelper m_helper;
 };
 
 // Checks that the domain is set to the actual MHTML file, not the URL it was
 // generated from.
 TEST_F(MHTMLTest, CheckDomain)
 {
     const char kFileURL[] = "file:///simple_test.mht";
 
     // Register the mocked frame and load it.
     WebURL url = toKURL(kFileURL);
     registerMockedURLLoad(kFileURL, WebString::fromUTF8("simple_test.mht"));
     loadURLInTopFrame(url);
     ASSERT_TRUE(page());
     LocalFrame* frame = toLocalFrame(page()->mainFrame());
     ASSERT_TRUE(frame);
     Document* document = frame->document();
     ASSERT_TRUE(document);
 
     EXPECT_STREQ(kFileURL, frame->domWindow()->location()->href().ascii().data());
 
     SecurityOrigin* origin = document->securityOrigin();
     EXPECT_STRNE("localhost", origin->domain().ascii().data());
 }
 
+TEST_F(MHTMLTest, TestMHTMLEncoding)
+{
+    addTestResources();
+    RefPtr<SharedBuffer> data  = serialize("Test Serialization", "text/html", MHTMLArchive::UseDefaultEncoding);
+
+    // Read the MHTML data line per line and do some pseudo-parsing to make sure the right encoding is used for the different sections.
+    LineReader lineReader(std::string(data->data()));
+    int sectionCheckedCount = 0;
+    const char* expectedEncoding = 0;
+    std::string line;
+    while (lineReader.getNextLine(&line)) {
+        if (!line.find("Content-Type:")) {
+            ASSERT_FALSE(expectedEncoding);
+            if (line.find("multipart/related;") != std::string::npos) {
+                // Skip this one, it's part of the MHTML header.
+                continue;
+            }
+            if (line.find("text/") != std::string::npos)
+                expectedEncoding = "quoted-printable";
+            else if (line.find("image/") != std::string::npos)
+                expectedEncoding = "base64";
+            else
+                FAIL() << "Unexpected Content-Type: " << line;
+            continue;
+        }
+        if (!line.find("Content-Transfer-Encoding:")) {
+            ASSERT_TRUE(expectedEncoding);
+            EXPECT_TRUE(line.find(expectedEncoding) != std::string::npos);

[hajimehoshi, 2015/06/10 09:02:03]

nit: EXPECT_NE

[Tiger (Sony Mobile), 2015/06/10 10:30:36]

Done.

+            expectedEncoding = 0;
+            sectionCheckedCount++;
+        }
+    }
+    EXPECT_EQ(12, sectionCheckedCount);
 }
+}