/*
 * 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.v7.tests.components.grid.basicfeatures.escalator;

import static org.junit.Assert.assertEquals;

import org.junit.Test;
import org.openqa.selenium.WebElement;

import com.vaadin.v7.tests.components.grid.basicfeatures.EscalatorBasicClientFeaturesTest;

public class EscalatorColspanTest extends EscalatorBasicClientFeaturesTest {
    private static final int NO_COLSPAN = 1;

    @Test
    public void testNoColspan() {
        openTestURL();
        populate();

        assertEquals(NO_COLSPAN, getColSpan(getHeaderCell(0, 0)));
        assertEquals(NO_COLSPAN, getColSpan(getBodyCell(0, 0)));
        assertEquals(NO_COLSPAN, getColSpan(getFooterCell(0, 0)));
    }

    @Test
    public void testColspan() {
        openTestURL();
        populate();

        int firstCellWidth = getBodyCell(0, 0).getSize().getWidth();
        int secondCellWidth = getBodyCell(0, 1).getSize().getWidth();
        int doubleCellWidth = firstCellWidth + secondCellWidth;

        selectMenuPath(FEATURES, COLUMN_SPANNING, COLSPAN_NORMAL);

        WebElement bodyCell = getBodyCell(0, 0);
        assertEquals("Cell was not spanned correctly", 2, getColSpan(bodyCell));
        assertEquals(
                "Spanned cell's width was not the sum of the previous cells ("
                        + firstCellWidth + " + " + secondCellWidth + ")",
                doubleCellWidth, bodyCell.getSize().getWidth(), 1);
    }

    @Test
    public void testColspanToggle() {
        openTestURL();
        populate();

        int singleCellWidth = getBodyCell(0, 0).getSize().getWidth();

        selectMenuPath(FEATURES, COLUMN_SPANNING, COLSPAN_NORMAL);
        selectMenuPath(FEATURES, COLUMN_SPANNING, COLSPAN_NONE);

        WebElement bodyCell = getBodyCell(0, 0);
        assertEquals(NO_COLSPAN, getColSpan(bodyCell));
        assertEquals(singleCellWidth, bodyCell.getSize().getWidth(), 1);
    }

    private static int getColSpan(WebElement cell) {
        String attribute = cell.getAttribute("colspan");
        if (attribute == null) {
            return NO_COLSPAN;
        } else {
            return Integer.parseInt(attribute);
        }
    }
}