uitest/src/test/java/com/vaadin/tests/components/treegrid/TreeGridInWindowTest.java


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

package com.vaadin.tests.components.treegrid;

import org.junit.Test;

import com.vaadin.testbench.elements.ButtonElement;
import com.vaadin.testbench.elements.TreeGridElement;
import com.vaadin.tests.tb3.MultiBrowserTest;

public class TreeGridInWindowTest extends MultiBrowserTest {

    @Test
    public void collapse_and_expand_first_child_multiple_times() {
        setDebug(true);
        openTestURL();

        ButtonElement openWindowButton = $(ButtonElement.class).first();
        openWindowButton.click();

        TreeGridElement grid = $(TreeGridElement.class).first();

        for (int i = 0; i < 10; i++) {
            // Collapse first child node
            grid.getExpandElement(1, 0).click();
            waitUntil(webDriver -> grid.getRowCount() == 5);

            // Expand first child node
            grid.getExpandElement(1, 0).click();
            waitUntil(webDriver -> grid.getRowCount() == 7);
        }

        assertNoErrorNotifications();
    }
}