ElementListInstanceLocatorTest.java

package org.fluentlenium.core.proxy;

import org.assertj.core.api.Assertions;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.openqa.selenium.WebElement;

import java.util.Arrays;
import java.util.Collections;

@RunWith(MockitoJUnitRunner.class)
public class ElementListInstanceLocatorTest {
    @Mock
    private WebElement element1;

    @Mock
    private WebElement element2;

    @Mock
    private WebElement element3;

    @Mock
    private WebElement element4;

    @Test
    public void testWithMockElements() {
        ElementListInstanceLocator locator = new ElementListInstanceLocator(
                Arrays.asList(element1, element2, element3, element4));

        Assertions.assertThat(locator.findElement()).isSameAs(element1);
        Assertions.assertThat(locator.findElements()).containsExactly(element1, element2, element3, element4);

        Assertions.assertThat(locator.getWrappedElements()).containsExactly(element1, element2, element3, element4);
    }

    @Test
    public void testWithNoElement() {
        ElementListInstanceLocator locator = new ElementListInstanceLocator(Collections.<WebElement>emptyList());

        Assertions.assertThat(locator.findElement()).isNull();
        Assertions.assertThat(locator.findElements()).isEmpty();

        Assertions.assertThat(locator.getWrappedElements()).isEmpty();
    }
}