test: fix grouped labels in the fake prometheus (#146)

form3tech-oss · Nov 9, 2023 · 937eb76 · 937eb76
1 parent 8ee2967
commit 937eb76
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/internal/run/prom_push_gw_test.go b/internal/run/prom_push_gw_test.go
@@ -55,13 +55,15 @@ func (f *FakePrometheus) ServeHTTP(response http.ResponseWriter, request *http.R
 		metricFamily := &io_prometheus_client.MetricFamily{}
 		expfmt.NewDecoder(request.Body, expfmt.ResponseFormat(request.Header)).Decode(metricFamily)
 		mf, ok := f.metricFamilies.Load(*metricFamily.Name)
-		if !ok {
-			if metricFamily.Metric != nil {
-				groupedLabels := parseGroupedLabels()
-				for _, m := range metricFamily.Metric {
-					m.Label = append(m.Label, groupedLabels...)
-				}
+
+		if metricFamily.Metric != nil {
+			groupedLabels := parseGroupedLabels()
+			for _, m := range metricFamily.Metric {
+				m.Label = append(m.Label, groupedLabels...)
 			}
+		}
+
+		if !ok {
 			f.metricFamilies.Store(*metricFamily.Name, metricFamily)
 		} else {
 			value, ok := mf.(*io_prometheus_client.MetricFamily)