Provide a method to remove inserted style sheet

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

 // Copyright 2014 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 "public/web/WebDocument.h"
 
 #include "core/CSSPropertyNames.h"
 #include "core/HTMLNames.h"
 #include "core/dom/NodeComputedStyle.h"
 #include "core/dom/StyleEngine.h"
@@ skipping 42 matching lines @@
 
 Document* WebDocumentTest::TopDocument() const {
   return ToLocalFrame(web_view_helper_.WebView()->GetPage()->MainFrame())
       ->GetDocument();
 }
 
 WebDocument WebDocumentTest::TopWebDocument() const {
   return web_view_helper_.WebView()->MainFrame()->GetDocument();
 }
 
-TEST_F(WebDocumentTest, InsertStyleSheet) {
+TEST_F(WebDocumentTest, InsertAndRemoveStyleSheet) {
   LoadURL("about:blank");
 
   WebDocument web_doc = TopWebDocument();
   Document* core_doc = TopDocument();
 
   unsigned start_count = core_doc->GetStyleEngine().StyleForElementCount();
 
-  web_doc.InsertStyleSheet("body { color: green }");
+  WebStyleSheetId stylesheet_id =
+      web_doc.InsertStyleSheet("body { color: green }");
 
   // Check insertStyleSheet did not cause a synchronous style recalc.
   unsigned element_count =
       core_doc->GetStyleEngine().StyleForElementCount() - start_count;
   ASSERT_EQ(0U, element_count);
 
   HTMLElement* body_element = core_doc->body();
   DCHECK(body_element);
 
   const ComputedStyle& style_before_insertion =
       body_element->ComputedStyleRef();
 
   // Inserted stylesheet not yet applied.
   ASSERT_EQ(Color(0, 0, 0),
             style_before_insertion.VisitedDependentColor(CSSPropertyColor));
 
   // Apply inserted stylesheet.
   core_doc->UpdateStyleAndLayoutTree();
 
   const ComputedStyle& style_after_insertion = body_element->ComputedStyleRef();
 
   // Inserted stylesheet applied.
   ASSERT_EQ(Color(0, 128, 0),
             style_after_insertion.VisitedDependentColor(CSSPropertyColor));
+
+  start_count = core_doc->GetStyleEngine().StyleForElementCount();
+
+  // Check RemoveInsertedStyleSheet did not cause a synchronous style recalc.
+  web_doc.RemoveInsertedStyleSheet(stylesheet_id);
+  element_count =
+      core_doc->GetStyleEngine().StyleForElementCount() - start_count;
+  ASSERT_EQ(0U, element_count);
+
+  const ComputedStyle& style_before_removing = body_element->ComputedStyleRef();
+
+  // Removed stylesheet not yet applied.
+  ASSERT_EQ(Color(0, 128, 0),
+            style_before_removing.VisitedDependentColor(CSSPropertyColor));
+
+  // Apply removed stylesheet.
+  core_doc->UpdateStyleAndLayoutTree();
+
+  const ComputedStyle& style_after_removing = body_element->ComputedStyleRef();
+  ASSERT_EQ(Color(0, 0, 0),
+            style_after_removing.VisitedDependentColor(CSSPropertyColor));
 }
 
 TEST_F(WebDocumentTest, ManifestURL) {
   LoadURL(std::string(kDefaultOrigin) + kManifestDummyFilePath);
 
   WebDocument web_doc = TopWebDocument();
   Document* document = TopDocument();
   HTMLLinkElement* link_manifest = document->LinkManifest();
 
   // No href attribute was set.
@@ skipping 248 matching lines @@
 
   ASSERT_EQ(ToOriginA(g_nested_origin_a_in_origin_b),
             TopDocument()->FirstPartyForCookies());
   ASSERT_EQ(ToOriginA(g_nested_origin_a_in_origin_b),
             NestedDocument()->FirstPartyForCookies());
   ASSERT_EQ(ToOriginA(g_nested_origin_a_in_origin_b),
             NestedNestedDocument()->FirstPartyForCookies());
 }
 
 }  // namespace blink