summaryrefslogtreecommitdiffstats
path: root/third_party/WebKit/Source/core/layout/MultiColumnFragmentainerGroupTest.cpp
blob: 3e2b6deed67ce0a09f0f2f0882eb963e624f2ffc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
// Copyright 2015 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/MultiColumnFragmentainerGroup.h"

#include "core/layout/LayoutMultiColumnFlowThread.h"
#include "core/layout/LayoutMultiColumnSet.h"
#include "core/layout/LayoutTestHelper.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace blink {

namespace {

class MultiColumnFragmentainerGroupTest : public RenderingTest {
public:
    MultiColumnFragmentainerGroupTest() : m_flowThread(nullptr), m_columnSet(nullptr) { }

protected:
    void SetUp() override;
    void TearDown() override;

    LayoutMultiColumnSet& columnSet() { return *m_columnSet; }

    static int groupCount(const MultiColumnFragmentainerGroupList&);

private:
    LayoutMultiColumnFlowThread* m_flowThread;
    LayoutMultiColumnSet* m_columnSet;
};

void MultiColumnFragmentainerGroupTest::SetUp()
{
    RenderingTest::SetUp();
    RefPtr<ComputedStyle> style = ComputedStyle::create();
    m_flowThread = LayoutMultiColumnFlowThread::createAnonymous(document(), *style.get());
    m_columnSet = LayoutMultiColumnSet::createAnonymous(*m_flowThread, *m_flowThread->style());
}

void MultiColumnFragmentainerGroupTest::TearDown()
{
    m_columnSet->destroy();
    m_flowThread->destroy();
    RenderingTest::TearDown();
}

int MultiColumnFragmentainerGroupTest::groupCount(const MultiColumnFragmentainerGroupList& groupList)
{
    int count = 0;
    for (const auto& dummyGroup : groupList) {
        (void) dummyGroup;
        count++;
    }
    return count;
}

TEST_F(MultiColumnFragmentainerGroupTest, Create)
{
    MultiColumnFragmentainerGroupList groupList(columnSet());
    EXPECT_EQ(groupCount(groupList), 1);
}

TEST_F(MultiColumnFragmentainerGroupTest, DeleteExtra)
{
    MultiColumnFragmentainerGroupList groupList(columnSet());
    EXPECT_EQ(groupCount(groupList), 1);
    groupList.deleteExtraGroups();
    EXPECT_EQ(groupCount(groupList), 1);
}

TEST_F(MultiColumnFragmentainerGroupTest, AddThenDeleteExtra)
{
    MultiColumnFragmentainerGroupList groupList(columnSet());
    EXPECT_EQ(groupCount(groupList), 1);
    groupList.addExtraGroup();
    EXPECT_EQ(groupCount(groupList), 2);
    groupList.deleteExtraGroups();
    EXPECT_EQ(groupCount(groupList), 1);
}

TEST_F(MultiColumnFragmentainerGroupTest, AddTwoThenDeleteExtraThenAddThreeThenDeleteExtra)
{
    MultiColumnFragmentainerGroupList groupList(columnSet());
    EXPECT_EQ(groupCount(groupList), 1);
    groupList.addExtraGroup();
    EXPECT_EQ(groupCount(groupList), 2);
    groupList.addExtraGroup();
    EXPECT_EQ(groupCount(groupList), 3);
    groupList.deleteExtraGroups();
    EXPECT_EQ(groupCount(groupList), 1);
    groupList.addExtraGroup();
    EXPECT_EQ(groupCount(groupList), 2);
    groupList.addExtraGroup();
    EXPECT_EQ(groupCount(groupList), 3);
    groupList.addExtraGroup();
    EXPECT_EQ(groupCount(groupList), 4);
    groupList.deleteExtraGroups();
    EXPECT_EQ(groupCount(groupList), 1);
}

} // anonymous namespace

} // namespace blink