| // 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(toNGBlockNode(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(toNGBlockNode(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( |
| toNGBlockNode(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(toNGBlockNode(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 |