Provide a method to remove inserted style sheet

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

Index: third_party/WebKit/Source/web/tests/WebDocumentTest.cpp
diff --git a/third_party/WebKit/Source/web/tests/WebDocumentTest.cpp b/third_party/WebKit/Source/web/tests/WebDocumentTest.cpp
index efa3c2171fb45a0f32c206efcdfc021fdde42221..50bed1a5f304937e263c160f72c1a6ed5ae0c747 100644
--- a/third_party/WebKit/Source/web/tests/WebDocumentTest.cpp
+++ b/third_party/WebKit/Source/web/tests/WebDocumentTest.cpp
@@ -60,7 +60,7 @@ 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();
@@ -68,7 +68,7 @@ TEST_F(WebDocumentTest, InsertStyleSheet) {
 
   unsigned start_count = core_doc->GetStyleEngine().StyleForElementCount();
 
-  web_doc.InsertStyleSheet("body { color: green }");
+  unsigned stylesheet_id = web_doc.InsertStyleSheet("body { color: green }");

[rune, 2017/05/18 13:13:21]

WebStyleSheetId?

[limasdf, 2017/05/18 14:32:26]

Done.

 
   // Check insertStyleSheet did not cause a synchronous style recalc.
   unsigned element_count =
@@ -93,6 +93,27 @@ TEST_F(WebDocumentTest, InsertStyleSheet) {
   // 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) {