@Test public void testCassandraBatchInsertAndSelectStatement() throws Exception { List<Book> books = getBookList(5); this.cassandraMessageHandler2.handleMessage(new GenericMessage<>(books)); Message<?> message = MessageBuilder.withPayload("Cassandra Guru").setHeader("limit", 2).build(); this.cassandraMessageHandler4.handleMessage(message); Message<?> receive = this.resultChannel.receive(10000); assertNotNull(receive); assertThat(receive.getPayload(), instanceOf(ResultSet.class)); ResultSet resultSet = (ResultSet) receive.getPayload(); assertNotNull(resultSet); List<Row> rows = resultSet.all(); assertEquals(2, rows.size()); this.cassandraMessageHandler1.handleMessage( new GenericMessage<>(QueryBuilder.truncate("book"))); }