diff --git a/hutool-core/src/test/java/cn/hutool/core/stream/EasyStreamTest.java b/hutool-core/src/test/java/cn/hutool/core/stream/EasyStreamTest.java index 098fc12b7..6312d17e3 100644 --- a/hutool-core/src/test/java/cn/hutool/core/stream/EasyStreamTest.java +++ b/hutool-core/src/test/java/cn/hutool/core/stream/EasyStreamTest.java @@ -208,6 +208,26 @@ public class EasyStreamTest { Assert.assertEquals(Arrays.asList(-1, -1, -1), EasyStream.of(1, 2, 3).parallel().flatMapIdx((e, i) -> EasyStream.of(i)).toList()); } + @Test + public void testPeek(){ + EasyStream.of("one", "two", "three", "four") + .filter(e -> e.length() == 4) + .peek(e -> Assert.assertEquals("four", e)) + .map(String::toUpperCase) + .peek(e -> Assert.assertEquals("FOUR", e)) + .collect(Collectors.toList()); + } + + @Test + public void testPeekIdx(){ + EasyStream.of("one", "two", "three", "four") + .filter(e -> e.length() == 4) + .peekIdx((e,i) -> Assert.assertEquals("four:0", e + ":" + i)) + .map(String::toUpperCase) + .peekIdx((e,i) -> Assert.assertEquals("FOUR:0", e + ":" + i)) + .collect(Collectors.toList()); + } + @Test public void testFlat() { final List list = Arrays.asList(1, 2, 3);