-
-
Notifications
You must be signed in to change notification settings - Fork 65
/
FlatMappingExamples.java
62 lines (52 loc) · 1.94 KB
/
FlatMappingExamples.java
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package net.jqwik.docs;
import java.util.*;
import org.assertj.core.api.*;
import net.jqwik.api.*;
import net.jqwik.api.Tuple.*;
class FlatMappingExamples {
@Property
boolean arbitrarySizedStrings(@ForAll("arbitraryString") String aString) {
int length = aString.length();
// Should fail and shrink to ["aaa"]
return length > 6 || length < 3;
}
@Provide
Arbitrary<String> arbitraryString() {
return Arbitraries.integers().between(1, 10) //
.flatMap(size -> Arbitraries.strings().withCharRange('a', 'z') //
.ofMinLength(size).ofMaxLength(size));
}
@Property @Report(Reporting.GENERATED)
boolean fixedSizedStringLists(@ForAll("listsOfEqualSizedStrings") List<String> strings) {
Assume.that(!strings.isEmpty());
return strings.stream().map(String::length).distinct().count() == 1;
}
@Provide
Arbitrary<List<String>> listsOfEqualSizedStrings() {
Arbitrary<Integer> integers2to5 = Arbitraries.integers().between(2, 5);
return integers2to5.flatMap(stringSize -> {
Arbitrary<String> strings = Arbitraries.strings() //
.withCharRange('a', 'z') //
.ofMinLength(stringSize).ofMaxLength(stringSize);
return strings.list();
});
}
@Property @Report(Reporting.GENERATED)
void substringLength(@ForAll("stringWithBeginEnd") Tuple3<String, Integer, Integer> stringBeginEnd) {
String aString = stringBeginEnd.get1();
int begin = stringBeginEnd.get2();
int end = stringBeginEnd.get3();
Assertions.assertThat(aString.substring(begin, end).length()) //
.isEqualTo(end - begin);
}
@Provide
Arbitrary<Tuple3<String, Integer, Integer>> stringWithBeginEnd() {
Arbitrary<String> stringArbitrary = Arbitraries.strings() //
.withCharRange('a', 'z') //
.ofMinLength(2).ofMaxLength(20);
return stringArbitrary //
.flatMap(aString -> Arbitraries.integers().between(0, aString.length()) //
.flatMap(end -> Arbitraries.integers().between(0, end) //
.map(begin -> Tuple.of(aString, begin, end))));
}
}