Sample IN test

Author: samueldlightfoot

spring-data-cassandra/src/test/java/org/springframework/data/cassandra/repository/ReactiveCassandraRepositoryIntegrationTests.java

@@ -21,6 +21,7 @@
 
 import java.util.Arrays;
 import java.util.HashSet;
+import java.util.List;
 import java.util.Set;
 
 import org.junit.jupiter.api.BeforeEach;
@@ -125,6 +126,14 @@ void setUp() throws Exception {
 				.verifyComplete();
 	}
 
+	@Test
+	void shouldFindByIdIn() {
+		// Statement generated does not use a bind marker for the IN clause (uses hard-coded values, causing issues
+		// with prepared statement generation)
+		List<String> ids = Arrays.asList(dave.getId(), carter.getId());
+		repository.findByIdIn(new HashSet<>(ids)).as(StepVerifier::create).expectNextCount(2).verifyComplete();
+	}
+
 	@Test // DATACASS-335
 	void shouldFindByLastName() {
 		repository.findByLastname(dave.getLastname()).as(StepVerifier::create).expectNextCount(2).verifyComplete();
@@ -232,6 +241,8 @@ void shouldApplyExistsProjection() {
 
 	interface UserRepository extends ReactiveCassandraRepository<User, String> {
 
+		Flux<User> findByIdIn(Set<String> ids);
+
 		Flux<User> findByLastname(String lastname);
 
 		Mono<Slice<User>> findByLastname(String firstname, Pageable pageable);