Support serializing shadow DOM to MHTML

third_party/WebKit/Source/web/tests/WebFrameSerializerSanitizationTest.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 83 matching lines @@
  protected:
   WebFrameSerializerSanitizationTest() { helper_.Initialize(); }
 
   ~WebFrameSerializerSanitizationTest() override {
     Platform::Current()
         ->GetURLLoaderMockFactory()
         ->UnregisterAllURLsAndClearMemoryCache();
   }
 
   String GenerateMHTMLFromHtml(const String& url, const String& file_name) {
-    return GenerateMHTML(url, file_name, "text/html", false);
+    LoadFrame(url, file_name, "text/html");
+    return GenerateMHTML(false);
   }
 
   String GenerateMHTMLPartsFromPng(const String& url, const String& file_name) {
-    return GenerateMHTML(url, file_name, "image/png", true);
+    LoadFrame(url, file_name, "image/png");
+    return GenerateMHTML(true);
   }
 
-  String GenerateMHTML(const String& url,
-                       const String& file_name,
-                       const String& mime_type,
-                       const bool only_body_parts) {
+  void LoadFrame(const String& url,
+                 const String& file_name,
+                 const String& mime_type) {
     KURL parsed_url(kParsedURLString, url);
     String file_path("frameserialization/" + file_name);
     RegisterMockedFileURLLoad(parsed_url, file_path, mime_type);
     FrameTestHelpers::LoadFrame(MainFrameImpl(), url.Utf8().data());
+    MainFrameImpl()->GetFrame()->View()->UpdateAllLifecyclePhases();
+  }
+
+  String GenerateMHTML(const bool only_body_parts) {
     // Boundaries are normally randomly generated but this one is predefined for
     // simplicity and as good as any other. Plus it gets used in almost all the
     // examples in the MHTML spec - RFC 2557.
     const WebString boundary("boundary-example");
     StringBuilder mhtml;
     if (!only_body_parts) {
       WebThreadSafeData header_result = WebFrameSerializer::GenerateMHTMLHeader(
           boundary, MainFrameImpl(), &mhtml_delegate_);
       mhtml.Append(header_result.Data(), header_result.size());
     }
@@ skipping 13 matching lines @@
     if (!only_body_parts) {
       // Validate the generated MHTML.
       MHTMLParser parser(SharedBuffer::Create(mhtml_string.Characters8(),
                                               size_t(mhtml_string.length())));
       EXPECT_FALSE(parser.ParseArchive().IsEmpty())
           << "Generated MHTML is not well formed";
     }
     return mhtml_string;
   }
 
+  ShadowRoot* SetShadowContent(TreeScope& scope,
+                               const char* host,
+                               ShadowRootType shadow_type,
+                               const char* shadow_content,
+                               bool delegates_focus = false) {
+    ShadowRoot* shadow_root =
+        scope.getElementById(AtomicString::FromUTF8(host))
+            ->CreateShadowRootInternal(shadow_type, ASSERT_NO_EXCEPTION);
+    shadow_root->SetDelegatesFocus(delegates_focus);
+    shadow_root->setInnerHTML(String::FromUTF8(shadow_content),
+                              ASSERT_NO_EXCEPTION);
+    scope.GetDocument().View()->UpdateAllLifecyclePhases();
+    return shadow_root;
+  }
+
   void SetRemovePopupOverlay(bool remove_popup_overlay) {
     mhtml_delegate_.SetRemovePopupOverlay(remove_popup_overlay);
   }
 
   void RegisterMockedFileURLLoad(const KURL& url,
                                  const String& file_path,
                                  const String& mime_type = "image/png") {
     URLTestHelpers::RegisterMockedURLLoad(
         url, testing::WebTestDataPath(file_path.Utf8().data()), mime_type);
   }
@@ skipping 177 matching lines @@
   EXPECT_EQ(WTF::kNotFound,
             mhtml.Find("<meta http-equiv=3D\"Content-Security-Policy"));
   EXPECT_NE(WTF::kNotFound, mhtml.Find("<meta name=3D\"description"));
   EXPECT_NE(WTF::kNotFound, mhtml.Find("<meta http-equiv=3D\"refresh"));
 
   // If an element is removed, its children should also be skipped.
   EXPECT_EQ(WTF::kNotFound, mhtml.Find("<select"));
   EXPECT_EQ(WTF::kNotFound, mhtml.Find("<option"));
 }
 
+TEST_F(WebFrameSerializerSanitizationTest, ShadowDOM) {
+  LoadFrame("http://www.test.com", "shadow_dom.html", "text/html");
+  Document* document = MainFrameImpl()->GetFrame()->GetDocument();
+  SetShadowContent(*document, "h1", ShadowRootType::V0, "V0 shadow");
+  ShadowRoot* shadowRoot =
+      SetShadowContent(*document, "h2", ShadowRootType::kOpen,
+                       "Parent shadow\n<p id=\"h3\">Foo</p>", true);
+  SetShadowContent(*shadowRoot, "h3", ShadowRootType::kClosed, "Nested shadow");
+  String mhtml = GenerateMHTML(false);
+
+  // Template with special attribute should be created for each shadow DOM tree.
+  EXPECT_NE(WTF::kNotFound, mhtml.Find("<template shadowmode=3D\"v0\">"));
+  EXPECT_NE(WTF::kNotFound,
+            mhtml.Find("<template shadowmode=3D\"open\" shadowdelegatesfocus"));
+  EXPECT_NE(WTF::kNotFound, mhtml.Find("<template shadowmode=3D\"closed\">"));
+
+  // The special attribute present in the original page should be removed.
+  EXPECT_EQ(WTF::kNotFound, mhtml.Find("<template shadowmode=3D\"foo\">"));
+}
+
 }  // namespace blink