@Test
public void testInsertStatement() {
final CassandraSchema schema = new CassandraSchema(KEY_SPACE, TABLE, fields);
final CassandraSinkSchemaManager schemaManagerNoTTL = new CassandraSinkSchemaManager(
schema,
Arrays.asList("country_code", "state_province"),
Collections.EMPTY_LIST,
Optional.of(10000L));
final String insertStmt = schemaManagerNoTTL.generateInsertStmt();
final String expected = "INSERT INTO marmaray.crossfit_gyms ( country_code, state_province, city, "
+ "capacity, gym_name ) VALUES ( ?,?,?,?,? ) USING TTL 10000";
Assert.assertEquals(expected, insertStmt);
final CassandraSinkSchemaManager schemaManagerWithTTL = new CassandraSinkSchemaManager(
schema,
Arrays.asList("country_code", "state_province"),
Collections.EMPTY_LIST,
Optional.absent());
final String insertStmtWithTTL = schemaManagerWithTTL.generateInsertStmt();
Assert.assertEquals(expected.replace("USING TTL 10000", StringTypes.EMPTY), insertStmtWithTTL);
}