vaadin-framework/uitest/src/main/java/com/vaadin/tests/components/treegrid/TreeGridDetailsManager.java

package com.vaadin.tests.components.treegrid;

import java.util.ArrayList;
import java.util.List;

import com.vaadin.data.TreeData;
import com.vaadin.data.provider.TreeDataProvider;
import com.vaadin.server.VaadinRequest;
import com.vaadin.tests.components.AbstractTestUI;
import com.vaadin.ui.Button;
import com.vaadin.ui.HorizontalLayout;
import com.vaadin.ui.Label;
import com.vaadin.ui.TreeGrid;

public class TreeGridDetailsManager extends AbstractTestUI {

    private TreeGrid<String> treeGrid;
    private TreeDataProvider<String> treeDataProvider;
    private List<String> items = new ArrayList<String>();

    private void initializeDataProvider() {
        TreeData<String> data = new TreeData<>();
        for (int i = 0; i < 2; i++) {
            String root = "Root " + i;
            items.add(root);
            data.addItem(null, root);
            for (int j = 0; j < 2; j++) {
                String leaf = "Leaf " + i + "/" + j;
                items.add(leaf);
                data.addItem(root, leaf);
            }
        }
        treeDataProvider = new TreeDataProvider<>(data);
    }

    @Override
    protected void setup(VaadinRequest request) {
        initializeDataProvider();
        treeGrid = new TreeGrid<>();
        treeGrid.setDataProvider(treeDataProvider);
        treeGrid.setSizeFull();
        treeGrid.addColumn(String::toString).setCaption("String")
                .setId("string");
        treeGrid.addColumn((i) -> "--").setCaption("Nothing");
        treeGrid.setHierarchyColumn("string");
        treeGrid.setDetailsGenerator(
                row -> new Label("details for " + row.toString()));
        treeGrid.addItemClickListener(event -> {
            treeGrid.setDetailsVisible(event.getItem(),
                    !treeGrid.isDetailsVisible(event.getItem()));
        });

        Button showDetails = new Button("Show all details", event -> {
            for (String id : items) {
                treeGrid.setDetailsVisible(id, true);
            }
        });
        showDetails.setId("showDetails");
        Button hideDetails = new Button("Hide all details", event -> {
            for (String id : items) {
                treeGrid.setDetailsVisible(id, false);
            }
        });
        hideDetails.setId("hideDetails");
        Button expandAll = new Button("Expand all", event -> {
            treeGrid.expand(items);
        });
        expandAll.setId("expandAll");
        Button collapseAll = new Button("Collapse all", event -> {
            treeGrid.collapse(items);
        });
        collapseAll.setId("collapseAll");
        Button addGrid = new Button("Add grid", event -> {
            addComponent(treeGrid);
            getLayout().setExpandRatio(treeGrid, 2);
        });
        addGrid.setId("addGrid");

        addComponents(new HorizontalLayout(showDetails, hideDetails, expandAll,
                collapseAll), addGrid);

        getLayout().getParent().setHeight("100%");
        getLayout().setHeight("100%");
        treeGrid.setHeight("100%");
        setHeight("100%");
    }

    @Override
    protected String getTestDescription() {
        return "Expanding and collapsing with and without open details rows shouldn't cause exceptions. "
                + "Details row should be reopened upon expanding if it was open before collapsing.";
    }

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