vaadin-framework/uitest/src/test/java/com/vaadin/tests/components/table/SortableHeaderStylesTest.java

package com.vaadin.tests.components.table;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import org.junit.Test;

import com.vaadin.testbench.elements.OptionGroupElement;
import com.vaadin.testbench.elements.TableElement;
import com.vaadin.tests.tb3.SingleBrowserTest;

public class SortableHeaderStylesTest extends SingleBrowserTest {
    @Test
    public void testSortableHeaderStyles() {
        openTestURL();

        assertFalse(hasSortableStyle(0));
        for (int i = 1; i < 8; i++) {
            assertTrue(hasSortableStyle(i));
        }

        OptionGroupElement sortableSelector = $(OptionGroupElement.class)
                .first();

        // Toggle sortability
        sortableSelector.selectByText("lastName");
        assertFalse(hasSortableStyle(3));

        // Toggle back
        sortableSelector.selectByText("lastName");
        assertTrue(hasSortableStyle(3));
    }

    private boolean hasSortableStyle(int column) {
        return $(TableElement.class).first().getHeaderCell(column)
                .getAttribute("class").contains("sortable");
    }
}