vaadin-framework/uitest/src/main/java/com/vaadin/tests/contextclick/GridContextClick.java

/*
 * Copyright 2000-2016 Vaadin Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.vaadin.tests.contextclick;

import com.vaadin.tests.util.PersonContainer;
import com.vaadin.ui.Button;
import com.vaadin.ui.Button.ClickEvent;
import com.vaadin.ui.HorizontalLayout;
import com.vaadin.v7.data.Item;
import com.vaadin.v7.shared.ui.grid.GridConstants.Section;
import com.vaadin.v7.ui.Grid;
import com.vaadin.v7.ui.Grid.GridContextClickEvent;

public class GridContextClick
        extends AbstractContextClickUI<Grid, GridContextClickEvent> {

    @Override
    protected Grid createTestComponent() {
        Grid grid = new Grid(PersonContainer.createWithTestData());
        grid.setFooterVisible(true);
        grid.appendFooterRow();

        grid.setColumnOrder("address", "email", "firstName", "lastName",
                "phoneNumber", "address.streetAddress", "address.postalCode",
                "address.city");

        grid.setWidth("100%");
        grid.setHeight("400px");

        return grid;
    }

    @Override
    protected void handleContextClickEvent(GridContextClickEvent event) {
        String value = "";
        Object propertyId = event.getPropertyId();
        if (event.getItemId() != null) {
            Item item = event.getComponent().getContainerDataSource()
                    .getItem(event.getItemId());
            value += item.getItemProperty("firstName").getValue();
            value += " " + item.getItemProperty("lastName").getValue();
        } else if (event.getSection() == Section.HEADER) {
            value = event.getComponent().getHeaderRow(event.getRowIndex())
                    .getCell(propertyId).getText();
        } else if (event.getSection() == Section.FOOTER) {
            value = event.getComponent().getFooterRow(event.getRowIndex())
                    .getCell(propertyId).getText();
        }
        log("ContextClickEvent value: " + value + ", propertyId: " + propertyId
                + ", section: " + event.getSection());
    }

    @Override
    protected HorizontalLayout createContextClickControls() {
        HorizontalLayout controls = super.createContextClickControls();
        controls.addComponent(
                new Button("Remove all content", new Button.ClickListener() {

                    @Override
                    public void buttonClick(ClickEvent event) {
                        testComponent.getContainerDataSource().removeAllItems();
                    }
                }));
        return controls;
    }
}