152 lines
4.2 KiB
Java
152 lines
4.2 KiB
Java
package com.baeldung;
|
|
|
|
|
|
import org.jooq.lambda.Seq;
|
|
import org.junit.Test;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.HashMap;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import static junit.framework.Assert.assertTrue;
|
|
import static junit.framework.TestCase.assertEquals;
|
|
import static org.jooq.lambda.tuple.Tuple.tuple;
|
|
|
|
public class JOOLTest {
|
|
@Test
|
|
public void givenSeq_whenCheckContains_shouldReturnTrue() {
|
|
List<Integer> concat = Seq.of(1, 2, 3).concat(Seq.of(4, 5, 6)).toList();
|
|
|
|
assertEquals(concat, Arrays.asList(1, 2, 3, 4, 5, 6));
|
|
|
|
|
|
assertTrue(Seq.of(1, 2, 3, 4).contains(2));
|
|
|
|
|
|
assertTrue(Seq.of(1, 2, 3, 4).containsAll(2, 3));
|
|
|
|
|
|
assertTrue(Seq.of(1, 2, 3, 4).containsAny(2, 5));
|
|
}
|
|
|
|
@Test
|
|
public void givenSeq_whenJoin_shouldHaveElementsFromBothSeq() {
|
|
assertEquals(
|
|
Seq.of(1, 2).crossJoin(Seq.of("A", "B")).toList(),
|
|
Arrays.asList(tuple(1, "A"), tuple(1, "B"), tuple(2, "A"), tuple(2, "B"))
|
|
);
|
|
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 4).innerJoin(Seq.of(1, 2, 3), (a, b) -> a == b).toList(),
|
|
Arrays.asList(tuple(1, 1), tuple(2, 2))
|
|
);
|
|
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 4).leftOuterJoin(Seq.of(1, 2, 3), (a, b) -> a == b).toList(),
|
|
Arrays.asList(tuple(1, 1), tuple(2, 2), tuple(4, null))
|
|
);
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 4).rightOuterJoin(Seq.of(1, 2, 3), (a, b) -> a == b).toList(),
|
|
Arrays.asList(tuple(1, 1), tuple(2, 2), tuple(null, 3))
|
|
);
|
|
}
|
|
|
|
@Test
|
|
public void givenSeq_whenManipulateSeq_seqShouldHaveNewElementsInIt() {
|
|
assertEquals(
|
|
Seq.of(1, 2, 3).cycle().limit(9).toList(),
|
|
Arrays.asList(1, 2, 3, 1, 2, 3, 1, 2, 3)
|
|
);
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3).duplicate().map((first, second) -> tuple(first.toList(), second.toList())),
|
|
tuple(Arrays.asList(1, 2, 3), Arrays.asList(1, 2, 3))
|
|
);
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3, 4).intersperse(0).toList(),
|
|
Arrays.asList(1, 0, 2, 0, 3, 0, 4)
|
|
);
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3, 4, 5).shuffle().toList().size(),
|
|
5
|
|
);
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3, 4).partition(i -> i % 2 != 0).map((first, second) -> tuple(first.toList(), second.toList())),
|
|
tuple(Arrays.asList(1, 3), Arrays.asList(2, 4))
|
|
|
|
);
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3, 4).reverse().toList(),
|
|
Arrays.asList(4, 3, 2, 1)
|
|
);
|
|
}
|
|
|
|
@Test
|
|
public void givenSeq_whenGroupByAndFold_shouldReturnProperSeq() {
|
|
|
|
Map<Integer, List<Integer>> expectedAfterGroupBy = new HashMap<>();
|
|
expectedAfterGroupBy.put(1, Arrays.asList(1, 3));
|
|
expectedAfterGroupBy.put(0, Arrays.asList(2, 4));
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3, 4).groupBy(i -> i % 2),
|
|
expectedAfterGroupBy
|
|
);
|
|
|
|
|
|
assertEquals(
|
|
Seq.of("a", "b", "c").foldLeft("!", (u, t) -> u + t),
|
|
"!abc"
|
|
);
|
|
|
|
|
|
assertEquals(
|
|
Seq.of("a", "b", "c").foldRight("!", (t, u) -> t + u),
|
|
"abc!"
|
|
);
|
|
}
|
|
|
|
@Test
|
|
public void givenSeq_whenUsingSeqWhile_shouldBehaveAsWhileLoop() {
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3, 4, 5).skipWhile(i -> i < 3).toList(),
|
|
Arrays.asList(3, 4, 5)
|
|
);
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3, 4, 5).skipUntil(i -> i == 3).toList(),
|
|
Arrays.asList(3, 4, 5)
|
|
);
|
|
}
|
|
|
|
@Test
|
|
public void givenSeq_whenZip_shouldHaveZippedSeq() {
|
|
|
|
assertEquals(
|
|
Seq.of(1, 2, 3).zip(Seq.of("a", "b", "c")).toList(),
|
|
Arrays.asList(tuple(1, "a"), tuple(2, "b"), tuple(3, "c"))
|
|
);
|
|
|
|
// ("1:a", "2:b", "3:c")
|
|
assertEquals(
|
|
Seq.of(1, 2, 3).zip(Seq.of("a", "b", "c"), (x, y) -> x + ":" + y).toList(),
|
|
Arrays.asList("1:a", "2:b", "3:c")
|
|
);
|
|
|
|
|
|
assertEquals(
|
|
Seq.of("a", "b", "c").zipWithIndex().toList(),
|
|
Arrays.asList(tuple("a", 0L), tuple("b", 1L), tuple("c", 2L))
|
|
);
|
|
}
|
|
}
|