1 /*
2  * Copyright (c) 2014, 2015, 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 
24 /**
25  * @test
26  * @summary Tests counting of streams containing Integer.MAX_VALUE + 1 elements
27  * @bug 8031187 8067969
28  */
29 
30 package org.openjdk.tests.java.util.stream;
31 
32 import java.util.stream.LongStream;
33 
34 import org.testng.annotations.Test;
35 
36 import static org.testng.Assert.*;
37 
38 @Test
39 public class CountLargeTest {
40 
41     static final long EXPECTED_LARGE_COUNT = 1L + Integer.MAX_VALUE;
42 
testRefLarge()43     public void testRefLarge() {
44         // Test known sized stream
45         {
46             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
47                     .mapToObj(e -> null).count();
48             assertEquals(count, EXPECTED_LARGE_COUNT);
49         }
50         // Test unknown sized stream
51         {
52             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
53                     .mapToObj(e -> null).filter(e -> true).count();
54             assertEquals(count, EXPECTED_LARGE_COUNT);
55         }
56     }
57 
testIntLarge()58     public void testIntLarge() {
59         // Test known sized stream
60         {
61             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
62                     .mapToInt(e -> 0).count();
63             assertEquals(count, EXPECTED_LARGE_COUNT);
64         }
65         // Test unknown sized stream
66         {
67             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
68                     .mapToInt(e -> 0).filter(e -> true).count();
69             assertEquals(count, EXPECTED_LARGE_COUNT);
70         }
71     }
72 
testLongLarge()73     public void testLongLarge() {
74         // Test known sized stream
75         {
76             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
77                     .count();
78             assertEquals(count, EXPECTED_LARGE_COUNT);
79         }
80         // Test unknown sized stream
81         {
82             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
83                     .filter(e -> true).count();
84             assertEquals(count, EXPECTED_LARGE_COUNT);
85         }
86     }
87 
testDoubleLarge()88     public void testDoubleLarge() {
89         // Test known sized stream
90         {
91             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
92                     .mapToDouble(e -> 0.0).count();
93             assertEquals(count, EXPECTED_LARGE_COUNT);
94         }
95         // Test unknown sized stream
96         {
97             long count = LongStream.range(0, EXPECTED_LARGE_COUNT)
98                     .mapToDouble(e -> 0.0).filter(e -> true).count();
99             assertEquals(count, EXPECTED_LARGE_COUNT);
100         }
101     }
102 }
103