@Test
public void getMapForNull() {
MetricFetcher fetcher =
new MetricFetcher(
mock(ActorSystem.class), mock(JobManagerRetriever.class), mock(ExecutionContext.class));
MetricStore store = fetcher.getMetricStore();
TaskManagerMetricsHandler handler = new TaskManagerMetricsHandler(fetcher);
Map<String, String> pathParams = new HashMap<>();
Map<String, Object> metrics = handler.getMapFor(pathParams, store);
assertNull(metrics);
}
@Test
public void getMapFor() throws Exception {
MetricFetcher fetcher =
new MetricFetcher(
mock(ActorSystem.class), mock(JobManagerRetriever.class), mock(ExecutionContext.class));
MetricStore store = MetricStoreTest.setupStore(fetcher.getMetricStore());
TaskManagerMetricsHandler handler = new TaskManagerMetricsHandler(fetcher);
Map<String, String> pathParams = new HashMap<>();
pathParams.put(PARAMETER_TM_ID, "tmid");
Map<String, Object> metrics = handler.getMapFor(pathParams, store);
assertEquals(1L, metrics.get("abc.metric2"));
}