18
18
import static com .google .datastore .v1 .client .DatastoreHelper .makeFilter ;
19
19
import static com .google .datastore .v1 .client .DatastoreHelper .makeValue ;
20
20
21
+ import com .google .common .truth .Truth ;
21
22
import com .google .datastore .v1 .Filter ;
22
23
import com .google .datastore .v1 .KindExpression ;
23
24
import com .google .datastore .v1 .PartitionId ;
29
30
import java .io .IOException ;
30
31
import java .security .GeneralSecurityException ;
31
32
import java .util .List ;
32
- import org .junit .Assert ;
33
33
import org .junit .Before ;
34
34
import org .junit .Test ;
35
35
@@ -61,10 +61,11 @@ public void testQuerySplitterWithDefaultDb() throws DatastoreException {
61
61
62
62
List <Query > splits =
63
63
DatastoreHelper .getQuerySplitter ().getSplits (query , PARTITION , 2 , DATASTORE );
64
+ Truth .assertThat (splits ).isNotEmpty ();
64
65
splits .forEach (
65
66
split -> {
66
- Assert . assertEquals ( KIND , split .getKind (0 ).getName ());
67
- Assert . assertEquals ( propertyFilter , split .getFilter ());
67
+ Truth . assertThat ( split .getKind (0 ).getName ()). isEqualTo ( KIND );
68
+ Truth . assertThat ( split .getFilter ()). isEqualTo ( propertyFilter );
68
69
});
69
70
}
70
71
@@ -82,10 +83,12 @@ public void testQuerySplitterWithDb() throws DatastoreException {
82
83
83
84
List <Query > splits =
84
85
DatastoreHelper .getQuerySplitter ().getSplits (query , PARTITION , 2 , DATASTORE );
86
+
87
+ Truth .assertThat (splits ).isNotEmpty ();
85
88
splits .forEach (
86
89
split -> {
87
- Assert . assertEquals ( KIND , split .getKind (0 ).getName ());
88
- Assert . assertEquals ( propertyFilter , split .getFilter ());
90
+ Truth . assertThat ( split .getKind (0 ).getName ()). isEqualTo ( KIND );
91
+ Truth . assertThat ( split .getFilter ()). isEqualTo ( propertyFilter );
89
92
});
90
93
}
91
94
}
0 commit comments