Sibling invalidation sets

Source/core/css/RuleFeatureSetTest.cpp

+
+#include "config.h"
+#include "core/css/RuleFeature.h"
+
+#include "core/css/CSSSelectorList.h"
+#include "core/css/RuleSet.h"
+#include "core/css/StylePropertySet.h"
+#include "core/css/StyleRule.h"
+#include "core/css/parser/CSSParser.h"
+#include "core/dom/ElementTraversal.h"
+#include "core/html/HTMLBodyElement.h"
+#include "core/html/HTMLDocument.h"
+#include "core/html/HTMLElement.h"
+#include "core/html/HTMLHtmlElement.h"
+#include <gtest/gtest.h>
+
+namespace blink {
+
+class RuleFeatureSetTest : public ::testing::Test {
+public:
+    RuleFeatureSetTest()
+    {
+    }
+
+    void SetUp()
+    {
+        m_document = HTMLDocument::create();
+        RefPtrWillBeRawPtr<HTMLHtmlElement> html = HTMLHtmlElement::create(*m_document);
+        html->appendChild(HTMLBodyElement::create(*m_document));
+        m_document->appendChild(html.release());
+
+        m_document->body()->setInnerHTML("<b><i></i></b>", ASSERT_NO_EXCEPTION);
+    }
+
+    void updateInvalidationSets(const String& selectorText)
+    {
+        CSSSelectorList selectorList;
+        CSSParser::parseSelector(strictCSSParserContext(), selectorText, selectorList);
+
+        RefPtrWillBeRawPtr<StyleRule> styleRule = StyleRule::create(selectorList, MutableStylePropertySet::create(HTMLStandardMode));
+        RuleData ruleData(styleRule.get(), 0, 0, RuleHasNoSpecialState);
+        m_ruleFeatureSet.updateInvalidationSets(ruleData);
+    }
+
+    void collectInvalidationSetsForClass(InvalidationSetVector& descendant, InvalidationSetVector& sibling, const AtomicString& className) const
+    {
+        Element* element = Traversal<HTMLElement>::firstChild(*Traversal<HTMLElement>::firstChild(*m_document->body()));
+        m_ruleFeatureSet.collectInvalidationSetsForClass(descendant, sibling, *element, className);
+    }
+
+    WillBeHeapHashSet<AtomicString>& ensureClassSet(DescendantInvalidationSet& invalidationSet)
+    {
+        return invalidationSet.ensureClassSet();
+    }
+
+    void expectNoInvalidation(InvalidationSetVector& invalidationSets)
+    {
+        EXPECT_EQ(0u, invalidationSets.size());
+    }
+
+    void expectSelfInvalidation(InvalidationSetVector& invalidationSets)
+    {
+        EXPECT_EQ(1u, invalidationSets.size());
+        EXPECT_TRUE(invalidationSets[0]->isEmpty());
+    }
+
+    void expectClassInvalidation(const AtomicString& className, InvalidationSetVector& invalidationSets)
+    {
+        EXPECT_EQ(1u, invalidationSets.size());
+        WillBeHeapHashSet<AtomicString> classes = ensureClassSet(*invalidationSets[0]);
+        EXPECT_EQ(1u, classes.size());
+        EXPECT_TRUE(classes.contains(className));
+    }
+
+    void expectSiblingInvalidation(unsigned maxDirectAdjacentSelectors, const AtomicString& siblingName, InvalidationSetVector& invalidationSets)
+    {
+        EXPECT_EQ(1u, invalidationSets.size());
+        WillBeHeapHashSet<AtomicString> classes = ensureClassSet(*invalidationSets[0]);
+        EXPECT_EQ(1u, classes.size());
+        EXPECT_TRUE(classes.contains(siblingName));
+        EXPECT_EQ(maxDirectAdjacentSelectors, invalidationSets[0]->maxDirectAdjacentSelectors());
+        EXPECT_TRUE(invalidationSets[0]->siblingInvalid());
+        EXPECT_EQ(nullptr, invalidationSets[0]->descendants());
+    }
+
+    void expectSiblingDescendantInvalidation(unsigned maxDirectAdjacentSelectors, const AtomicString& siblingName, const AtomicString& descendantName, InvalidationSetVector& invalidationSets)
+    {
+        EXPECT_EQ(1u, invalidationSets.size());
+        WillBeHeapHashSet<AtomicString> classes = ensureClassSet(*invalidationSets[0]);
+        EXPECT_EQ(1u, classes.size());
+        EXPECT_TRUE(classes.contains(siblingName));
+        EXPECT_EQ(maxDirectAdjacentSelectors, invalidationSets[0]->maxDirectAdjacentSelectors());
+        EXPECT_FALSE(invalidationSets[0]->siblingInvalid());
+
+        WillBeHeapHashSet<AtomicString> descendantClasses = ensureClassSet(invalidationSets[0]->ensureDescendantInvalidationSet());
+        EXPECT_EQ(1u, descendantClasses.size());
+        EXPECT_TRUE(descendantClasses.contains(descendantName));
+    }
+
+    void expectClassesInvalidation(const AtomicString& firstClassName, const AtomicString& secondClassName, InvalidationSetVector& invalidationSets)
+    {
+        EXPECT_EQ(1u, invalidationSets.size());
+        WillBeHeapHashSet<AtomicString> classes = ensureClassSet(*invalidationSets[0]);
+        EXPECT_EQ(2u, classes.size());
+        EXPECT_TRUE(classes.contains(firstClassName));
+        EXPECT_TRUE(classes.contains(secondClassName));
+    }
+
+    DEFINE_INLINE_TRACE()
+    {
+#if ENABLE(OILPAN)
+        visitor->trace(m_ruleFeatureSet);
+        visitor->trace(m_document);
+#endif
+    }
+
+private:
+    RuleFeatureSet m_ruleFeatureSet;
+    RefPtrWillBePersistent<Document> m_document;
+};
+
+TEST_F(RuleFeatureSetTest, interleavedDescendantSibling)
+{
+    updateInvalidationSets(".k > .l ~ .m + .n .o + .p");
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "k");
+        expectClassInvalidation("p", descendant);
+        expectNoInvalidation(sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "l");
+        expectNoInvalidation(descendant);
+        expectSiblingDescendantInvalidation(std::numeric_limits<unsigned>::max(), "n", "p", sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "m");
+        expectNoInvalidation(descendant);
+        expectSiblingDescendantInvalidation(1, "n", "p", sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "n");
+        expectClassInvalidation("p", descendant);
+        expectNoInvalidation(sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "o");
+        expectNoInvalidation(descendant);
+        expectSiblingInvalidation(1, "p", sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "p");
+        expectSelfInvalidation(descendant);
+        expectNoInvalidation(sibling);
+    }
+}
+
+TEST_F(RuleFeatureSetTest, anySibling)
+{
+    updateInvalidationSets(":-webkit-any(.q, .r) ~ .s .t");
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "q");
+        expectNoInvalidation(descendant);
+        expectSiblingDescendantInvalidation(std::numeric_limits<unsigned>::max(), "s", "t", sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "r");
+        expectNoInvalidation(descendant);
+        expectSiblingDescendantInvalidation(std::numeric_limits<unsigned>::max(), "s", "t", sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "s");
+        expectClassInvalidation("t", descendant);
+        expectNoInvalidation(sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "t");
+        expectSelfInvalidation(descendant);
+        expectNoInvalidation(sibling);
+    }
+}
+
+TEST_F(RuleFeatureSetTest, siblingAny)
+{
+    updateInvalidationSets(".u .v ~ :-webkit-any(.w, .x)");
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "u");
+        expectClassesInvalidation("w", "x", descendant);
+        expectNoInvalidation(sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "v");
+        expectNoInvalidation(descendant);
+        expectClassesInvalidation("w", "x", sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "w");
+        expectSelfInvalidation(descendant);
+        expectNoInvalidation(sibling);
+    }
+    {
+        InvalidationSetVector descendant;
+        InvalidationSetVector sibling;
+        collectInvalidationSetsForClass(descendant, sibling, "x");
+        expectSelfInvalidation(descendant);
+        expectNoInvalidation(sibling);
+    }
+}
+
+} // namespace blink