18
18
*/
19
19
package neo4j .org .testkit .backend .messages .requests ;
20
20
21
+ import java .util .Objects ;
21
22
import java .util .Set ;
22
23
import java .util .concurrent .CompletionStage ;
23
24
import java .util .function .BiFunction ;
24
- import java .util .function .Supplier ;
25
25
import java .util .stream .Collectors ;
26
26
import lombok .RequiredArgsConstructor ;
27
27
import neo4j .org .testkit .backend .TestkitState ;
@@ -37,20 +37,21 @@ class TestkitBookmarkSupplier implements BookmarkSupplier {
37
37
38
38
@ Override
39
39
public Set <Bookmark > getBookmarks (String database ) {
40
- return getBookmarksFromTestkit (() -> database );
40
+ Objects .requireNonNull (database , "database may not be null" );
41
+ return getBookmarksFromTestkit (database );
41
42
}
42
43
43
44
@ Override
44
45
public Set <Bookmark > getAllBookmarks () {
45
46
return getBookmarksFromTestkit (null );
46
47
}
47
48
48
- private Set <Bookmark > getBookmarksFromTestkit (Supplier < String > databaseSupplier ) {
49
+ private Set <Bookmark > getBookmarksFromTestkit (String database ) {
49
50
var callbackId = testkitState .newId ();
50
51
var bodyBuilder =
51
52
BookmarksSupplierRequest .BookmarksSupplierRequestBody .builder ().id (callbackId );
52
- if (databaseSupplier != null ) {
53
- bodyBuilder = bodyBuilder .database (databaseSupplier . get () );
53
+ if (database != null ) {
54
+ bodyBuilder = bodyBuilder .database (database );
54
55
}
55
56
var callback =
56
57
BookmarksSupplierRequest .builder ().data (bodyBuilder .build ()).build ();
0 commit comments