Merge page serializers [11/12]

Source/web/tests/PageSerializerTest.cpp

 /*
  * Copyright (c) 2013, Opera Software ASA. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
@@ skipping 63 matching lines @@
     virtual void TearDown()
     {
         Platform::current()->unitTestSupport()->unregisterAllMockedURLs();
     }
 
     void setBaseFolder(const char* folder)
     {
         m_folder = WebString::fromUTF8(folder);
     }
 
+    void setRewriteURLFolder(const char* folder)
+    {
+        m_rewriteFolder = folder;
+    }
+
     void registerURL(const char* url, const char* file, const char* mimeType)
     {
         registerMockedURLLoad(KURL(m_baseUrl, url), WebString::fromUTF8(file), m_folder, WebString::fromUTF8(mimeType));
     }
 
     void registerURL(const char* file, const char* mimeType)
     {
         registerURL(file, file, mimeType);
     }
 
     void registerErrorURL(const char* file, int statusCode)
     {
         WebURLError error;
         error.reason = 0xdead + statusCode;
         error.domain = "PageSerializerTest";
 
         WebURLResponse response;
         response.initialize();
         response.setMIMEType("text/html");
         response.setHTTPStatusCode(statusCode);
 
         Platform::current()->unitTestSupport()->registerMockedErrorURL(KURL(m_baseUrl, file), response, error);
     }
 
+    void registerRewriteURL(const char* fromURL, const char* toURL)
+    {
+        m_rewriteURLs.add(fromURL, toURL);
+    }
+
     void serialize(const char* url)
     {
         FrameTestHelpers::loadFrame(m_helper.webView()->mainFrame(), KURL(m_baseUrl, url).string().utf8().data());
         PageSerializer serializer(&m_resources, nullptr);
+
+        serializer.setRewriteURLFolder(m_rewriteFolder);
+        for (const auto& rewriteURL: m_rewriteURLs)
+            serializer.registerRewriteURL(rewriteURL.key, rewriteURL.value);
+
         serializer.serialize(m_helper.webViewImpl()->mainFrameImpl()->frame()->page());
     }
 
     Vector<SerializedResource>& getResources()
     {
         return m_resources;
     }
 
 
     const SerializedResource* getResource(const char* url, const char* mimeType)
@@ skipping 27 matching lines @@
     {
         settings->setImagesEnabled(true);
         settings->setLoadsImagesAutomatically(true);
         settings->setJavaScriptEnabled(true);
     }
 
     FrameTestHelpers::WebViewHelper m_helper;
     WebString m_folder;
     KURL m_baseUrl;
     Vector<SerializedResource> m_resources;
+    HashMap<String, String> m_rewriteURLs;
+    String m_rewriteFolder;
 };
 
 TEST_F(PageSerializerTest, HTMLElements)
 {
     setBaseFolder("pageserializer/elements/");
 
     registerURL("elements.html", "text/html");
     registerURL("style.css", "style.css", "text/css");
     registerURL("copyright.html", "text.txt", "text/html");
     registerURL("script.js", "text.txt", "text/javascript");
@@ skipping 215 matching lines @@
     setBaseFolder("pageserializer/datauri/");
 
     registerURL("page_with_morphing_data.html", "text/html");
 
     serialize("page_with_morphing_data.html");
 
     EXPECT_EQ(2U, getResources().size());
     EXPECT_TRUE(isSerialized("page_with_morphing_data.html", "text/html"));
 }
 
+TEST_F(PageSerializerTest, RewriteLinksSimple)
+{
+    setBaseFolder("pageserializer/rewritelinks/");
+    setRewriteURLFolder("folder");
+
+    registerURL("rewritelinks_simple.html", "text/html");
+    registerURL("absolute.png", "image.png", "image/png");
+    registerURL("relative.png", "image.png", "image/png");
+    registerRewriteURL("http://www.test.com/absolute.png", "a.png");
+    registerRewriteURL("http://www.test.com/relative.png", "b.png");
+
+    serialize("rewritelinks_simple.html");
+
+    EXPECT_EQ(3U, getResources().size());
+    EXPECT_NE(getSerializedData("rewritelinks_simple.html", "text/html").find("./folder/a.png"), kNotFound);
+    EXPECT_NE(getSerializedData("rewritelinks_simple.html", "text/html").find("./folder/b.png"), kNotFound);
+}
+
+TEST_F(PageSerializerTest, RewriteLinksBase)
+{
+    setBaseFolder("pageserializer/rewritelinks/");
+    setRewriteURLFolder("folder");
+
+    registerURL("rewritelinks_base.html", "text/html");
+    registerURL("images/here/image.png", "image.png", "image/png");
+    registerURL("images/here/or/in/here/image.png", "image.png", "image/png");
+    registerURL("or/absolute.png", "image.png", "image/png");
+    registerRewriteURL("http://www.test.com/images/here/image.png", "a.png");
+    registerRewriteURL("http://www.test.com/images/here/or/in/here/image.png", "b.png");
+    registerRewriteURL("http://www.test.com/or/absolute.png", "c.png");
+
+    serialize("rewritelinks_base.html");
+
+    EXPECT_EQ(4U, getResources().size());
+    EXPECT_NE(getSerializedData("rewritelinks_base.html", "text/html").find("./folder/a.png"), kNotFound);
+    EXPECT_NE(getSerializedData("rewritelinks_base.html", "text/html").find("./folder/b.png"), kNotFound);
+    EXPECT_NE(getSerializedData("rewritelinks_base.html", "text/html").find("./folder/c.png"), kNotFound);
+}
+
 // Test that we don't regress https://bugs.webkit.org/show_bug.cgi?id=99105
 TEST_F(PageSerializerTest, SVGImageDontCrash)
 {
     setBaseFolder("pageserializer/svg/");
 
     registerURL("page_with_svg_image.html", "text/html");
     registerURL("green_rectangle.svg", "image/svg+xml");
 
     serialize("page_with_svg_image.html");
 
@@ skipping 24 matching lines @@
     registerURL("namespace_element.html", "text/html");
 
     serialize("namespace_element.html");
 
     EXPECT_EQ(1U, getResources().size());
     EXPECT_TRUE(isSerialized("namespace_element.html", "text/html"));
     EXPECT_GT(getSerializedData("namespace_element.html", "text/html").length(), 0U);
 }
 
 }