Add DataProvider.setFilter method

Change-Id: I87193d3d2c66a880c2be165cfd49926fa290010b

1 files changed, 17 insertions, 0 deletions

diff --git a/server/src/test/java/com/vaadin/server/data/provider/ListDataProviderTest.java b/server/src/test/java/com/vaadin/server/data/provider/ListDataProviderTest.java
index 6a53aae528..29a234061f 100644
--- a/server/src/test/java/com/vaadin/server/data/provider/ListDataProviderTest.java
+++ b/server/src/test/java/com/vaadin/server/data/provider/ListDataProviderTest.java
@@ -11,6 +11,7 @@ import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 
+import com.vaadin.server.SerializablePredicate;
 import com.vaadin.server.data.DataProvider;
 import com.vaadin.server.data.ListDataProvider;
 import com.vaadin.server.data.Query;
@@ -216,4 +217,20 @@ public class ListDataProviderTest {
                 dataProvider.size(new Query<>(
                         strBean -> strBean.getValue().contains("Foo"))));
     }
+
+    @Test
+    public void filteringListDataProvider_defaultFilter() {
+        SerializablePredicate<StrBean> filter = s -> s.getRandomNumber() > 4;
+        // Intentionally lost filter type. Not actually filterable anymore.
+        DataProvider<StrBean, ?> filtered = dataProvider.setFilter(filter);
+
+        Assert.assertEquals("Filter not applied, unexpected item count",
+                dataProvider.size(new Query<>(filter)),
+                filtered.size(new Query<>()));
+
+        Assert.assertEquals("Further filtering succeeded",
+                filtered.size(new Query<>()),
+                filtered.size((Query) new Query<SerializablePredicate<StrBean>>(
+                        s -> s.getValue().equals("Foo"))));
+    }
 }