Skip to content

Commit 80bc979

Browse files
committed
test: mock sink init context for DorisSinkTest
1 parent f95c1af commit 80bc979

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

flink-doris-connector/flink-doris-connector-flink1/src/test/java/org/apache/doris/flink/sink/DorisSinkTest.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,9 @@
3636
import org.mockito.MockedStatic;
3737

3838
import java.util.Collections;
39+
import java.util.OptionalLong;
40+
41+
import org.apache.flink.metrics.groups.SinkWriterMetricGroup;
3942

4043
import static org.mockito.ArgumentMatchers.any;
4144
import static org.mockito.Mockito.mock;
@@ -57,6 +60,9 @@ public void testDorisSink() {
5760
DorisReadOptions dorisReadOptions = OptionUtils.buildDorisReadOptions();
5861
DorisRecordSerializer<String> serializer = new SimpleStringSerializer();
5962
Sink.InitContext initContext = mock(Sink.InitContext.class);
63+
when(initContext.metricGroup()).thenReturn(mock(SinkWriterMetricGroup.class));
64+
when(initContext.getRestoredCheckpointId()).thenReturn(OptionalLong.empty());
65+
when(initContext.getSubtaskId()).thenReturn(0);
6066

6167
DorisExecutionOptions dorisExecutionOptions =
6268
DorisExecutionOptions.builder().disable2PC().build();

0 commit comments

Comments
 (0)