@Test @SuppressWarnings("unchecked") public void testKafkaTableSink() throws Exception { DataStream dataStream = mock(DataStream.class); KafkaTableSink kafkaTableSink = createTableSink(); kafkaTableSink.emitDataStream(dataStream); verify(dataStream).addSink(kafkaProducer); }
@Test @SuppressWarnings("unchecked") public void testCreatedProducer() throws Exception { DataStream dataStream = mock(DataStream.class); KafkaTableSink kafkaTableSink = spy(createTableSink()); kafkaTableSink.emitDataStream(dataStream); verify(kafkaTableSink) .createKafkaProducer( eq(TOPIC), eq(properties), any(getSerializationSchema()), eq(partitioner)); }
@Test public void testConfiguration() { KafkaTableSink kafkaTableSink = createTableSink(); KafkaTableSink newKafkaTableSink = kafkaTableSink.configure(FIELD_NAMES, FIELD_TYPES); assertNotSame(kafkaTableSink, newKafkaTableSink); assertArrayEquals(FIELD_NAMES, newKafkaTableSink.getFieldNames()); assertArrayEquals(FIELD_TYPES, newKafkaTableSink.getFieldTypes()); assertEquals(new RowTypeInfo(FIELD_TYPES), newKafkaTableSink.getOutputType()); }