From 234516a980fbb1cab28a6b62e4511f4e9a53403d Mon Sep 17 00:00:00 2001 From: avinash Date: Wed, 22 Dec 2021 10:38:37 +0530 Subject: [PATCH] Added test to cover scenario where there is a single ENTRY span in a trace --- .../enrichers/TraceStatsEnricherTest.java | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/test/java/org/hypertrace/traceenricher/enrichment/enrichers/TraceStatsEnricherTest.java b/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/test/java/org/hypertrace/traceenricher/enrichment/enrichers/TraceStatsEnricherTest.java index 72c4fa10f..38389b3b6 100644 --- a/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/test/java/org/hypertrace/traceenricher/enrichment/enrichers/TraceStatsEnricherTest.java +++ b/hypertrace-trace-enricher/hypertrace-trace-enricher-impl/src/test/java/org/hypertrace/traceenricher/enrichment/enrichers/TraceStatsEnricherTest.java @@ -9,6 +9,7 @@ import static org.junit.jupiter.api.Assertions.assertNull; import java.nio.ByteBuffer; +import java.util.Collections; import java.util.HashMap; import org.hypertrace.core.datamodel.Event; import org.hypertrace.core.datamodel.StructuredTrace; @@ -20,6 +21,21 @@ class TraceStatsEnricherTest { private static final String UNIQUE_API_NODES_COUNT = "unique.apis.count"; private static final String API_ID = "API_ID"; + @Test + void headSpanIdIsAddedToTraceAttributeForTraceWithSingleEntrySpan() { + Event aEntryHeadSpanEvent = createEntryEventWithName("aEvent"); // 0 + + StructuredTrace trace = + createTraceWithEventsAndEdges(new Event[] {aEntryHeadSpanEvent}, Collections.emptyMap()); + + TraceStatsEnricher traceStatsEnricher = new TraceStatsEnricher(); + traceStatsEnricher.enrichTrace(trace); + ByteBuffer actualHeadSpanId = aEntryHeadSpanEvent.getEventId(); + assertEquals( + actualHeadSpanId, + trace.getAttributes().getAttributeMap().get(HEAD_EVENT_ID).getBinaryValue()); + } + @Test void headSpanIdIsAddedToTraceAttribute() { Event aEntryHeadSpanEvent = createEntryEventWithName("aEvent"); // 0