Implemented scoped invalidation for added/removed RuleSets.

third_party/WebKit/Source/core/dom/StyleEngineTest.cpp

 // Copyright 2016 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 "core/dom/StyleEngine.h"
 
 #include "core/css/StyleSheetContents.h"
 #include "core/dom/Document.h"
 #include "core/dom/NodeComputedStyle.h"
+#include "core/dom/shadow/ShadowRootInit.h"
 #include "core/frame/FrameView.h"
 #include "core/html/HTMLElement.h"
 #include "core/html/HTMLStyleElement.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/heap/Heap.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include <memory>
 
 namespace blink {
 
 class StyleEngineTest : public ::testing::Test {
 protected:
     void SetUp() override;
 
     Document& document() { return m_dummyPageHolder->document(); }
     StyleEngine& styleEngine() { return document().styleEngine(); }
 
     bool isDocumentStyleSheetCollectionClean() { return !styleEngine().shouldUpdateDocumentStyleSheetCollection(AnalyzedStyleUpdate); }
 
+    void scheduleInvalidationsForRules(TreeScope&, const String& cssText);
+
 private:
     std::unique_ptr<DummyPageHolder> m_dummyPageHolder;
 };
 
 void StyleEngineTest::SetUp()
 {
     m_dummyPageHolder = DummyPageHolder::create(IntSize(800, 600));
 }
 
+void StyleEngineTest::scheduleInvalidationsForRules(TreeScope& treeScope, const String& cssText)
+{
+    StyleSheetContents* sheet = StyleSheetContents::create(CSSParserContext(HTMLStandardMode, nullptr));
+    sheet->parseString(cssText);
+    HeapVector<Member<const RuleSet>> ruleSets;
+    RuleSet& ruleSet = sheet->ensureRuleSet(MediaQueryEvaluator(), RuleHasDocumentSecurityOrigin);
+    ruleSet.compactRulesIfNeeded();
+    ruleSets.append(&ruleSet);
+    styleEngine().scheduleInvalidationsForRuleSets(treeScope, ruleSets);
+}
+
 TEST_F(StyleEngineTest, DocumentDirtyAfterInject)
 {
     StyleSheetContents* parsedSheet = StyleSheetContents::create(CSSParserContext(document(), nullptr));
     parsedSheet->parseString("div {}");
     styleEngine().injectAuthorSheet(parsedSheet);
     document().view()->updateAllLifecyclePhases();
 
     EXPECT_TRUE(isDocumentStyleSheetCollectionClean());
 }
 
@@ skipping 47 matching lines @@
     // Garbage collection should clear the weak reference in the StyleSheetContents cache.
     ThreadState::current()-> collectAllGarbage();
 
     element = HTMLStyleElement::create(document(), false);
     sheet1 = styleEngine().createSheet(element, sheetText, minPos, context);
 
     // Check that we did not use a cached StyleSheetContents after the garbage collection.
     EXPECT_FALSE(sheet1->contents()->isUsedFromTextCache());
 }
 
+TEST_F(StyleEngineTest, RuleSetInvalidationTypeSelectors)
+{
+    document().body()->setInnerHTML(
+        "<div>"
+        "  <span></span>"
+        "  <div></div>"
+        "</div>", ASSERT_NO_EXCEPTION);
+
+    document().view()->updateAllLifecyclePhases();
+
+    unsigned beforeCount = styleEngine().styleForElementCount();
+
+    scheduleInvalidationsForRules(document(), "span { background: green}");
+    document().view()->updateAllLifecyclePhases();
+
+    unsigned afterCount = styleEngine().styleForElementCount();
+
+    EXPECT_EQ(1u, afterCount - beforeCount);
+
+    beforeCount = afterCount;
+    scheduleInvalidationsForRules(document(), "body div { background: green}");
+    document().view()->updateAllLifecyclePhases();
+
+    afterCount = styleEngine().styleForElementCount();
+
+    EXPECT_EQ(2u, afterCount - beforeCount);
+}
+
+TEST_F(StyleEngineTest, RuleSetInvalidationHost)
+{
+    document().body()->setInnerHTML("<div id=nohost></div><div id=host></div>", ASSERT_NO_EXCEPTION);
+    Element* host = document().getElementById("host");
+    ASSERT_TRUE(host);
+
+    ShadowRootInit init;
+    init.setMode("open");
+    ShadowRoot* shadowRoot = host->attachShadow(ScriptState::forMainWorld(document().frame()), init, ASSERT_NO_EXCEPTION);
+    ASSERT_TRUE(shadowRoot);
+
+    shadowRoot->setInnerHTML("<div></div><div></div><div></div>", ASSERT_NO_EXCEPTION);
+    document().view()->updateAllLifecyclePhases();
+
+    unsigned beforeCount = styleEngine().styleForElementCount();
+    scheduleInvalidationsForRules(*shadowRoot, ":host(#nohost), #nohost { background: green}");
+    document().view()->updateAllLifecyclePhases();
+    unsigned afterCount = styleEngine().styleForElementCount();
+    EXPECT_EQ(0u, afterCount - beforeCount);
+
+    beforeCount = afterCount;
+    scheduleInvalidationsForRules(*shadowRoot, ":host(#host) { background: green}");
+    document().view()->updateAllLifecyclePhases();
+    afterCount = styleEngine().styleForElementCount();
+    EXPECT_EQ(1u, afterCount - beforeCount);
+}
+
+TEST_F(StyleEngineTest, RuleSetInvalidationSlotted)
+{
+    document().body()->setInnerHTML(
+        "<div id=host>"
+        "  <span slot=other class=s1></span>"
+        "  <span class=s2></span>"
+        "  <span class=s1></span>"
+        "  <span></span>"
+        "</div>", ASSERT_NO_EXCEPTION);
+
+    Element* host = document().getElementById("host");
+    ASSERT_TRUE(host);
+
+    ShadowRootInit init;
+    init.setMode("open");
+    ShadowRoot* shadowRoot = host->attachShadow(ScriptState::forMainWorld(document().frame()), init, ASSERT_NO_EXCEPTION);
+    ASSERT_TRUE(shadowRoot);
+
+    shadowRoot->setInnerHTML("<slot name=other></slot><slot></slot>", ASSERT_NO_EXCEPTION);
+    document().view()->updateAllLifecyclePhases();
+
+    unsigned beforeCount = styleEngine().styleForElementCount();
+    scheduleInvalidationsForRules(*shadowRoot, "::slotted(.s1) { background: green}");
+    document().view()->updateAllLifecyclePhases();
+    unsigned afterCount = styleEngine().styleForElementCount();
+    EXPECT_EQ(2u, afterCount - beforeCount);
+}
+
 } // namespace blink