aboutsummaryrefslogtreecommitdiffstats
path: root/uitest/src/main/java/com/vaadin/tests/components/treetable/KeepAllItemsVisible.java
blob: 33e0873ee9f506e6e15d0e871d5fa211f2987be9 (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
105
106
107
108
109
110
111
112
113
114
package com.vaadin.tests.components.treetable;

import com.vaadin.tests.components.TestBase;
import com.vaadin.ui.Button;
import com.vaadin.v7.data.Item;
import com.vaadin.v7.ui.Tree.CollapseEvent;
import com.vaadin.v7.ui.Tree.CollapseListener;
import com.vaadin.v7.ui.Tree.ExpandEvent;
import com.vaadin.v7.ui.Tree.ExpandListener;
import com.vaadin.v7.ui.TreeTable;

public class KeepAllItemsVisible extends TestBase
        implements ExpandListener, CollapseListener {

    private static final String CAPTION = "caption";

    @Override
    protected void setup() {
        getMainWindow().getContent().setHeight(null);
        final TreeTable tt = new TreeTable();
        tt.setWidth("400px");
        tt.addContainerProperty(CAPTION, String.class, "");
        for (int i = 0; i < 20; i++) {
            String id = "Root " + i;
            Item item = tt.addItem(id);
            item.getItemProperty(CAPTION).setValue(id);
            addChildren(tt, id, 5, 2);
        }

        tt.setSelectable(true);
        tt.setImmediate(true);

        tt.addListener((ExpandListener) this);
        tt.addListener((CollapseListener) this);
        tt.addValueChangeListener(event -> {
            recalculatePageLength(tt);
            tt.markAsDirty();
        });
        addComponent(tt);

        recalculatePageLength(tt);

        Button b = new Button("Set pagelength to 10",
                event -> tt.setPageLength(10));
        addComponent(b);
        b = new Button("Set pagelength to 20", event -> tt.setPageLength(20));
        addComponent(b);
        b = new Button("Set pagelength to 0", event -> tt.setPageLength(0));
        addComponent(b);
    }

    private void recalculatePageLength(TreeTable tt) {
        tt.setPageLength(getVisibleNodeCount(tt));
    }

    private int getVisibleNodeCount(TreeTable tt) {
        int count = 0;
        for (Object rootId : tt.rootItemIds()) {
            count += countVisibleNodes(tt, rootId);
        }

        return count;

    }

    private int countVisibleNodes(TreeTable tt, Object node) {
        int count = 1;
        if (!tt.isCollapsed(node) && tt.hasChildren(node)) {
            for (Object childId : tt.getChildren(node)) {
                count += countVisibleNodes(tt, childId);
            }
        }

        return count;
    }

    private void addChildren(TreeTable tt, String parentId, int nr,
            int maxNesting) {
        if (maxNesting < 1) {
            return;
        }

        for (int i = 1; i <= nr; i++) {
            String id = parentId + "/" + i;
            Item item = tt.addItem(id);
            item.getItemProperty(CAPTION).setValue(id);
            tt.setParent(id, parentId);
            addChildren(tt, id, nr, maxNesting - 1);
        }

    }

    @Override
    protected String getDescription() {
        return "Keeps the TreeTable pagelength so that all expanded items are shown.";
    }

    @Override
    protected Integer getTicketNumber() {
        return 7292;
    }

    @Override
    public void nodeCollapse(CollapseEvent event) {
        recalculatePageLength((TreeTable) event.getSource());

    }

    @Override
    public void nodeExpand(ExpandEvent event) {
        recalculatePageLength((TreeTable) event.getSource());
    }

}