@Test
public void testUpdateBulkInsertParallelism() {
final String basePath = "/basePath";
final String tableName = "test-table";
final String schemaStr = getSchema("TS", "RECORD_KEY", 4, 8).toString();
final HoodieConfiguration hoodieConf =
HoodieConfiguration.newBuilder(tableName).withTableName(tableName).withMetricsPrefix("test")
.withBasePath(basePath).withSchema(schemaStr).enableMetrics(false).build();
final HoodieSink mockSink =
spy(new HoodieSink(hoodieConf, mock(HoodieSinkDataConverter.class),
mock(JavaSparkContext.class), HoodieSink.HoodieSinkOp.NO_OP, new NoOpMetadataManager()));
when(mockSink.calculateNewBulkInsertParallelism(anyLong())).thenReturn(18);
Assert.assertTrue(mockSink.updateBulkInsertParallelism(1000));
Assert.assertEquals(18, hoodieConf.getBulkInsertParallelism());
Assert.assertEquals(HoodieConfiguration.DEFAULT_HOODIE_PARALLELISM, hoodieConf.getInsertParallelism());
}