[LayoutNG] Introduce block child iterator.

third_party/WebKit/Source/core/layout/ng/ng_block_child_iterator_test.cc

Index: third_party/WebKit/Source/core/layout/ng/ng_block_child_iterator_test.cc
diff --git a/third_party/WebKit/Source/core/layout/ng/ng_block_child_iterator_test.cc b/third_party/WebKit/Source/core/layout/ng/ng_block_child_iterator_test.cc
new file mode 100644
index 0000000000000000000000000000000000000000..0d6fb038469252f8e54d09285abab5fab3be95b4
--- /dev/null
+++ b/third_party/WebKit/Source/core/layout/ng/ng_block_child_iterator_test.cc
@@ -0,0 +1,124 @@
+// Copyright 2017 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/layout/ng/ng_block_child_iterator.h"
+
+#include "core/layout/LayoutTestHelper.h"
+#include "core/layout/ng/ng_block_break_token.h"
+#include "core/layout/ng/ng_block_node.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace blink {
+namespace {
+
+#define CHILD_TOKEN_PAIR(child, token)                   \
+  std::make_pair(static_cast<NGLayoutInputNode*>(child), \
+                 static_cast<NGBreakToken*>(token))
+
+class NGBlockChildIteratorTest : public RenderingTest {};
+
+TEST_F(NGBlockChildIteratorTest, testNullFirstChild) {
+  NGBlockChildIterator iterator(nullptr, nullptr);
+  EXPECT_EQ(CHILD_TOKEN_PAIR(nullptr, nullptr), iterator.NextChild());
+}
+
+TEST_F(NGBlockChildIteratorTest, testNoBreakToken) {
+  setBodyInnerHTML(R"HTML(
+      <div id='child1'></div>
+      <div id='child2'></div>
+      <div id='child3'></div>
+    )HTML");
+  NGLayoutInputNode* node1 =
+      new NGBlockNode(toLayoutBlockFlow(getLayoutObjectByElementId("child1")));
+  NGLayoutInputNode* node2 = node1->NextSibling();
+  NGLayoutInputNode* node3 = node2->NextSibling();
+
+  // The iterator should loop through three children.
+  NGBlockChildIterator iterator(node1, nullptr);
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node1, nullptr), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node2, nullptr), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node3, nullptr), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(nullptr, nullptr), iterator.NextChild());
+}
+
+TEST_F(NGBlockChildIteratorTest, testBreakTokenWithFinishedChild) {
+  setBodyInnerHTML(R"HTML(
+      <div id='container'>
+        <div id='child1'></div>
+        <div id='child2'></div>
+        <div id='child3'></div>
+      </div>
+    )HTML");
+  NGBlockNode* container = new NGBlockNode(
+      toLayoutBlockFlow(getLayoutObjectByElementId("container")));
+  NGLayoutInputNode* node1 = container->FirstChild();
+  NGLayoutInputNode* node2 = node1->NextSibling();
+  NGLayoutInputNode* node3 = node2->NextSibling();
+
+  HeapVector<Member<NGBreakToken>> child_break_tokens;
+  child_break_tokens.push_back(new NGBlockBreakToken(toNGBlockNode(node1)));
+  NGBlockBreakToken* parent_token =
+      new NGBlockBreakToken(container, LayoutUnit(50), child_break_tokens);
+
+  // The iterator should loop through two children.
+  NGBlockChildIterator iterator(node1, parent_token);
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node2, nullptr), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node3, nullptr), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(nullptr, nullptr), iterator.NextChild());
+
+  child_break_tokens.push_back(new NGBlockBreakToken(toNGBlockNode(node2)));
+  parent_token =
+      new NGBlockBreakToken(container, LayoutUnit(50), child_break_tokens);
+
+  // The iterator should loop through two children.
+  NGBlockChildIterator iterator2(node1, parent_token);
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node1, nullptr), iterator2.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node3, nullptr), iterator2.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(nullptr, nullptr), iterator2.NextChild());
+}
+
+TEST_F(NGBlockChildIteratorTest, testBreakTokenWithUnFinishedChild) {
+  setBodyInnerHTML(R"HTML(
+      <div id='container'>
+        <div id='child1'></div>
+        <div id='child2'></div>
+        <div id='child3'></div>
+      </div>
+    )HTML");
+  NGBlockNode* container = new NGBlockNode(
+      toLayoutBlockFlow(getLayoutObjectByElementId("container")));
+  NGLayoutInputNode* node1 = container->FirstChild();
+  NGLayoutInputNode* node2 = node1->NextSibling();
+  NGLayoutInputNode* node3 = node2->NextSibling();
+
+  HeapVector<Member<NGBreakToken>> child_break_tokens;
+  NGBreakToken* child_token = new NGBlockBreakToken(
+      toNGBlockNode(node1), LayoutUnit(), child_break_tokens);
+  child_break_tokens.push_back(child_token);
+  NGBlockBreakToken* parent_token =
+      new NGBlockBreakToken(container, LayoutUnit(50), child_break_tokens);
+
+  // The iterator should loop through three children, one with a break token.
+  NGBlockChildIterator iterator(node1, parent_token);
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node1, child_token), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node2, nullptr), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node3, nullptr), iterator.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(nullptr, nullptr), iterator.NextChild());
+
+  child_token = new NGBlockBreakToken(toNGBlockNode(node2), LayoutUnit(),
+                                      child_break_tokens);
+  child_break_tokens.push_back(child_token);
+  parent_token =
+      new NGBlockBreakToken(container, LayoutUnit(50), child_break_tokens);
+
+  // The iterator should loop through three children, one with a break token.
+  NGBlockChildIterator iterator2(node1, parent_token);
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node1, nullptr), iterator2.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node2, child_token), iterator2.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(node3, nullptr), iterator2.NextChild());
+  EXPECT_EQ(CHILD_TOKEN_PAIR(nullptr, nullptr), iterator2.NextChild());
+}
+
+}  // namespace
+}  // namespace blink