Introduce EphemeralRange::nodes() helper to traverse over a range.

third_party/WebKit/Source/core/editing/EphemeralRangeTest.cpp

Index: third_party/WebKit/Source/core/editing/EphemeralRangeTest.cpp
diff --git a/third_party/WebKit/Source/core/editing/EphemeralRangeTest.cpp b/third_party/WebKit/Source/core/editing/EphemeralRangeTest.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..036d8efd2f5de9b6d1cc41acdc24ffd973ce1c04
--- /dev/null
+++ b/third_party/WebKit/Source/core/editing/EphemeralRangeTest.cpp
@@ -0,0 +1,187 @@
+// Copyright 2015 The Chromium Authors. All rights reserved.

[tkent, 2016/08/17 02:05:02]

2015 -> 2016

[Andrey Kraynov, 2016/08/17 09:58:16]

Done.

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "core/editing/EphemeralRange.h"
+
+#include "core/dom/Range.h"
+#include "core/editing/EditingTestBase.h"
+#include <sstream>
+
+namespace blink {
+
+namespace {
+
+class EphemeralRangeTest : public EditingTestBase {
+protected:
+    template <typename Traversal>
+    std::string traverseRange(Range*) const;
+
+    template <typename Strategy>
+    std::string traverseRange(const EphemeralRangeTemplate<Strategy>&) const;
+
+    template <typename Strategy>
+    std::string traverseRange(const PositionTemplate<Strategy>&, const PositionTemplate<Strategy>&) const;
+
+    Range* getBodyRange() const;
+};
+
+template <typename Traversal = NodeTraversal>
+std::string EphemeralRangeTest::traverseRange(Range* range) const
+{
+    std::stringstream nodesContent;
+    for (Node* node = range->firstNode(); node != range->pastLastNode(); node = Traversal::next(*node)) {
+        nodesContent << "[" << *node << "]";
+    }
+
+    return nodesContent.str();
+}
+
+template <typename Strategy>
+std::string EphemeralRangeTest::traverseRange(const EphemeralRangeTemplate<Strategy>& range) const
+{
+    std::stringstream nodesContent;
+    for (Node& node : inRange(range)) {
+        nodesContent << "[" << node << "]";
+    }
+
+    return nodesContent.str();
+}
+
+template <typename Strategy>
+std::string EphemeralRangeTest::traverseRange(const PositionTemplate<Strategy>& start, const PositionTemplate<Strategy>& end) const
+{
+    std::stringstream nodesContent;
+    for (Node& node : inRange(start, end)) {
+        nodesContent << "[" << node << "]";
+    }
+
+    return nodesContent.str();
+}
+
+Range* EphemeralRangeTest::getBodyRange() const
+{
+    Range* range = Range::create(document());
+    range->selectNode(document().body());
+    return range;
+}
+
+} // namespace
+
+// Tests that |inRange| helper will traverse the whole range exactly as
+// |for (Node* n = firstNode(); n != pastLastNode(); n = Traversal::next(*n))|
+// does.
+TEST_F(EphemeralRangeTest, rangeTraversal)
+{
+    const char* bodyContent = "<p id='host'><b id='one'></b><b id='two'>22</b></p>";
+    setBodyContent(bodyContent);
+
+    const std::string expectedNodes("[BODY][P id=\"host\"][B id=\"one\"][B id=\"two\"][#text \"22\"]");
+
+    // Check all three ways to traverse.
+    EXPECT_EQ(expectedNodes, traverseRange<>(getBodyRange()));
+    EXPECT_EQ(traverseRange<>(getBodyRange()), traverseRange(EphemeralRange(getBodyRange())));
+    EXPECT_EQ(
+        traverseRange<>(getBodyRange()),
+        traverseRange(getBodyRange()->startPosition(), getBodyRange()->endPosition()));
+
+    // The same with FlatTree traversing.
+    EXPECT_EQ(expectedNodes, traverseRange<FlatTreeTraversal>(getBodyRange()));
+    EXPECT_EQ(
+        traverseRange<FlatTreeTraversal>(getBodyRange()),
+        traverseRange(EphemeralRangeInFlatTree(getBodyRange())));
+    EXPECT_EQ(
+        traverseRange<FlatTreeTraversal>(getBodyRange()),
+        traverseRange(toPositionInFlatTree(getBodyRange()->startPosition()), toPositionInFlatTree(getBodyRange()->endPosition())));
+}
+
+// Tests that |inRange| helper will traverse the whole range with shadow DOM.
+TEST_F(EphemeralRangeTest, rangeShadowTraversal)
+{
+    const char* bodyContent = "<p id='host'><b id='one'></b><input type='text' value='some'></p>";
+    setBodyContent(bodyContent);
+
+    EXPECT_EQ(
+        std::string("[BODY][P id=\"host\"][B id=\"one\"][INPUT]"),
+        traverseRange<>(getBodyRange()));
+    EXPECT_EQ(traverseRange<>(getBodyRange()), traverseRange(EphemeralRange(getBodyRange())));
+    EXPECT_EQ(
+        traverseRange<>(getBodyRange()),
+        traverseRange(getBodyRange()->startPosition(), getBodyRange()->endPosition()));
+
+    // In this case FlatTree traverse should differs from DOM tree traverse.
+    EXPECT_EQ(
+        std::string("[BODY][P id=\"host\"][B id=\"one\"][INPUT][DIV id=\"inner-editor\"][#text \"some\"]"),
+        traverseRange<FlatTreeTraversal>(getBodyRange()));
+    EXPECT_EQ(
+        traverseRange<FlatTreeTraversal>(getBodyRange()),
+        traverseRange(toPositionInFlatTree(getBodyRange()->startPosition()), toPositionInFlatTree(getBodyRange()->endPosition())));
+}
+
+// Limit a range and check that it will be traversed correctly.
+TEST_F(EphemeralRangeTest, rangeTraversalLimited)
+{
+    const char* bodyContent = "<p id='host'><b id='one'></b><input type='text' value='some'><span id='two'></p>";
+    setBodyContent(bodyContent);
+
+    // Get a limited range from <p> to <b> nodes.
+    Range* untilB = getBodyRange();
+    untilB->setEnd(document().getElementById("one"), 0, IGNORE_EXCEPTION);
+    EXPECT_EQ(
+        std::string("[BODY][P id=\"host\"][B id=\"one\"]"),
+        traverseRange<>(untilB));
+
+    EXPECT_EQ(traverseRange<>(untilB), traverseRange(EphemeralRange(untilB)));
+    EXPECT_EQ(
+        traverseRange<>(untilB),
+        traverseRange(untilB->startPosition(), untilB->endPosition()));
+
+    EXPECT_EQ(
+        std::string("[BODY][P id=\"host\"][B id=\"one\"]"),
+        traverseRange<FlatTreeTraversal>(untilB));
+    EXPECT_EQ(
+        traverseRange<FlatTreeTraversal>(untilB),
+        traverseRange(EphemeralRangeInFlatTree(untilB)));
+
+    // Get a limited range from <b> to <span> nodes.
+    Range* fromBToSpan = getBodyRange();
+    fromBToSpan->setStart(document().getElementById("one"), 0, IGNORE_EXCEPTION);
+    fromBToSpan->setEnd(document().getElementById("two"), 0, IGNORE_EXCEPTION);
+
+    EXPECT_EQ(
+        std::string("[B id=\"one\"][INPUT][SPAN id=\"two\"]"),
+        traverseRange<>(fromBToSpan));
+    EXPECT_EQ(traverseRange<>(fromBToSpan), traverseRange(EphemeralRange(fromBToSpan)));
+    EXPECT_EQ(
+        traverseRange<>(fromBToSpan),
+        traverseRange(fromBToSpan->startPosition(), fromBToSpan->endPosition()));
+
+    EXPECT_EQ(
+        std::string("[B id=\"one\"][INPUT][DIV id=\"inner-editor\"][#text \"some\"][SPAN id=\"two\"]"),
+        traverseRange<FlatTreeTraversal>(fromBToSpan));
+    EXPECT_EQ(
+        traverseRange<FlatTreeTraversal>(fromBToSpan),
+        traverseRange(EphemeralRangeInFlatTree(fromBToSpan)));
+    EXPECT_EQ(
+        traverseRange<FlatTreeTraversal>(fromBToSpan),
+        traverseRange(toPositionInFlatTree(fromBToSpan->startPosition()), toPositionInFlatTree(fromBToSpan->endPosition())));
+}
+
+TEST_F(EphemeralRangeTest, traversalWithNullCheck)
+{
+    const char* bodyContent = "<p id='host'><b id='one'></b></p>";
+    setBodyContent(bodyContent);
+
+    Range* inner = getBodyRange();
+    inner->setEnd(document().getElementById("one"), 0, IGNORE_EXCEPTION);
+    auto iterable = inRange(inner->endPosition(), inner->startPosition());
+    auto it = iterable.begin();
+    const auto end = iterable.end();
+    while (!it.isNull() && it != end)
+        ++it;
+
+    EXPECT_TRUE(it.isNull());
+    EXPECT_NE(end, it);
+}
+
+} // namespace blink