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
|
package com.vaadin.tests.components.treegrid;
import org.junit.Before;
import org.junit.Test;
import com.vaadin.testbench.elements.ButtonElement;
import com.vaadin.testbench.elements.RadioButtonGroupElement;
import com.vaadin.testbench.elements.TreeGridElement;
import com.vaadin.tests.tb3.SingleBrowserTest;
import static org.junit.Assert.assertEquals;
public class TreeGridExpandCollapseRecursivelyTest extends SingleBrowserTest {
private static final int rowCount0 = 5;
private static final int rowCount1 = rowCount0 + rowCount0 * 5;
private static final int rowCount2 = rowCount1
+ (rowCount1 - rowCount0) * 5;
private static final int rowCount3 = rowCount2
+ (rowCount2 - rowCount1) * 5;
private static final int rowCount4 = rowCount3
+ (rowCount3 - rowCount2) * 5;
private TreeGridElement grid;
private RadioButtonGroupElement depthSelector;
private ButtonElement expandButton;
private ButtonElement collapseButton;
@Before
public void before() {
openTestURL();
grid = $(TreeGridElement.class).first();
depthSelector = $(RadioButtonGroupElement.class).first();
expandButton = $(ButtonElement.class).get(0);
collapseButton = $(ButtonElement.class).get(1);
}
@Test
public void expandVariousDepth() {
assertEquals(rowCount0, grid.getRowCount());
selectDepth(0);
expandButton.click();
assertEquals(rowCount1, grid.getRowCount());
selectDepth(1);
expandButton.click();
assertEquals(rowCount2, grid.getRowCount());
selectDepth(2);
expandButton.click();
assertEquals(rowCount3, grid.getRowCount());
selectDepth(3);
expandButton.click();
assertEquals(rowCount4, grid.getRowCount());
}
@Test(timeout = 5000)
public void expandAndCollapseAllItems() {
assertEquals(rowCount0, grid.getRowCount());
selectDepth(3);
expandButton.click();
assertEquals(rowCount4, grid.getRowCount());
collapseButton.click();
assertEquals(rowCount0, grid.getRowCount());
}
@Test
public void partialCollapse() {
assertEquals(rowCount0, grid.getRowCount());
selectDepth(3);
expandButton.click();
assertEquals(rowCount4, grid.getRowCount());
selectDepth(1);
collapseButton.click();
assertEquals(rowCount0, grid.getRowCount());
selectDepth(0);
expandButton.click();
assertEquals(rowCount1, grid.getRowCount());
// Open just one subtree to see if it is still fully expanded
grid.getExpandElement(2, 0).click();
assertEquals(rowCount1 + rowCount2, grid.getRowCount());
}
private void selectDepth(int depth) {
depthSelector.setValue(String.valueOf(depth));
}
}
|
