Make Editor::tidyUpHTMLStructure() handles BODY element correctly

third_party/WebKit/Source/core/editing/EditorTest.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/editing/Editor.h"
+
+#include "core/dom/Document.h"
+#include "core/editing/EditingTestBase.h"
+#include "core/html/HTMLBodyElement.h"
+#include "core/html/HTMLDivElement.h"
+#include "core/html/HTMLHeadElement.h"
+#include "core/html/HTMLHtmlElement.h"
+
+namespace blink {
+
+class EditorTest : public EditingTestBase {
+protected:
+    void makeDocumentEmpty();
+};
+
+void EditorTest::makeDocumentEmpty()
+{
+    while (document().firstChild())
+        document().removeChild(document().firstChild());
+}
+
+TEST_F(EditorTest, tidyUpHTMLStructureFromBody)
+{
+    RefPtrWillBeRawPtr<Element> body = HTMLBodyElement::create(document());
+    makeDocumentEmpty();
+    document().setDesignMode("on");
+    document().appendChild(body);
+    Editor::tidyUpHTMLStructure(document());
+
+    EXPECT_TRUE(isHTMLHtmlElement(document().documentElement()));
+    EXPECT_EQ(body, document().body());
+    EXPECT_EQ(document().documentElement(), body->parentNode());
+}
+
+TEST_F(EditorTest, tidyUpHTMLStructureFromDiv)
+{
+    RefPtrWillBeRawPtr<Element> div = HTMLDivElement::create(document());
+    makeDocumentEmpty();
+    document().setDesignMode("on");
+    document().appendChild(div);
+    Editor::tidyUpHTMLStructure(document());
+
+    EXPECT_TRUE(isHTMLHtmlElement(document().documentElement()));
+    EXPECT_TRUE(isHTMLBodyElement(document().body()));
+    EXPECT_EQ(document().body(), div->parentNode());
+}
+
+TEST_F(EditorTest, tidyUpHTMLStructureFromHead)
+{
+    RefPtrWillBeRawPtr<Element> head = HTMLHeadElement::create(document());
+    makeDocumentEmpty();
+    document().setDesignMode("on");
+    document().appendChild(head);
+    Editor::tidyUpHTMLStructure(document());
+
+    EXPECT_TRUE(isHTMLHtmlElement(document().documentElement()));
+    EXPECT_TRUE(isHTMLBodyElement(document().body()));
+    EXPECT_EQ(document().documentElement(), head->parentNode());
+}
+
+} // namespace blink