vaadin-framework/uitest/src/test/java/com/vaadin/tests/components/listselect/ListSelectAddRemoveItemsTest.java

package com.vaadin.tests.components.listselect;

import static org.junit.Assert.assertEquals;

import java.util.Arrays;

import org.junit.Test;

import com.vaadin.testbench.elements.ButtonElement;
import com.vaadin.testbench.elements.ListSelectElement;
import com.vaadin.tests.tb3.SingleBrowserTest;

public class ListSelectAddRemoveItemsTest extends SingleBrowserTest {
    @Test
    public void testAddAndRemove() {
        openTestURL();
        assertOptions("a", "b", "c");

        click("Add first");
        assertOptions("first", "a", "b", "c");

        click("Swap");
        assertOptions("c", "a", "b", "first");

        click("Remove first");
        assertOptions("a", "b", "first");

        click("Add middle");
        assertOptions("a", "middle", "b", "first");

        click("Add last");
        assertOptions("a", "middle", "b", "first", "last");

        click("Remove middle");
        assertOptions("a", "middle", "first", "last");

        click("Reset");
        assertOptions("a", "b", "c");
    }

    private void assertOptions(String... options) {
        assertEquals(Arrays.asList(options),
                $(ListSelectElement.class).first().getOptions());
    }

    private void click(String caption) {
        $(ButtonElement.class).caption(caption).first().click();
    }
}