diff --git a/documentation/src/test/java/org/hibernate/search/documentation/search/projection/ProjectionDslIT.java b/documentation/src/test/java/org/hibernate/search/documentation/search/projection/ProjectionDslIT.java index a1f3afe025b..e54ee648472 100644 --- a/documentation/src/test/java/org/hibernate/search/documentation/search/projection/ProjectionDslIT.java +++ b/documentation/src/test/java/org/hibernate/search/documentation/search/projection/ProjectionDslIT.java @@ -134,6 +134,36 @@ public void reference() { } ); } + @Test + public void id_object() { + withinSearchSession( searchSession -> { + // tag::id-object[] + List hits = searchSession.search( Book.class ) + .select( f -> f.id() ) + .where( f -> f.matchAll() ) + .fetchHits( 20 ); + // end::id-object[] + assertThat( hits ).containsExactlyInAnyOrder( + BOOK1_ID, BOOK2_ID, BOOK3_ID, BOOK4_ID + ); + } ); + } + + @Test + public void id_int() { + withinSearchSession( searchSession -> { + // tag::id-int[] + List hits = searchSession.search( Book.class ) + .select( f -> f.id( Integer.class ) ) + .where( f -> f.matchAll() ) + .fetchHits( 20 ); + // end::id-int[] + assertThat( hits ).containsExactlyInAnyOrder( + BOOK1_ID, BOOK2_ID, BOOK3_ID, BOOK4_ID + ); + } ); + } + @Test public void entity() { withinSearchSession( searchSession -> {