Save-Page-As-Complete-Html: Each frame links to a distinct local file.

third_party/WebKit/Source/web/tests/WebFrameSerializerTest.cpp

 /*
  * Copyright (C) 2011 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 61 matching lines @@
     {
         Platform::current()->unitTestSupport()->unregisterAllMockedURLs();
     }
 
     void registerMockedImageURL(const String& url)
     {
         // Image resources need to be mocked, but irrelevant here what image they map to.
         URLTestHelpers::registerMockedURLLoad(KURL(ParsedURLString, url), "frameserialization/awesome.png");
     }
 
+    class SingleLinkRewritingDelegate : public WebFrameSerializer::LinkRewritingDelegate {
+    public:
+        SingleLinkRewritingDelegate(const WebURL& url, const WebString& localPath)
+            : m_url(url)
+            , m_localPath(localPath)
+        {
+        }
+
+        bool rewriteFrameSource(WebFrame* frame, WebString* rewrittenLink) override
+        {
+            return false;
+        }
+
+        bool rewriteLink(const WebURL& url, WebString* rewrittenLink) override
+        {
+            if (url != m_url)
+                return false;
+
+            *rewrittenLink = m_localPath;
+            return true;
+        }
+
+    private:
+        const WebURL m_url;
+        const WebString m_localPath;
+    };
+
     String serializeFile(const String& url, const String& fileName)
     {
         KURL parsedURL(ParsedURLString, url);
         URLTestHelpers::registerMockedURLLoad(parsedURL, fileName, "frameserialization/", "text/html");
         FrameTestHelpers::loadFrame(mainFrameImpl(), url.utf8().data());
-        std::vector<std::pair<WebURL, WebString>> urlsToLocalPaths;
-        urlsToLocalPaths.push_back(std::make_pair(parsedURL, WebString("local")));
+        SingleLinkRewritingDelegate delegate(parsedURL, WebString("local"));
         SimpleWebFrameSerializerClient serializerClient;
-        WebFrameSerializer::serialize(mainFrameImpl(), &serializerClient, urlsToLocalPaths);
+        WebFrameSerializer::serialize(
+            mainFrameImpl(), &serializerClient, &delegate);
         return serializerClient.toString();
     }
 
     WebLocalFrameImpl* mainFrameImpl()
     {
         return m_helper.webViewImpl()->mainFrameImpl();
     }
 
 private:
     FrameTestHelpers::WebViewHelper m_helper;
@@ skipping 28 matching lines @@
     EXPECT_EQ(expectedHTML, actualHTML);
 }
 
 TEST_F(WebFrameSerializerTest, FromUrlWithMinusMinus)
 {
     String actualHTML = serializeFile("http://www.test.com?--x--", "text_only_page.html");
     EXPECT_EQ("<!-- saved from url=(0030)http://www.test.com/?-%2Dx-%2D -->", actualHTML.substring(1, 60));
 }
 
 } // namespace blink