java.util.stream.TestData#OfLong ( )源码实例Demo

@Test(dataProvider = "LongStreamTestData", dataProviderClass = LongStreamTestDataProvider.class)
public void testLongStream(String name, TestData.OfLong data) {
    for (LongPredicate p : LONG_PREDICATES) {
        setContext("p", p);
        for (Kind kind : Kind.values()) {
            setContext("kind", kind);
            exerciseTerminalOps(data, longKinds.get(kind).apply(p));
            exerciseTerminalOps(data, s -> s.filter(lpFalse), longKinds.get(kind).apply(p));
            exerciseTerminalOps(data, s -> s.filter(lpEven), longKinds.get(kind).apply(p));
        }
    }
}
 

@Test(dataProvider = "LongStreamTestData", dataProviderClass = LongStreamTestDataProvider.class)
public void testOps(String name, TestData.OfLong data) {
    AtomicLong expectedCount = new AtomicLong();
    data.stream().forEach(e -> expectedCount.incrementAndGet());

    withData(data).
            terminal(LongStream::count).
            expectedResult(expectedCount.get()).
            exercise();
}
 

@Test(dataProvider = "LongStream.limit")
public void testLongUnorderedGenerator(String description, UnaryOperator<LongStream> fs) {
    // Source is spliterator of infinite size
    TestData.OfLong generator = TestData.Factory.ofLongSupplier(
            "[1L, 1L, ...]", () -> LongStream.generate(() -> 1));

    withData(generator).
            stream(s -> fs.apply(s.filter(i -> true).unordered())).
            exercise();
}
 

@Test(dataProvider = "LongStream.limit")
public void testLongUnorderedIteration(String description, UnaryOperator<LongStream> fs) {
    // Source is a right-balanced tree of infinite size
    TestData.OfLong iterator = TestData.Factory.ofLongSupplier(
            "[1L, 2L, 3L, ...]", () -> LongStream.iterate(1, i -> i + 1));

    // Ref
    withData(iterator).
            stream(s -> fs.apply(s.unordered())).
            resultAsserter(unorderedAsserter()).
            exercise();
}
 

@Test
public void testLongSingleton() {
    TestData.OfLong data = TestData.Factory.ofLongSupplier("[0, 1)",
                                                           () -> LongStream.of(1));

    withData(data).
            stream(s -> s).
            expectedResult(Collections.singletonList(1L)).
            exercise();

    withData(data).
            stream(s -> s.map(i -> i)).
            expectedResult(Collections.singletonList(1L)).
            exercise();
}
 

private void testLongStreamBuilder(int size, Function<Integer, LongStream> supplier) {
    TestData.OfLong data = TestData.Factory.ofLongSupplier(String.format("[0, %d)", size),
                                                           () -> supplier.apply(size));

    withData(data).
            stream(s -> s).
            expectedResult(LongStream.range(0, size).toArray()).
            exercise();

    withData(data).
            stream(s -> s.map(i -> i)).
            expectedResult(LongStream.range(0, size).toArray()).
            exercise();
}
 

@Test(dataProvider = "LongStreamTestData", dataProviderClass = LongStreamTestDataProvider.class)
public void testOps(String name, TestData.OfLong data) {
    AtomicLong expectedCount = new AtomicLong();
    data.stream().forEach(e -> expectedCount.incrementAndGet());

    withData(data).
            terminal(LongStream::count).
            expectedResult(expectedCount.get()).
            exercise();
}
 

@Test(dataProvider = "LongStreamTestData", dataProviderClass = LongStreamTestDataProvider.class)
public void testLongStream(String name, TestData.OfLong data) {
    for (LongPredicate p : LONG_PREDICATES) {
        setContext("p", p);
        for (Kind kind : Kind.values()) {
            setContext("kind", kind);
            exerciseTerminalOps(data, longKinds.get(kind).apply(p));
            exerciseTerminalOps(data, s -> s.filter(lpFalse), longKinds.get(kind).apply(p));
            exerciseTerminalOps(data, s -> s.filter(lpEven), longKinds.get(kind).apply(p));
        }
    }
}
 

private void testLongStreamBuilder(int size, Function<Integer, LongStream> supplier) {
    TestData.OfLong data = TestData.Factory.ofLongSupplier(String.format("[0, %d)", size),
                                                           () -> supplier.apply(size));

    withData(data).
            stream(s -> s).
            expectedResult(LongStream.range(0, size).toArray()).
            exercise();

    withData(data).
            stream(s -> s.map(i -> i)).
            expectedResult(LongStream.range(0, size).toArray()).
            exercise();
}
 

@Test(dataProvider = "LongStream.limit")
public void testLongUnorderedIteration(String description, UnaryOperator<LongStream> fs) {
    // Source is a right-balanced tree of infinite size
    TestData.OfLong iterator = TestData.Factory.ofLongSupplier(
            "[1L, 2L, 3L, ...]", () -> LongStream.iterate(1, i -> i + 1));

    // Ref
    withData(iterator).
            stream(s -> fs.apply(s.unordered())).
            resultAsserter(unorderedAsserter()).
            exercise();
}
 

private TestData.OfLong longRange(long l, long u) {
    return TestData.Factory.ofLongSupplier(
            String.format("[%d, %d)", l, u),
            () -> LongStream.range(l, u));
}
 

private TestData.OfLong proxiedLongRange(long l, long u) {
    return TestData.Factory.ofLongSupplier(
            String.format("[%d, %d)", l, u),
            () -> StreamSupport.longStream(proxyNotSubsized(LongStream.range(l, u).spliterator()), false));
}
 

TestData.OfLong longRangeData(long start, long end) {
    return TestData.Factory.ofLongSupplier("long range", () -> LongStream.range(start, end));
}
 

TestData.OfLong longRangeClosedData(long start, long end) {
    return TestData.Factory.ofLongSupplier("long rangeClosed", () -> LongStream.rangeClosed(start, end));
}
 

@Test(dataProvider = "LongStreamTestData", dataProviderClass = LongStreamTestDataProvider.class)
public void testLongSpliterators(String name, TestData.OfLong data) {
    for (Function<LongStream, LongStream> f : longStreamFunctions()) {
        SpliteratorTestHelper.testLongSpliterator(() -> f.apply(data.stream()).spliterator());
    }
}
 

@Test(dataProvider = "LongStreamTestData", dataProviderClass = LongStreamTestDataProvider.class)
public void testLongParSpliterators(String name, TestData.OfLong data) {
    for (Function<LongStream, LongStream> f : longStreamFunctions()) {
        SpliteratorTestHelper.testLongSpliterator(() -> f.apply(data.parallelStream()).spliterator());
    }
}
 

TestData.OfLong longRangeClosedData(long start, long end) {
    return TestData.Factory.ofLongSupplier("long rangeClosed", () -> LongStream.rangeClosed(start, end));
}
 

private TestData.OfLong longs() {
    return longRange(0, Long.MAX_VALUE);
}
 

private TestData.OfLong longRange(long l, long u) {
    return TestData.Factory.ofLongSupplier(
            String.format("[%d, %d)", l, u),
            () -> LongStream.range(l, u));
}
 

@Test(dataProvider = "LongStreamTestData", dataProviderClass = LongStreamTestDataProvider.class)
public void testLongParSpliterators(String name, TestData.OfLong data) {
    for (Function<LongStream, LongStream> f : longStreamFunctions()) {
        SpliteratorTestHelper.testLongSpliterator(() -> f.apply(data.parallelStream()).spliterator());
    }
}