Merge pull request #19038 from agolubev/agolubev-#18807-range-Source-for-Java-DSL

+str #18807 add Source.range for Java DSL
This commit is contained in:
Roland Kuhn 2015-12-12 17:38:12 +01:00
commit d5d099725a
2 changed files with 50 additions and 0 deletions

View file

@ -483,6 +483,28 @@ public class SourceTest extends StreamTest {
assertEquals("A", result);
}
@Test
public void mustWorkFromRange() throws Exception {
Future<List<Integer>> f = Source.range(0, 10).grouped(20).runWith(Sink.<List<Integer>> head(), materializer);
final List<Integer> result = Await.result(f, FiniteDuration.create(3, TimeUnit.SECONDS));
assertEquals(11, result.size());
Integer counter = 0;
for (Integer i: result)
assertEquals(i, counter++);
}
@Test
public void mustWorkFromRangeWithStep() throws Exception {
Future<List<Integer>> f = Source.range(0, 10, 2).grouped(20).runWith(Sink.<List<Integer>> head(), materializer);
final List<Integer> result = Await.result(f, FiniteDuration.create(3, TimeUnit.SECONDS));
assertEquals(6, result.size());
Integer counter = 0;
for (Integer i: result) {
assertEquals(i, counter);
counter+=2;
}
}
@Test
public void mustRepeat() throws Exception {
final Future<List<Integer>> f = Source.repeat(42).grouped(10000).runWith(Sink.<List<Integer>> head(), materializer);