aboutsummaryrefslogtreecommitdiffstats
path: root/uitest/src/com/vaadin/tests/components/treetable/DynamicallyModified.java
blob: 6dc5417d28c0a11f6ed57fb437247f23640dbd0a (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
115
116
117
118
119
120
121
122
123
124
125
126
127
package com.vaadin.tests.components.treetable;

import java.util.Collection;

import com.vaadin.tests.components.TestBase;
import com.vaadin.ui.Tree;
import com.vaadin.ui.Tree.CollapseEvent;
import com.vaadin.ui.Tree.ExpandEvent;
import com.vaadin.ui.TreeTable;

public class DynamicallyModified extends TestBase implements
        Tree.ExpandListener, Tree.CollapseListener {
    protected static final String NAME_PROPERTY = "Name";
    protected static final String HOURS_PROPERTY = "Hours done";
    protected static final String MODIFIED_PROPERTY = "Last Modified";
    private static final String DEFAULT_DATE = "Wed Nov 30 14:40:26 EET 2011";

    protected TreeTable treetable;

    @Override
    protected void setup() {
        getLayout().setWidth("100%");

        // Create the treetable
        treetable = new TreeTable();
        treetable.setWidth("100%");
        treetable.addListener((Tree.ExpandListener) this);
        treetable.addListener((Tree.CollapseListener) this);

        addComponent(treetable);

        // Add Table columns
        treetable.addContainerProperty(NAME_PROPERTY, String.class, "");
        treetable.addContainerProperty(HOURS_PROPERTY, Integer.class, 0);
        treetable.addContainerProperty(MODIFIED_PROPERTY, String.class,
                DEFAULT_DATE);

        // Populate table
        Object allProjects = treetable.addItem(new Object[] { "All Projects",
                18, DEFAULT_DATE }, null);
        Object year2010 = treetable.addItem(new Object[] { "Year 2010", 18,
                DEFAULT_DATE }, null);
        Object customerProject1 = treetable.addItem(new Object[] {
                "Customer Project 1", 13, DEFAULT_DATE }, null);
        Object customerProject1Implementation = treetable.addItem(new Object[] {
                "Implementation", 5, DEFAULT_DATE }, null);
        Object customerProject1Planning = treetable.addItem(new Object[] {
                "Planning", 2, DEFAULT_DATE }, null);
        Object customerProject1Prototype = treetable.addItem(new Object[] {
                "Prototype", 5, DEFAULT_DATE }, null);
        Object customerProject2 = treetable.addItem(new Object[] {
                "Customer Project 2", 5, DEFAULT_DATE }, null);
        Object customerProject2Planning = treetable.addItem(new Object[] {
                "Planning", 5, DEFAULT_DATE }, null);

        // Set hierarchy
        treetable.setParent(year2010, allProjects);
        treetable.setParent(customerProject1, year2010);
        treetable.setParent(customerProject1Implementation, customerProject1);
        treetable.setParent(customerProject1Planning, customerProject1);
        treetable.setParent(customerProject1Prototype, customerProject1);
        treetable.setParent(customerProject2, year2010);
        treetable.setParent(customerProject2Planning, customerProject2);

        // Disallow children from leaves
        treetable.setChildrenAllowed(customerProject1Implementation, false);
        treetable.setChildrenAllowed(customerProject1Planning, false);
        treetable.setChildrenAllowed(customerProject1Prototype, false);
        treetable.setChildrenAllowed(customerProject2Planning, false);

        // Expand all
        treetable.setCollapsed(allProjects, false);
        treetable.setCollapsed(year2010, false);
        treetable.setCollapsed(customerProject1, false);
        treetable.setCollapsed(customerProject2, false);
    }

    @Override
    protected String getDescription() {
        return "Collaps 'Customer Project 1' will cause the first child if it to be removed. Expanding 'Custom Project 2' will cause a new child to be added. These events should be rendered correctly.";
    }

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

    private int newChild = 1;

    @Override
    public void nodeExpand(ExpandEvent event) {
        Object expandedItemId = event.getItemId();
        // 7 == "Customer Project 1"
        if (expandedItemId != Integer.valueOf(7)) {
            return;
        }
        Object newChildId = treetable.addItem(new Object[] {
                "New child " + newChild++, 5, "Fri Nov 04 10:53:16 EET 2011" },
                null);
        treetable.setParent(newChildId, expandedItemId);
        treetable.setChildrenAllowed(newChildId, false);
    }

    @Override
    public void nodeCollapse(CollapseEvent event) {

        Object collapsedItemId = event.getItemId();

        // 3 == "Customer Project 1"
        if (collapsedItemId != Integer.valueOf(3)) {
            return;
        }
        @SuppressWarnings("unchecked")
        Collection<Object> childs = (Collection<Object>) treetable
                .getChildren(event.getItemId());

        if (childs == null) {
            return;
        }
        Object[] arr = childs.toArray();

        if (arr.length > 0) {
            treetable.removeItem(arr[0]);
        }

    }
}