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

package com.vaadin.tests.components.uitest.components;

import com.vaadin.tests.components.uitest.TestSampler;
import com.vaadin.v7.data.Item;
import com.vaadin.v7.data.util.HierarchicalContainer;
import com.vaadin.v7.ui.TreeTable;

@SuppressWarnings("deprecation")
public class TreeTableCssTest {
    private int debugIdCounter = 0;

    public TreeTableCssTest(TestSampler parent) {
        TreeTable treeTable = new TreeTable();
        treeTable.setId("treetable" + debugIdCounter++);
        treeTable.setWidth("100%");
        treeTable.setSelectable(true);
        parent.addComponent(treeTable);

        HierarchicalContainer hc = createHierarchicalContainer();

        treeTable.setContainerDataSource(hc);

        for (Object itemId : treeTable.getItemIds()) {
            treeTable.setCollapsed(itemId, false);
        }

        parent.addReadOnlyChangeListener(
                event -> treeTable.setReadOnly(!treeTable.isReadOnly()));
    }

    private HierarchicalContainer createHierarchicalContainer() {
        String[] itemNames = { "Foo", "Baar" };

        HierarchicalContainer hc = new HierarchicalContainer();
        hc.addContainerProperty("NAME", String.class, null);

        for (String parentId : itemNames) {
            Item parent = hc.addItem(parentId);
            parent.getItemProperty("NAME").setValue(parentId);
            hc.setChildrenAllowed(parent, true);
            for (int i = 0; i < 5; i++) {
                String childId = parentId + i;
                Item child = hc.addItem(childId);
                child.getItemProperty("NAME").setValue(childId);
                if (!hc.setParent(childId, parentId)) {
                    System.out.println("Unable to set parent \"" + parentId
                            + "\" for child with id: \"" + childId + "\"");
                }
            }
        }
        return hc;
    }

}