blob: 89bb76abe7068dfa8df702614923d11f833ecf07 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
package com.vaadin.tests.util;
import java.util.Locale;
import java.util.stream.IntStream;
import java.util.stream.Stream;
import com.vaadin.data.provider.CallbackDataProvider;
import com.vaadin.data.provider.Query;
/**
* A data provider for tests that creates "Item n" strings on the fly.
*
* @author Vaadin Ltd
*/
public class ItemDataProvider extends CallbackDataProvider<String, String> {
public ItemDataProvider(int size) {
super(q -> itemStream(q, size).skip(q.getOffset()).limit(q.getLimit()),
q -> size(q, 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);
if (filterText.isEmpty()) {
return stream;
}
return stream.filter(
text -> text.toLowerCase(Locale.US).contains(filterText));
}
private static int size(Query<String, String> q, int size) {
if (!q.getFilter().orElse("").isEmpty()) {
return (int) itemStream(q, size).count();
}
return size;
}
}
|