Provide in-memory sorting information in Query

Change-Id: Iebafff6079816c08e5a4d144f6891d1379751f12

1 files changed, 3 insertions, 2 deletions

diff --git a/uitest/src/main/java/com/vaadin/tests/util/ItemDataProvider.java b/uitest/src/main/java/com/vaadin/tests/util/ItemDataProvider.java
index f050d1f454..5f02135591 100644
--- a/uitest/src/main/java/com/vaadin/tests/util/ItemDataProvider.java
+++ b/uitest/src/main/java/com/vaadin/tests/util/ItemDataProvider.java
@@ -19,7 +19,8 @@ public class ItemDataProvider extends BackEndDataProvider<String, String> {
                 q -> size(q, size));
     }
 
-    private static Stream<String> itemStream(Query<String> q, int size) {
+    private static Stream<String> itemStream(Query<String, String> q,
+            int size) {
         Stream<String> stream = IntStream.range(0, size)
                 .mapToObj(i -> "Item " + i);
         String filterText = q.getFilter().orElse("").toLowerCase(Locale.US);
@@ -32,7 +33,7 @@ public class ItemDataProvider extends BackEndDataProvider<String, String> {
                 text -> text.toLowerCase(Locale.US).contains(filterText));
     }
 
-    private static int size(Query<String> q, int size) {
+    private static int size(Query<String, String> q, int size) {
         if (!q.getFilter().orElse("").isEmpty()) {
             return (int) itemStream(q, size).count();
         }