FindByOfListTest.java

package org.fluentlenium.integration;

import org.fluentlenium.core.FluentPage;
import org.fluentlenium.core.annotation.Page;
import org.fluentlenium.core.domain.FluentList;
import org.fluentlenium.core.domain.FluentWebElement;
import org.fluentlenium.integration.localtest.IntegrationFluentTest;
import org.junit.Test;
import org.openqa.selenium.support.FindAll;
import org.openqa.selenium.support.FindBy;

import static org.assertj.core.api.Assertions.assertThat;

public class FindByOfListTest extends IntegrationFluentTest {

    @Page
    private PageIndex page;

    @Test
    public void shouldFindByRetrieveList() {
        page.go();
        page.isAt();
        assertThat(page.smalls).hasSize(3);
        assertThat(page.smalls.texts()).containsExactly("Small 1", "Small 2", "Small 3");
    }

    @Test
    public void shouldFindAllFindByRetrieveList() {
        page.go();
        page.isAt();
        assertThat(page.findAllElements).hasSize(4);
        assertThat(page.findAllElements.texts()).containsExactly("Pharmacy", "Small 1", "Small 2", "Small 3");
    }

    private static class PageIndex extends FluentPage {

        @FindBy(className = "small")
        private FluentList<FluentWebElement> smalls;

        @FindAll({@FindBy(id = "location"), @FindBy(className = "small")})
        private FluentList<FluentWebElement> findAllElements;

        @Override
        public String getUrl() {
            return IntegrationFluentTest.DEFAULT_URL;
        }

        @Override
        public void isAt() {
            assertThat(getDriver().getTitle()).contains("Selenium");
        }
    }
}