private void insertWithKeyspaceOnStatement(Cluster cluster, String sessionKeyspace) {
Session session;
if (sessionKeyspace != null) {
session = cluster.connect(sessionKeyspace);
} else {
session = cluster.connect();
}
try {
int id = idCounter.incrementAndGet();
BatchStatement batch = new BatchStatement();
batch.add(
new SimpleStatement("INSERT INTO users (id, id2, name) values (?, 1, 'hello')", id));
batch.add(
new SimpleStatement("INSERT INTO users (id, id2, name) values (?, 2, 'sweet')", id));
batch.add(
new SimpleStatement("INSERT INTO users (id, id2, name) values (?, 3, 'world')", id));
batch.setKeyspace(keyspace2Internal);
assertThat(batch.getKeyspace()).isEqualTo(keyspace2Internal);
session.execute(batch);
validateData(session, id);
} finally {
session.close();
if (cluster != cluster()) {
cluster.close();
}
}
}