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

// 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 {

class NGBlockChildIteratorTest : public RenderingTest {};

TEST_F(NGBlockChildIteratorTest, NullFirstChild) {
  NGBlockChildIterator iterator(nullptr, nullptr);
  ASSERT_EQ(NGBlockChildIterator::Entry(nullptr, nullptr),
            iterator.NextChild());
}

TEST_F(NGBlockChildIteratorTest, NoBreakToken) {
  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);
  ASSERT_EQ(NGBlockChildIterator::Entry(node1, nullptr), iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node2, nullptr), iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node3, nullptr), iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(nullptr, nullptr),
            iterator.NextChild());
}

TEST_F(NGBlockChildIteratorTest, BreakTokenWithFinishedChild) {
  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();

  Vector<RefPtr<NGBreakToken>> child_break_tokens;
  child_break_tokens.push_back(NGBlockBreakToken::create(node1));
  RefPtr<NGBlockBreakToken> parent_token =
      NGBlockBreakToken::create(container, LayoutUnit(50), child_break_tokens);

  // The iterator should loop through two children.
  NGBlockChildIterator iterator(node1, parent_token.get());
  ASSERT_EQ(NGBlockChildIterator::Entry(node2, nullptr), iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node3, nullptr), iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(nullptr, nullptr),
            iterator.NextChild());

  child_break_tokens.push_back(NGBlockBreakToken::create(node2));
  parent_token =
      NGBlockBreakToken::create(container, LayoutUnit(50), child_break_tokens);

  // The iterator should loop through two children.
  NGBlockChildIterator iterator2(node1, parent_token.get());
  ASSERT_EQ(NGBlockChildIterator::Entry(node1, nullptr), iterator2.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node3, nullptr), iterator2.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(nullptr, nullptr),
            iterator2.NextChild());
}

TEST_F(NGBlockChildIteratorTest, BreakTokenWithUnFinishedChild) {
  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();

  Vector<RefPtr<NGBreakToken>> child_break_tokens;
  RefPtr<NGBreakToken> child_token =
      NGBlockBreakToken::create(node1, LayoutUnit(), child_break_tokens);
  child_break_tokens.push_back(child_token);
  RefPtr<NGBlockBreakToken> parent_token =
      NGBlockBreakToken::create(container, LayoutUnit(50), child_break_tokens);

  // The iterator should loop through three children, one with a break token.
  NGBlockChildIterator iterator(node1, parent_token.get());
  ASSERT_EQ(NGBlockChildIterator::Entry(node1, child_token.get()),
            iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node2, nullptr), iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node3, nullptr), iterator.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(nullptr, nullptr),
            iterator.NextChild());

  child_token =
      NGBlockBreakToken::create(node2, LayoutUnit(), child_break_tokens);
  child_break_tokens.push_back(child_token);
  parent_token =
      NGBlockBreakToken::create(container, LayoutUnit(50), child_break_tokens);

  // The iterator should loop through three children, one with a break token.
  NGBlockChildIterator iterator2(node1, parent_token.get());
  ASSERT_EQ(NGBlockChildIterator::Entry(node1, nullptr), iterator2.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node2, child_token.get()),
            iterator2.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(node3, nullptr), iterator2.NextChild());
  ASSERT_EQ(NGBlockChildIterator::Entry(nullptr, nullptr),
            iterator2.NextChild());
}

}  // namespace
}  // namespace blink