public Map<String, Long> getDurationMetrics() {
        return durationMetrics;
    }
@Test
    public void testDurationMetrics() {
        SubTaskInitializationMetricsBuilder subTaskInitializationMetricsBuilder =
                new SubTaskInitializationMetricsBuilder(10);
        subTaskInitializationMetricsBuilder.addDurationMetric("A", 20);
        subTaskInitializationMetricsBuilder.addDurationMetric("B", 10);
        assertThat(subTaskInitializationMetricsBuilder.build(50).getDurationMetrics())
                .containsOnlyKeys("A", "B")
                .containsEntry("A", 20L)
                .containsEntry("B", 10L);
    }