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

package com.vaadin.tests.components.grid;

import java.util.LinkedList;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.stream.Collectors;
import java.util.stream.IntStream;

import com.vaadin.annotations.Widgetset;
import com.vaadin.data.ValueProvider;
import com.vaadin.data.converter.StringToIntegerConverter;
import com.vaadin.data.provider.DataProvider;
import com.vaadin.data.provider.ListDataProvider;
import com.vaadin.server.VaadinRequest;
import com.vaadin.tests.components.AbstractTestUIWithLog;
import com.vaadin.ui.Button;
import com.vaadin.ui.Grid.SelectionMode;
import com.vaadin.ui.Label;
import com.vaadin.ui.TabSheet;
import com.vaadin.ui.TextField;
import com.vaadin.ui.renderers.NumberRenderer;

@Widgetset("com.vaadin.DefaultWidgetSet")
public class GridInTabSheet extends AbstractTestUIWithLog {

    public class DataCommunicator<T>
            extends com.vaadin.data.provider.DataCommunicator<T> {
        @Override
        protected void onRequestRows(int firstRowIndex, int numberOfRows,
                int firstCachedRowIndex, int cacheSize) {
            log("RequestRows: [" + firstRowIndex + "," + numberOfRows + ","
                    + firstCachedRowIndex + "," + cacheSize + "]");
            super.onRequestRows(firstRowIndex, numberOfRows,
                    firstCachedRowIndex, cacheSize);
        }
    }

    public class Grid<T> extends com.vaadin.ui.Grid<T> {

        public Grid() {
            super(new DataCommunicator<>());
        }
    }

    private AtomicInteger index = new AtomicInteger(0);

    @Override
    protected void setup(VaadinRequest request) {
        TabSheet sheet = new TabSheet();

        final Grid<Integer> grid = new Grid<>();
        grid.setSelectionMode(SelectionMode.MULTI);
        grid.addColumn(ValueProvider.identity(), new NumberRenderer())
                .setId("count").setEditorBinding(
                        grid.getEditor().getBinder().forField(new TextField())
                                .withConverter(new StringToIntegerConverter(""))
                                .bind(i -> i, (i, v) -> {
                                }));
        grid.getEditor().setEnabled(true);

        LinkedList<Integer> items = IntStream.range(0, 3).boxed()
                .collect(Collectors.toCollection(LinkedList::new));
        ListDataProvider<Integer> provider = DataProvider.ofCollection(items);
        grid.setDataProvider(provider);

        sheet.addTab(grid, "Grid");
        sheet.addTab(new Label("Hidden"), "Label");

        addComponent(sheet);
        addComponent(new Button("Add row to Grid", event -> {
            items.add(100 + index.getAndIncrement());
            provider.refreshAll();
        }));

        Button remove = new Button("Remove row from Grid", event -> {
            Object firstItemId = items.get(0);
            if (firstItemId != null) {
                items.remove(0);
                provider.refreshAll();
            }
        });
        addComponent(remove);
        Button addGenerator = new Button("Add CellStyleGenerator", event -> {
            grid.setStyleGenerator(item -> {
                if (item % 2 == 1) {
                    return null;
                }
                return "count";
            });
        });

        addComponent(addGenerator);
    }
}