1 /*
2  * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.
8  *
9  * This code is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12  * version 2 for more details (a copy is included in the LICENSE file that
13  * accompanied this code).
14  *
15  * You should have received a copy of the GNU General Public License version
16  * 2 along with this work; if not, write to the Free Software Foundation,
17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20  * or visit www.oracle.com if you need additional information or have any
21  * questions.
22  */
23 package org.openjdk.tests.java.util.stream;
24 
25 import org.openjdk.testlib.java.util.stream.LambdaTestHelpers;
26 import org.openjdk.testlib.java.util.stream.OpTestCase;
27 import org.openjdk.testlib.java.util.stream.StreamTestDataProvider;
28 import org.openjdk.testlib.java.util.stream.TestData;
29 
30 import java.util.List;
31 import java.util.stream.Stream;
32 import org.testng.annotations.Test;
33 
34 import java.util.HashSet;
35 import java.util.Map;
36 
37 import static org.openjdk.testlib.java.util.stream.LambdaTestHelpers.*;
38 import static java.util.stream.Collectors.groupingBy;
39 import static java.util.stream.Collectors.reducing;
40 
41 /**
42  * ReduceByOpTest
43  *
44  * @author Brian Goetz
45  */
46 @Test
47 public class ReduceByOpTest extends OpTestCase {
48 
49     @Test(dataProvider = "StreamTestData<Integer>", dataProviderClass = StreamTestDataProvider.class)
testOps(String name, TestData.OfRef<Integer> data)50     public void testOps(String name, TestData.OfRef<Integer> data) {
51         Map<Boolean,List<Integer>> gbResult = data.stream().collect(groupingBy(LambdaTestHelpers.forPredicate(pEven, true, false)));
52         Map<Boolean, Integer> result = data.stream().collect(groupingBy(LambdaTestHelpers.forPredicate(pEven, true, false), reducing(0, rPlus)));
53         assertEquals(result.size(), gbResult.size());
54         for (Map.Entry<Boolean, Integer> entry : result.entrySet()) {
55             setContext("entry", entry);
56             Boolean key = entry.getKey();
57             assertEquals(entry.getValue(), data.stream().filter(e -> pEven.test(e) == key).reduce(0, rPlus));
58         }
59 
60         int uniqueSize = data.into(new HashSet<Integer>()).size();
61         Map<Integer, List<Integer>> mgResult = exerciseTerminalOps(data, s -> s.collect(groupingBy(mId)));
62         Map<Integer, Integer> miResult = exerciseTerminalOps(data, s -> s.collect(groupingBy(mId, reducing(0, e -> 1, Integer::sum))));
63         assertEquals(miResult.keySet().size(), uniqueSize);
64         for (Map.Entry<Integer, Integer> entry : miResult.entrySet()) {
65             setContext("entry", entry);
66             assertEquals((int) entry.getValue(), mgResult.get(entry.getKey()).size());
67         }
68     }
69 }
70