@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")));
  }