public class FluentIterableExampleTest
{

    private FluentIterable<String> build()
    {
        ArrayList<String> list = Lists.newArrayList("Alex", "Wang", "Guava", "Scala");
        return FluentIterable.from(list);
    }

    @Test
    public void testFilter()
    {
        FluentIterable<String> fit = build();
        assertThat(fit.size(), equalTo(4));

        FluentIterable<String> result = fit.filter(e -> e != null && e.length() > 4);
        assertThat(result.size(), equalTo(2));
    }

    @Test
    public void testAppend()
    {
        FluentIterable<String> fit = build();
        ArrayList<String> append = Lists.newArrayList("APPEND");
        assertThat(fit.size(), equalTo(4));
        FluentIterable<String> appendFI = fit.append(append);
        assertThat(appendFI.size(), equalTo(5));
        assertThat(appendFI.contains("APPEND"), is(true));

        appendFI = appendFI.append("APPEND2");
        assertThat(appendFI.size(), equalTo(6));
        assertThat(appendFI.contains("APPEND2"), is(true));
        assertThat(appendFI.contains("Alex"), is(true));
    }

    @Test
    public void testMatch()
    {
        FluentIterable<String> fit = build();
        boolean result = fit.allMatch(e -> e != null && e.length() >= 4);
        assertThat(result, is(true));

        result = fit.anyMatch(e -> e != null && e.length() == 5);
        assertThat(result, is(true));

        Optional<String> optional = fit.firstMatch(e -> e != null && e.length() == 5);
        assertThat(optional.isPresent(), is(true));
        assertThat(optional.get(), equalTo("Guava"));
    }

    @Test
    public void testFirst$Last()
    {
        FluentIterable<String> fit = build();
        Optional<String> optional = fit.first();
        assertThat(optional.isPresent(), is(true));
        assertThat(optional.get(), equalTo("Alex"));

        optional = fit.last();
        assertThat(optional.isPresent(), is(true));
        assertThat(optional.get(), equalTo("Scala"));
    }

    @Test
    public void testLimit()
    {
        FluentIterable<String> fit = build();
        FluentIterable<String> limit = fit.limit(3);
        System.out.println(limit);
        assertThat(limit.contains("Scala"), is(false));

        limit = fit.limit(300);
        System.out.println(limit);
        assertThat(limit.contains("Scala"), is(true));

    }

    /**
     * DSL
     */
    @Test
    public void testCopyIn()
    {
        FluentIterable<String> fit = build();
        ArrayList<String> list = Lists.newArrayList("Java");
        ArrayList<String> result = fit.copyInto(list);

        assertThat(result.size(), equalTo(5));
        assertThat(result.contains("Scala"), is(true));
    }

    @Test
    public void testCycle()
    {
        FluentIterable<String> fit = build();
        FluentIterable<String> cycle = fit.cycle().limit(20);
        cycle.forEach(System.out::println);
    }

    @Test
    public void testTransform()
    {
        FluentIterable<String> fit = build();
        fit.transform(e -> e.length()).forEach(System.out::println);
    }

    @Test
    public void testTransformAndConcat()
    {
        FluentIterable<String> fit = build();
        List<Integer> list = Lists.newArrayList(1);
        FluentIterable<Integer> result = fit.transformAndConcat(e -> list);
        result.forEach(System.out::println);
    }

    /**
     * A----->API---->B(Server)
     * 1,2
     */
    @Test
    public void testTransformAndConcatInAction()
    {
        ArrayList<Integer> cTypes = Lists.newArrayList(1, 2);
        FluentIterable.from(cTypes).transformAndConcat(this::search)
                .forEach(System.out::println);
    }

    @Test
    public void testJoin()
    {
        FluentIterable<String> fit = build();
        String result = fit.join(Joiner.on(','));
        assertThat(result, equalTo("Alex,Wang,Guava,Scala"));
    }


    private List<Customer> search(int type)
    {
        if (type == 1)
        {
            return Lists.newArrayList(new Customer(type, "Alex"), new Customer(type, "Tina"));
        } else
        {
            return Lists.newArrayList(new Customer(type, "Wang"), new Customer(type, "Wen"), new Customer(type, "Jun"));
        }
    }

    class Customer
    {
        final int type;
        final String name;

        Customer(int type, String name)
        {
            this.type = type;
            this.name = name;
        }

        @Override
        public String toString()
        {
            return "Customer{" +
                    "type=" + type +
                    ", name='" + name + '\'' +
                    '}';
        }
    }
}