[GitHub] flink pull request #3748: [FLINK-6225] [Cassandra Connector] add CassandraTa...

[PangZhi <gi...@git.apache.org>]

Github user PangZhi commented on a diff in the pull request:

    https://github.com/apache/flink/pull/3748#discussion_r143098834
  
    --- Diff: flink-connectors/flink-connector-cassandra/src/test/java/org/apache/flink/streaming/connectors/cassandra/CassandraConnectorITCase.java ---
    @@ -392,6 +425,45 @@ public void testCassandraPojoAtLeastOnceSink() throws Exception {
     	}
     
     	@Test
    +	public void testCassandraTableSink() throws Exception {
    +		StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
    +		env.setParallelism(1);
    +
    +		DataStreamSource<Row> source = env.fromCollection(rowCollection);
    +		CassandraTableSink cassandraTableSink = new CassandraTableSink(new ClusterBuilder() {
    +			@Override
    +			protected Cluster buildCluster(Cluster.Builder builder) {
    +				return builder.addContactPointsWithPorts(new InetSocketAddress(HOST, PORT)).build();
    +			}
    +		}, injectTableName(INSERT_DATA_QUERY), new Properties());
    +		CassandraTableSink newCassandrTableSink = cassandraTableSink.configure(FIELD_NAMES, FIELD_TYPES);
    +
    +		newCassandrTableSink.emitDataStream(source);
    +
    +		env.execute();
    +		ResultSet rs = session.execute(injectTableName(SELECT_DATA_QUERY));
    +		Assert.assertEquals(20, rs.all().size());
    +	}
    +
    +	@Test
    +	public void testCassandraTableSinkE2E() throws Exception {
    --- End diff --
    
    this is added as another reviewer think better to add e2e test using sql api.


---