Skip to content

Commit f4e767b

Browse files
committed
fix: mock BackendUtil.getInstance in flink2 DorisSinkTest
Same NPE issue as flink1 — mockStatic intercepts all static methods but getInstance was not stubbed, returning null for batch/copy mode.
1 parent 7895574 commit f4e767b

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@
3939
import java.util.Collections;
4040

4141
import static org.mockito.ArgumentMatchers.any;
42+
import static org.mockito.ArgumentMatchers.anyInt;
4243
import static org.mockito.Mockito.mock;
4344
import static org.mockito.Mockito.mockStatic;
4445
import static org.mockito.Mockito.when;
@@ -49,8 +50,15 @@ public class DorisSinkTest {
4950

5051
@Before
5152
public void setUp() throws Exception {
53+
BackendUtil mockBackendUtil = mock(BackendUtil.class);
54+
when(mockBackendUtil.getAvailableBackend()).thenReturn("127.0.0.1:8040");
55+
when(mockBackendUtil.getAvailableBackend(anyInt())).thenReturn("127.0.0.1:8040");
56+
5257
backendUtilMockedStatic = mockStatic(BackendUtil.class);
5358
backendUtilMockedStatic.when(() -> BackendUtil.tryHttpConnection(any())).thenReturn(true);
59+
backendUtilMockedStatic
60+
.when(() -> BackendUtil.getInstance(any(), any(), any(), any()))
61+
.thenReturn(mockBackendUtil);
5462
}
5563

5664
@Test

0 commit comments

Comments
 (0)