[chore] [mdatagen] Fix generated telemetry tests (#12034)

Unblocks open-telemetry/opentelemetry-collector-contrib#37051
open-telemetry · Jan 7, 2025 · ced38e8 · ced38e8
1 parent 83c909f
commit ced38e8
Show file tree

Hide file tree

Showing 6 changed files with 44 additions and 4 deletions.
diff --git a/cmd/mdatagen/internal/loader_test.go b/cmd/mdatagen/internal/loader_test.go
@@ -288,6 +288,16 @@ func TestLoadMetadata(t *testing.T) {
 								Async: true,
 							},
 						},
+						"queue_capacity": {
+							Enabled:     true,
+							Description: "Queue capacity - sync gauge example.",
+							Unit:        strPtr("{items}"),
+							Gauge: &Gauge{
+								MetricValueType: MetricValueType{
+									ValueType: pmetric.NumberDataPointValueTypeInt,
+								},
+							},
+						},
 					},
 				},
 				ScopeName:       "go.opentelemetry.io/collector/internal/receiver/samplereceiver",

diff --git a/cmd/mdatagen/internal/samplereceiver/documentation.md b/cmd/mdatagen/internal/samplereceiver/documentation.md
@@ -134,6 +134,14 @@ Cumulative bytes allocated for heap objects (see 'go doc runtime.MemStats.TotalA
 | ---- | ----------- | ---------- | --------- |
 | By | Sum | Int | true |
 
+### otelcol_queue_capacity
+
+Queue capacity - sync gauge example.
+
+| Unit | Metric Type | Value Type |
+| ---- | ----------- | ---------- |
+| {items} | Gauge | Int |
+
 ### otelcol_queue_length
 
 This metric is optional and therefore not initialized in NewTelemetryBuilder.

diff --git a/cmd/mdatagen/internal/samplereceiver/internal/metadata/generated_telemetry.go b/cmd/mdatagen/internal/samplereceiver/internal/metadata/generated_telemetry.go
diff --git a/cmd/mdatagen/internal/samplereceiver/internal/metadatatest/generated_telemetrytest_test.go b/cmd/mdatagen/internal/samplereceiver/internal/metadatatest/generated_telemetrytest_test.go
diff --git a/cmd/mdatagen/internal/samplereceiver/metadata.yaml b/cmd/mdatagen/internal/samplereceiver/metadata.yaml
@@ -197,3 +197,9 @@ telemetry:
       gauge:
         async: true
         value_type: int
+    queue_capacity:
+      enabled: true
+      description: Queue capacity - sync gauge example.
+      unit: "{items}"
+      gauge:
+        value_type: int
diff --git a/cmd/mdatagen/internal/templates/telemetrytest_test.go.tmpl b/cmd/mdatagen/internal/templates/telemetrytest_test.go.tmpl
@@ -26,11 +26,9 @@ func TestSetupTelemetry(t *testing.T) {
 	require.NotNil(t, tb)
     {{- range $name, $metric := .Telemetry.Metrics }}
         {{- if (and (not $metric.Optional) (not $metric.Data.Async)) }}
-            {{ if eq $metric.Data.Type "Gauge" -}}
-            	tb.{{ $name.Render }}.Set(context.Background(), 1)
-            {{- else if eq $metric.Data.Type "Sum" -}}
+            {{- if eq $metric.Data.Type "Sum" }}
             	tb.{{ $name.Render }}.Add(context.Background(), 1)
-            {{- else if eq $metric.Data.Type "Histogram" -}}
+            {{- else }}
             	tb.{{ $name.Render }}.Record(context.Background(), 1)
             {{- end }}
         {{- end }}