Upgrade in-document custom elements when an element is defined.

third_party/WebKit/Source/core/dom/custom/CustomElementUpgradeSorterTest.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/custom/CustomElementUpgradeSorter.h"
 
 #include "bindings/core/v8/ExceptionState.h"
 #include "core/HTMLNames.h"
 #include "core/dom/Document.h"
 #include "core/dom/Element.h"
 #include "core/dom/shadow/ShadowRoot.h"
 #include "core/dom/shadow/ShadowRootInit.h"
 #include "core/html/HTMLDocument.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/heap/Handle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/OwnPtr.h"
 #include "wtf/text/AtomicString.h"
 
 namespace blink {
 
 class CustomElementUpgradeSorterTest : public ::testing::Test {
-public:
+protected:
+    void SetUp() override
+    {
+        m_page = DummyPageHolder::create(IntSize(1, 1));
+    }
+
+    void TearDown() override
+    {
+        m_page = nullptr;
+    }
+
     Element* createElementWithId(const char* localName, const char* id)
     {
         NonThrowableExceptionState noExceptions;
         Element* element =
             document()->createElement(localName, AtomicString(), noExceptions);
         element->setAttribute(HTMLNames::idAttr, id);
         return element;
     }
 
     Document* document() { return &m_page->document(); }
 
     ScriptState* scriptState()
     {
         return ScriptState::forMainWorld(&m_page->frame());
     }
 
     ShadowRoot* attachShadowTo(Element* element)
     {
         NonThrowableExceptionState noExceptions;
         ShadowRootInit shadowRootInit;
         return
             element->attachShadow(scriptState(), shadowRootInit, noExceptions);
     }
 
-protected:
-    void SetUp() override
-    {
-        m_page = DummyPageHolder::create(IntSize(1, 1));
-    }
-
-    void TearDown() override
-    {
-        m_page = nullptr;
-    }
-
 private:
     OwnPtr<DummyPageHolder> m_page;
 };
 
 TEST_F(CustomElementUpgradeSorterTest, inOtherDocument_notInSet)
 {
     NonThrowableExceptionState noExceptions;
     Element* element =
         document()->createElement("a-a", AtomicString(), noExceptions);
 
@@ skipping 150 matching lines @@
 
     HeapVector<Member<Element>> elements;
     sort.sorted(&elements, document());
     EXPECT_EQ(3u, elements.size());
     EXPECT_EQ(a, elements[0].get());
     EXPECT_EQ(c, elements[1].get());
     EXPECT_EQ(d, elements[2].get());
 }
 
 } // namespace blink