add global labels values test case

Add an e2e test for global labels values with multiple events. A batch of 2 events in input with different labels Values but same key should produce in output events with all different Values added to the same key.
elastic · Aug 14, 2024 · 57265b3 · 57265b3
1 parent 6c0a6ab
commit 57265b3
Showing 1 changed file with 134 additions and 0 deletions.
diff --git a/aggregators/aggregator_test.go b/aggregators/aggregator_test.go
@@ -1242,6 +1242,140 @@ func TestAggregateAndHarvest(t *testing.T) {
 		runTest(t, input, expected)
 	})
 
+	t.Run("with multiple global labels values", func(t *testing.T) {
+		input := modelpb.Batch{
+			{
+				Event: &modelpb.Event{
+					Outcome:  "success",
+					Duration: uint64(txnDuration),
+				},
+				Transaction: &modelpb.Transaction{
+					Name:                "foo",
+					Type:                "txtype",
+					RepresentativeCount: 1,
+				},
+				Service: &modelpb.Service{Name: "svc"},
+				Labels: modelpb.Labels{
+					"baz": &modelpb.LabelValue{Global: true, Values: []string{"asd", "qwe"}},
+				},
+				NumericLabels: modelpb.NumericLabels{
+					"bar": &modelpb.NumericLabelValue{Global: true, Values: []float64{1, 2}},
+				},
+			},
+			{
+				Event: &modelpb.Event{
+					Outcome:  "success",
+					Duration: uint64(txnDuration),
+				},
+				Transaction: &modelpb.Transaction{
+					Name:                "foo",
+					Type:                "txtype",
+					RepresentativeCount: 1,
+				},
+				Service: &modelpb.Service{Name: "svc"},
+				Labels: modelpb.Labels{
+					"baz": &modelpb.LabelValue{Global: true, Values: []string{"rty", "fgh"}},
+				},
+				NumericLabels: modelpb.NumericLabels{
+					"bar": &modelpb.NumericLabelValue{Global: true, Values: []float64{4, 5}},
+				},
+			},
+		}
+
+		expected := []*modelpb.APMEvent{
+			{
+				Timestamp: modelpb.FromTime(time.Unix(0, 0).UTC()),
+				Event: &modelpb.Event{
+					SuccessCount: &modelpb.SummaryMetric{
+						Count: 1,
+						Sum:   1,
+					},
+					Outcome: "success",
+				},
+				Transaction: &modelpb.Transaction{
+					Name: "foo",
+					Type: "txtype",
+					Root: true,
+					DurationSummary: &modelpb.SummaryMetric{
+						Count: 1,
+						Sum:   100351, // Estimate from histogram
+					},
+					DurationHistogram: &modelpb.Histogram{
+						Values: []float64{100351},
+						Counts: []uint64{1},
+					},
+				},
+				Service: &modelpb.Service{
+					Name: "svc",
+				},
+				Labels: modelpb.Labels{
+					"baz": &modelpb.LabelValue{Global: true, Values: []string{"asd", "qwe", "rty", "fgh"}},
+				},
+				NumericLabels: modelpb.NumericLabels{
+					"bar": &modelpb.NumericLabelValue{Global: true, Values: []float64{1, 2, 4, 5}},
+				},
+				Metricset: &modelpb.Metricset{
+					Name:     "transaction",
+					DocCount: 1,
+					Interval: "1s",
+				},
+			},
+			{
+				Timestamp: modelpb.FromTime(time.Unix(0, 0).UTC()),
+				Event:     &modelpb.Event{},
+				Service: &modelpb.Service{
+					Name: "svc",
+				},
+				Labels: modelpb.Labels{
+					"baz": &modelpb.LabelValue{Global: true, Values: []string{"asd", "qwe", "rty", "fgh"}},
+				},
+				NumericLabels: modelpb.NumericLabels{
+					"bar": &modelpb.NumericLabelValue{Global: true, Values: []float64{1, 2, 4, 5}},
+				},
+				Metricset: &modelpb.Metricset{
+					Name:     "service_summary",
+					Interval: "1s",
+				},
+			},
+			{
+				Timestamp: modelpb.FromTime(time.Unix(0, 0).UTC()),
+				Event: &modelpb.Event{
+					SuccessCount: &modelpb.SummaryMetric{
+						Count: 1,
+						Sum:   1,
+					},
+				},
+				Transaction: &modelpb.Transaction{
+					Type: "txtype",
+					DurationSummary: &modelpb.SummaryMetric{
+						Count: 1,
+						Sum:   100351, // Estimate from histogram
+					},
+					DurationHistogram: &modelpb.Histogram{
+						Values: []float64{100351},
+						Counts: []uint64{1},
+					},
+				},
+				Service: &modelpb.Service{
+					Name: "svc",
+				},
+				Labels: modelpb.Labels{
+					"baz": &modelpb.LabelValue{Global: true, Values: []string{"asd", "qwe", "rty", "fgh"}},
+				},
+				NumericLabels: modelpb.NumericLabels{
+					"bar": &modelpb.NumericLabelValue{Global: true, Values: []float64{1, 2, 4, 5}},
+				},
+				Metricset: &modelpb.Metricset{
+					Name:     "service_transaction",
+					DocCount: 1,
+					Interval: "1s",
+				},
+			},
+		}
+
+		runTest(t, input, expected)
+	})
+
 }
 
 func TestHarvestOverflowCount(t *testing.T) {