use of org.apache.samza.metrics.reporter.MetricsSnapshot in project samza by apache.
the class TestDiagnosticsStreamMessage method testSerdeEmptySamzaEpochIdInHeader.
@Test
public void testSerdeEmptySamzaEpochIdInHeader() {
DiagnosticsStreamMessage diagnosticsStreamMessage = getDiagnosticsStreamMessage(Optional.empty());
MetricsSnapshot metricsSnapshot = diagnosticsStreamMessage.convertToMetricsSnapshot();
MetricsHeader expectedHeader = new MetricsHeader(JOB_NAME, JOB_ID, CONTAINER_NAME, EXECUTION_ENV_CONTAINER_ID, Optional.empty(), DiagnosticsManager.class.getName(), TASK_CLASS_VERSION, SAMZA_VERSION, HOSTNAME, timestamp, resetTimestamp);
Assert.assertEquals(metricsSnapshot.getHeader(), expectedHeader);
DiagnosticsStreamMessage convertedDiagnosticsStreamMessage = DiagnosticsStreamMessage.convertToDiagnosticsStreamMessage(metricsSnapshot);
Assert.assertEquals(convertedDiagnosticsStreamMessage, diagnosticsStreamMessage);
}
use of org.apache.samza.metrics.reporter.MetricsSnapshot in project samza by apache.
the class TestDiagnosticsManager method validateMetricsHeader.
private void validateMetricsHeader(OutgoingMessageEnvelope outgoingMessageEnvelope, long sendTime) {
// Validate the outgoing message
Assert.assertTrue(outgoingMessageEnvelope.getSystemStream().equals(diagnosticsSystemStream));
MetricsSnapshot metricsSnapshot = new MetricsSnapshotSerdeV2().fromBytes((byte[]) outgoingMessageEnvelope.getMessage());
MetricsHeader expectedHeader = new MetricsHeader(JOB_NAME, JOB_ID, "samza-container-0", EXECUTION_ENV_CONTAINER_ID, Optional.of(SAMZA_EPOCH_ID), DiagnosticsManager.class.getName(), TASK_CLASS_VERSION, SAMZA_VERSION, HOSTNAME, sendTime, RESET_TIME);
Assert.assertEquals(expectedHeader, metricsSnapshot.getHeader());
}
Aggregations