iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update unit test for collectors.toMap method

This commit is contained in:
joe zhang 2020-06-29 22:34:19 +08:00
parent 1c61ba62ab
commit 672ffdbe9f
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
java-collections-conversions-2/src/test/java/com/baeldung/convertlisttomap/ListToMapUnitTest.java
View File

@ -9,6 +9,8 @@ import java.util.HashMap;
import java.util.List; import java.util.List;
import java.util.Map; import java.util.Map;
import java.util.function.BiConsumer; import java.util.function.BiConsumer;
import java.util.function.BinaryOperator;
import java.util.function.Function;
import java.util.function.Supplier; import java.util.function.Supplier;
import java.util.stream.Collectors; import java.util.stream.Collectors;
@ -84,15 +86,23 @@ public class ListToMapUnitTest {
return new ArrayList<>(); return new ArrayList<>();
}; };
Function<String, Integer> keyMapper = (element) -> {
return element.length();
};
Function<String, List<String>> valueMapper = (element) -> {
List<String> collection = listSupplier.get();
collection.add(element);
return collection;
};
BinaryOperator<List<String>> mergeFunction = (existing, replacement) -> {
existing.addAll(replacement);
return existing;
};
convertedMap = strings.stream() convertedMap = strings.stream()
.collect(Collectors.toMap(String::length, (p) -> { .collect(Collectors.toMap(keyMapper, valueMapper, mergeFunction, mapFactory));
List<String> strs = listSupplier.get();
strs.add(p);
return strs;
}, (existing, replacement) -> {
existing.addAll(replacement);
return existing;
}, mapFactory));
assertEquals(2, convertedMap.size()); assertEquals(2, convertedMap.size());
assertTrue(convertedMap.get(3) assertTrue(convertedMap.get(3)