From 381eec655e9c256ad9625f65d8580b0619495ba7 Mon Sep 17 00:00:00 2001 From: Looly Date: Thu, 1 Jun 2023 17:37:04 +0800 Subject: [PATCH] add test --- .../hutool/core/stream/EasyStream.java | 2 +- .../hutool/core/stream/EasyStreamTest.java | 32 +++++++++++++++++++ 2 files changed, 33 insertions(+), 1 deletion(-) diff --git a/hutool-core/src/main/java/org/dromara/hutool/core/stream/EasyStream.java b/hutool-core/src/main/java/org/dromara/hutool/core/stream/EasyStream.java index 006616a5f..4036e899f 100644 --- a/hutool-core/src/main/java/org/dromara/hutool/core/stream/EasyStream.java +++ b/hutool-core/src/main/java/org/dromara/hutool/core/stream/EasyStream.java @@ -84,7 +84,7 @@ public class EasyStream extends AbstractEnhancedWrappedStream emptyBigDecimalAvg = EasyStream.of(bigDecimalEmptyList) .avg(Function.identity(), 2, RoundingMode.HALF_UP); Assertions.assertFalse(emptyBigDecimalAvg.isPresent()); + } + /** + * https://github.com/dromara/hutool/pull/3128 + */ + @Test + void testStreamBigDecimal() { + final List testList = ListUtil.of(); + for (int i = 0; i < 5; i++) { + final BigDecimalTest test = BigDecimalTest.builder() + .name("test" + i) + .count(new BigDecimal(i+1)) + .build(); + testList.add(test); + } + final BigDecimal sum = EasyStream.of(testList).sum(BigDecimalTest::getCount); + Assertions.assertEquals(15, sum.intValue()); + + final BigDecimal avg = EasyStream.of(testList).avg(BigDecimalTest::getCount).get(); + Assertions.assertEquals(3, avg.intValue()); + } + + @Data + @Builder + private static class BigDecimalTest{ + private String name; + private BigDecimal count; } }