Remove unnecessary uses of HTMLDocument

third_party/WebKit/Source/core/dom/shadow/FlatTreeTraversalTest.cpp

 // Copyright 2015 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/shadow/FlatTreeTraversal.h"
 
 #include "bindings/core/v8/ExceptionStatePlaceholder.h"
 #include "core/dom/Document.h"
 #include "core/dom/Element.h"
 #include "core/dom/Node.h"
 #include "core/dom/NodeTraversal.h"
 #include "core/dom/shadow/ShadowRoot.h"
 #include "core/frame/FrameView.h"
-#include "core/html/HTMLDocument.h"
 #include "core/html/HTMLElement.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/geometry/IntSize.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/Compiler.h"
 #include "wtf/StdLibExtras.h"
 #include "wtf/Vector.h"
 #include <memory>
 
 namespace blink {
 
 class FlatTreeTraversalTest : public ::testing::Test {
 protected:
-    HTMLDocument& document() const;
+    Document& document() const;
 
     // Sets |mainHTML| to BODY element with |innerHTML| property and attaches
     // shadow root to child with |shadowHTML|, then update distribution for
     // calling member functions in |FlatTreeTraversal|.
     void setupSampleHTML(const char* mainHTML, const char* shadowHTML, unsigned);
 
     void setupDocumentTree(const char* mainHTML);
 
     void attachV0ShadowRoot(Element& shadowHost, const char* shadowInnerHTML);
     void attachOpenShadowRoot(Element& shadowHost, const char* shadowInnerHTML);
 
 private:
     void SetUp() override;
 
-    Persistent<HTMLDocument> m_document;
+    Persistent<Document> m_document;
     std::unique_ptr<DummyPageHolder> m_dummyPageHolder;
 };
 
 void FlatTreeTraversalTest::SetUp()
 {
     m_dummyPageHolder = DummyPageHolder::create(IntSize(800, 600));
-    m_document = toHTMLDocument(&m_dummyPageHolder->document());
+    m_document = &m_dummyPageHolder->document();
     DCHECK(m_document);
 }
 
-HTMLDocument& FlatTreeTraversalTest::document() const
+Document& FlatTreeTraversalTest::document() const
 {
     return *m_document;
 }
 
 void FlatTreeTraversalTest::setupSampleHTML(const char* mainHTML, const char* shadowHTML, unsigned index)
 {
     Element* body = document().body();
     body->setInnerHTML(String::fromUTF8(mainHTML), ASSERT_NO_EXCEPTION);
     Element* shadowHost = toElement(NodeTraversal::childAt(*body, index));
     ShadowRoot* shadowRoot = shadowHost->createShadowRootInternal(ShadowRootType::V0, ASSERT_NO_EXCEPTION);
@@ skipping 497 matching lines @@
 
     EXPECT_EQ(d11, FlatTreeTraversal::parent(*d111));
     EXPECT_EQ(d11, FlatTreeTraversal::parent(*d112));
     EXPECT_EQ(d11, FlatTreeTraversal::parent(*d2));
     EXPECT_EQ(d11, FlatTreeTraversal::parent(*d14));
     EXPECT_EQ(nullptr, FlatTreeTraversal::parent(*d3));
     EXPECT_EQ(nullptr, FlatTreeTraversal::parent(*d4));
 }
 
 } // namespace blink