Free HashSets on setWholeSubtreeInvalid.

Source/core/css/analyzer/DescendantInvalidationSetTest.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 "config.h"
+#include "core/css/analyzer/DescendantInvalidationSet.h"
+
+#include <gtest/gtest.h>
+
+using namespace WebCore;
+
+namespace {
+
+// Once we setWholeSubtreeInvalid, we should not keep the HashSets.
+TEST(DescendantInvalidationSetTest, SubtreeInvalid_AddBefore)
+{
+    RefPtr<DescendantInvalidationSet> set = DescendantInvalidationSet::create();
+    set->addClass("a");
+    set->setWholeSubtreeInvalid();
+
+    Vector<AtomicString> classes;
+    set->getClasses(classes);
+
+    ASSERT_TRUE(classes.isEmpty());
+}
+
+// Don't (re)create HashSets if we've already setWholeSubtreeInvalid.
+TEST(DescendantInvalidationSetTest, SubtreeInvalid_AddAfter)
+{
+    RefPtr<DescendantInvalidationSet> set = DescendantInvalidationSet::create();
+    set->setWholeSubtreeInvalid();
+    set->addClass("a");
+
+    Vector<AtomicString> classes;
+    set->getClasses(classes);
+
+    ASSERT_TRUE(classes.isEmpty());
+}
+
+// No need to keep the HashSets when combining with a wholeSubtreeInvalid set.
+TEST(DescendantInvalidationSetTest, SubtreeInvalid_Combine_1)
+{
+    RefPtr<DescendantInvalidationSet> set1 = DescendantInvalidationSet::create();
+    RefPtr<DescendantInvalidationSet> set2 = DescendantInvalidationSet::create();
+
+    set1->addClass("a");
+    set2->setWholeSubtreeInvalid();
+
+    set1->combine(*set2);
+
+    Vector<AtomicString> classes;
+    set1->getClasses(classes);
+
+    ASSERT_TRUE(set1->wholeSubtreeInvalid());
+    ASSERT_TRUE(classes.isEmpty());
+}
+
+// No need to add HashSets from combining set when we already have wholeSubtreeInvalid.
+TEST(DescendantInvalidationSetTest, SubtreeInvalid_Combine_2)
+{
+    RefPtr<DescendantInvalidationSet> set1 = DescendantInvalidationSet::create();
+    RefPtr<DescendantInvalidationSet> set2 = DescendantInvalidationSet::create();
+
+    set1->setWholeSubtreeInvalid();
+    set2->addClass("a");
+
+    set1->combine(*set2);
+
+    Vector<AtomicString> classes;
+    set1->getClasses(classes);
+
+    ASSERT_TRUE(set1->wholeSubtreeInvalid());
+    ASSERT_TRUE(classes.isEmpty());
+}
+
+} // namespace