1 files changed, 217 insertions, 0 deletions
diff --git a/core/fxcrt/xml/cfx_xmlelement_unittest.cpp b/core/fxcrt/xml/cfx_xmlelement_unittest.cpp
new file mode 100644
index 0000000000..79e067f511
--- /dev/null
+++ b/core/fxcrt/xml/cfx_xmlelement_unittest.cpp
@@ -0,0 +1,217 @@
+// Copyright 2018 PDFium 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 <utility>
+
+#include "core/fxcrt/xml/cfx_xmlchardata.h"
+#include "core/fxcrt/xml/cfx_xmlelement.h"
+#include "core/fxcrt/xml/cfx_xmltext.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/string_write_stream.h"
+#include "testing/test_support.h"
+#include "third_party/base/ptr_util.h"
+
+TEST(CFX_XMLElementTest, GetType) {
+  CFX_XMLElement node(L"node");
+  EXPECT_EQ(FX_XMLNODE_Element, node.GetType());
+}
+
+TEST(CFX_XMLElementTest, GetName) {
+  CFX_XMLElement node(L"node");
+  EXPECT_EQ(L"node", node.GetName());
+}
+
+TEST(CFX_XMLElementTest, GetLocalTagName) {
+  CFX_XMLElement node1(L"node1");
+  EXPECT_EQ(L"node1", node1.GetLocalTagName());
+
+  CFX_XMLElement node2(L"test:node2");
+  EXPECT_EQ(L"node2", node2.GetLocalTagName());
+}
+
+TEST(CFX_XMLElementTest, GetNamespacePrefix) {
+  CFX_XMLElement node1(L"node1");
+  EXPECT_EQ(L"", node1.GetNamespacePrefix());
+
+  CFX_XMLElement node2(L"test:node2");
+  EXPECT_EQ(L"test", node2.GetNamespacePrefix());
+}
+
+TEST(CFX_XMLElementTest, GetNamespaceURI) {
+  CFX_XMLElement node1(L"node1");
+  EXPECT_EQ(L"", node1.GetNamespaceURI());
+
+  node1.SetAttribute(L"xmlns", L"https://example.org/ns1");
+  EXPECT_EQ(L"https://example.org/ns1", node1.GetNamespaceURI());
+
+  CFX_XMLElement node2(L"test:node2");
+  EXPECT_EQ(L"", node2.GetNamespaceURI());
+
+  node2.SetAttribute(L"xmlns", L"https://example.org/ns2");
+  EXPECT_EQ(L"", node2.GetNamespaceURI());
+
+  node2.SetAttribute(L"xmlns:test", L"https://example.org/ns2");
+  EXPECT_EQ(L"https://example.org/ns2", node2.GetNamespaceURI());
+}
+
+TEST(CFX_XMLElementTest, Attributes) {
+  CFX_XMLElement node(L"test:node");
+  node.SetAttribute(L"first", L"one");
+  node.SetAttribute(L"second", L"two");
+
+  ASSERT_TRUE(node.HasAttribute(L"first"));
+  EXPECT_EQ(L"one", node.GetAttribute(L"first"));
+  ASSERT_TRUE(node.HasAttribute(L"second"));
+  EXPECT_EQ(L"two", node.GetAttribute(L"second"));
+
+  ASSERT_EQ(2U, node.GetAttributes().size());
+
+  node.RemoveAttribute(L"first");
+  EXPECT_FALSE(node.HasAttribute(L"first"));
+
+  ASSERT_EQ(1U, node.GetAttributes().size());
+}
+
+TEST(CFX_XMLElementTest, Clone) {
+  CFX_XMLElement node(L"test:node");
+  node.SetAttribute(L"first", L"one");
+  node.SetAttribute(L"second", L"two");
+  node.SetAttribute(L"xmlns:test", L"https://example.org/test");
+
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLText>(L"Text Child"));
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLElement>(L"Node child"));
+
+  auto clone = node.Clone();
+  EXPECT_TRUE(clone != nullptr);
+
+  ASSERT_EQ(FX_XMLNODE_Element, clone->GetType());
+  CFX_XMLElement* inst = static_cast<CFX_XMLElement*>(clone.get());
+
+  EXPECT_EQ(L"test:node", inst->GetName());
+  EXPECT_EQ(L"node", inst->GetLocalTagName());
+  EXPECT_EQ(L"test", inst->GetNamespacePrefix());
+  EXPECT_EQ(L"https://example.org/test", inst->GetNamespaceURI());
+
+  ASSERT_TRUE(inst->HasAttribute(L"first"));
+  EXPECT_EQ(L"one", inst->GetAttribute(L"first"));
+  ASSERT_TRUE(inst->HasAttribute(L"second"));
+  EXPECT_EQ(L"two", inst->GetAttribute(L"second"));
+
+  // Only clone the Text node, so expect only one child.
+  ASSERT_TRUE(inst->GetFirstChild() != nullptr);
+  EXPECT_TRUE(inst->GetFirstChild()->GetNextSibling() == nullptr);
+
+  ASSERT_EQ(FX_XMLNODE_Text, inst->GetFirstChild()->GetType());
+  auto* text = static_cast<CFX_XMLText*>(inst->GetFirstChild());
+  EXPECT_EQ(L"Text Child", text->GetText());
+}
+
+TEST(CFX_XMLElementTest, Save) {
+  auto stream = pdfium::MakeRetain<StringWriteStream>();
+  CFX_XMLElement node(L"root");
+
+  node.Save(stream);
+  EXPECT_EQ("<root />\n", stream->ToString());
+}
+
+TEST(CFX_XMLElementTest, SaveWithAttributes) {
+  auto stream = pdfium::MakeRetain<StringWriteStream>();
+  CFX_XMLElement node(L"root");
+  node.SetAttribute(L"first", L"one");
+  node.SetAttribute(L"second", L"two");
+
+  node.Save(stream);
+  EXPECT_EQ("<root first=\"one\" second=\"two\" />\n", stream->ToString());
+}
+
+TEST(CFX_XMLElementTest, SaveWithChildren) {
+  auto stream = pdfium::MakeRetain<StringWriteStream>();
+  CFX_XMLElement node(L"node");
+
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLText>(L"Text Child 1"));
+
+  auto child = pdfium::MakeUnique<CFX_XMLElement>(L"node-child");
+  CFX_XMLElement* node_child1 = child.get();
+  node.AppendChild(std::move(child));
+
+  node_child1->AppendChild(pdfium::MakeUnique<CFX_XMLText>(L"Text Child 2"));
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLCharData>(L"Char Data"));
+
+  node.Save(stream);
+  EXPECT_EQ(
+      "<node>\n"
+      "Text Child 1"
+      "<node-child>\nText Child 2</node-child>\n"
+      "<![CDATA[Char Data]]>"
+      "</node>\n",
+      stream->ToString());
+}
+
+TEST(CFX_XMLElementTest, SaveWithNamespace) {
+  auto stream = pdfium::MakeRetain<StringWriteStream>();
+  CFX_XMLElement node(L"test:root");
+  node.SetAttribute(L"xmlns:test", L"https://example.org/ns1");
+
+  node.Save(stream);
+  EXPECT_EQ("<test:root xmlns:test=\"https://example.org/ns1\" />\n",
+            stream->ToString());
+}
+
+TEST(CFX_XMLElementTest, GetFirstChildNamed) {
+  CFX_XMLElement node(L"node");
+  auto child = pdfium::MakeUnique<CFX_XMLElement>(L"node-child");
+  CFX_XMLElement* node_child1 = child.get();
+  node.AppendChild(std::move(child));
+
+  auto* found = node.GetFirstChildNamed(L"node-child");
+  EXPECT_TRUE(found != nullptr);
+  EXPECT_EQ(node_child1, found);
+}
+
+TEST(CFX_XMLElementTest, GetFirstChildNamedMissing) {
+  CFX_XMLElement node(L"node");
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLElement>(L"node-child"));
+
+  auto* found = node.GetFirstChildNamed(L"node-sibling");
+  EXPECT_TRUE(found == nullptr);
+}
+
+TEST(CFX_XMLElementTest, GetNthChildNamed) {
+  CFX_XMLElement node(L"node");
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLElement>(L"node-child"));
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLElement>(L"node-child"));
+
+  auto child = pdfium::MakeUnique<CFX_XMLElement>(L"node-child");
+  CFX_XMLElement* node_child3 = child.get();
+  node.AppendChild(std::move(child));
+
+  auto* found = node.GetNthChildNamed(L"node-child", 2);
+  EXPECT_TRUE(found != nullptr);
+  EXPECT_EQ(node_child3, found);
+}
+
+TEST(CFX_XMLElementTest, GetNthChildNamedMissingChild) {
+  CFX_XMLElement node(L"node");
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLElement>(L"node-child"));
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLElement>(L"node-child"));
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLElement>(L"node-child"));
+
+  auto* found = node.GetNthChildNamed(L"node-child", 5);
+  EXPECT_TRUE(found == nullptr);
+}
+
+TEST(CFX_XMLElementTest, GetTextData) {
+  CFX_XMLElement node(L"node");
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLText>(L"Text Child 1"));
+
+  auto child = pdfium::MakeUnique<CFX_XMLElement>(L"Node child");
+  CFX_XMLElement* node_child1 = child.get();
+  node.AppendChild(std::move(child));
+
+  node_child1->AppendChild(pdfium::MakeUnique<CFX_XMLText>(L"Text Child 2"));
+
+  node.AppendChild(pdfium::MakeUnique<CFX_XMLCharData>(L"Char Data"));
+
+  EXPECT_EQ(L"Text Child 1Char Data", node.GetTextData());
+}