@Test public void testCreateJobLongMetric() { Metric jobMetric1 = jobMetrics.createJobLongMetric(JobMetricType.DRIVER_MEMORY, 100l, Maps.newHashMap()); Metric jobMetric2 = jobMetrics.createJobLongMetric(JobMetricType.SIZE, 1028l, Maps.newHashMap()); assertEquals("job-driver_memory", jobMetric1.getMetricName()); assertEquals("job-size", jobMetric2.getMetricName()); assertEquals(new Long(100), jobMetric1.getMetricValue()); assertEquals(new Long(1028), jobMetric2.getMetricValue()); assertTrue(jobMetric1.getTags().containsKey("job")); assertEquals(jobMetric1.getTags().get("job"), "test-job"); assertFalse(jobMetric1.getTags().containsKey("topic")); assertFalse(jobMetric1.getTags().containsKey("hello")); jobMetric1.addTag("hello", "world"); assertTrue(jobMetric1.getTags().containsKey("hello")); assertTrue(jobMetrics.getMetricSet().contains(jobMetric1)); assertTrue(jobMetrics.getMetricSet().contains(jobMetric2)); }
@Test public void testCreateJobTimerMetric() { Metric jobMetric1 = jobMetrics.createJobTimerMetric(JobMetricType.RUNTIME, Maps.newHashMap()); Metric jobMetric2 = jobMetrics.createJobTimerMetric(JobMetricType.STAGE_RUNTIME, Maps.newHashMap()); assertEquals("job-runtime", jobMetric1.getMetricName()); assertEquals("job-stage_runtime", jobMetric2.getMetricName()); assertTrue(jobMetric1 instanceof TimerMetric); assertTrue(jobMetric2 instanceof TimerMetric); assertTrue(jobMetric1.getTags().containsKey("job")); assertEquals(jobMetric1.getTags().get("job"), "test-job"); assertFalse(jobMetric1.getTags().containsKey("topic")); assertFalse(jobMetric1.getTags().containsKey("hello")); jobMetric1.addTag("hello", "world"); assertTrue(jobMetric1.getTags().containsKey("hello")); assertTrue(jobMetrics.getMetricSet().contains(jobMetric1)); assertTrue(jobMetrics.getMetricSet().contains(jobMetric2)); }