vaadin-framework/compatibility-server/src/test/java/com/vaadin/v7/tests/server/component/table/TableStateTest.java

package com.vaadin.v7.tests.server.component.table;

import static org.junit.Assert.assertEquals;

import org.junit.Test;

import com.vaadin.v7.shared.ui.table.TableState;
import com.vaadin.v7.ui.Table;

/**
 * Tests for Table State.
 *
 */
public class TableStateTest {

    @Test
    public void getState_tableHasCustomState() {
        TestTable table = new TestTable();
        TableState state = table.getState();
        assertEquals("Unexpected state class", TableState.class,
                state.getClass());
    }

    @Test
    public void getPrimaryStyleName_tableHasCustomPrimaryStyleName() {
        Table table = new Table();
        TableState state = new TableState();
        assertEquals("Unexpected primary style name", state.primaryStyleName,
                table.getPrimaryStyleName());
    }

    @Test
    public void tableStateHasCustomPrimaryStyleName() {
        TableState state = new TableState();
        assertEquals("Unexpected primary style name", "v-table",
                state.primaryStyleName);
    }

    private static class TestTable extends Table {

        @Override
        public TableState getState() {
            return super.getState();
        }
    }
}