#1823 - ENH: Add query.findStream() and query.findLargeStream()

src/test/java/org/tests/query/TestQueryFindStream.java

@@ -0,0 +1,71 @@
+package org.tests.query;
+
+import io.ebean.BaseTestCase;
+import io.ebean.DB;
+import org.junit.Test;
+import org.tests.model.basic.Customer;
+import org.tests.model.basic.ResetBasicData;
+
+import java.util.List;
+import java.util.stream.Stream;
+
+import static java.util.stream.Collectors.toList;
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class TestQueryFindStream extends BaseTestCase {
+
+  @Test
+  public void findStream_basic() {
+
+    ResetBasicData.reset();
+
+    try (Stream<Customer> stream = DB.find(Customer.class)
+      .findStream()) {
+
+      // bad example, don't use a stream like this when we can
+      // use findSingleAttributeList() instead
+      final List<String> namesStream = stream
+        .map(Customer::getName)
+        .collect(toList());
+
+      final List<String> namesQuery = DB.find(Customer.class)
+        .select("name")
+        .findSingleAttributeList();
+
+      assertThat(namesStream).hasSize(namesQuery.size());
+      assertThat(namesStream).containsAll(namesQuery);
+    }
+  }
+
+  @Test
+  public void findLargeStream_basic() {
+
+    ResetBasicData.reset();
+
+    try (Stream<Customer> stream = DB.find(Customer.class)
+      .findLargeStream()) {
+
+      // bad example, don't use a stream like this when we can
+      // use findSingleAttributeList() instead
+      final List<String> namesStream = stream
+        .map(Customer::getName)
+        .collect(toList());
+
+      final List<String> namesQuery = DB.find(Customer.class)
+        .select("name")
+        .findSingleAttributeList();
+
+      assertThat(namesStream).hasSize(namesQuery.size());
+      assertThat(namesStream).containsAll(namesQuery);
+    }
+  }
+
+  @Test
+  public void manualTest_findSteam_when_streamNotClosed_connectionLeak() {
+
+    Stream<Customer> stream = DB.find(Customer.class).findStream();
+    // remember a steam MUST be closed or we leak resources
+    // comment out the close(); below to leak a connection
+    stream.close();
+  }
+}