fix test

open-telemetry · Mar 27, 2024 · 59a4e27 · 59a4e27
1 parent 55c94aa
commit 59a4e27
Show file tree

Hide file tree

Showing 4 changed files with 91 additions and 57 deletions.
diff --git a/cmd/mdatagen/internal/samplereceiver/internal/metadata/generated_metrics_test.go b/cmd/mdatagen/internal/samplereceiver/internal/metadata/generated_metrics_test.go
diff --git a/cmd/mdatagen/internal/samplereceiver/internal/metadata/testdata/config.yaml b/cmd/mdatagen/internal/samplereceiver/internal/metadata/testdata/config.yaml
@@ -57,12 +57,14 @@ none_set:
       enabled: false
     string.resource.attr_to_be_removed:
       enabled: false
-filter_set:
+filter_set_include:
   metrics:
     default.metric:
       enabled: true
     default.metric.to_be_removed:
       enabled: true
+    metric.input_type:
+      enabled: true
     optional.metric:
       enabled: true
     optional.metric.empty_unit:
@@ -72,63 +74,68 @@ filter_set:
       enabled: true
     optional.resource.attr:
       enabled: true
-      exclude:
-        - strict: "non-existent"
-        - regexp: "non-existent"
       include:
-        - strict: "non-existent"
-        - regexp: "non-existent"
-        - strict: "optional.resource.attr-val"
         - regexp: ".*"
     slice.resource.attr:
       enabled: true
     string.enum.resource.attr:
       enabled: true
-      exclude:
-        - strict: "non-existent"
-        - regexp: "non-existent"
       include:
-        - strict: "non-existent"
-        - regexp: "non-existent"
-        - strict: "one"
         - regexp: ".*"
     string.resource.attr:
       enabled: true
-      exclude:
-        - strict: "non-existent"
-        - regexp: "non-existent"
       include:
-        - strict: "non-existent"
-        - regexp: "non-existent"
-        - strict: "string.resource.attr-val"
         - regexp: ".*"
     string.resource.attr_disable_warning:
       enabled: true
-      exclude:
-        - strict: "non-existent"
-        - regexp: "non-existent"
       include:
-        - strict: "non-existent"
-        - regexp: "non-existent"
-        - strict: "string.resource.attr_disable_warning-val"
         - regexp: ".*"
     string.resource.attr_remove_warning:
       enabled: true
-      exclude:
-        - strict: "non-existent"
-        - regexp: "non-existent"
       include:
-        - strict: "non-existent"
-        - regexp: "non-existent"
-        - strict: "string.resource.attr_remove_warning-val"
         - regexp: ".*"
     string.resource.attr_to_be_removed:
       enabled: true
-      exclude:
-        - strict: "non-existent"
-        - regexp: "non-existent"
       include:
-        - strict: "non-existent"
-        - regexp: "non-existent"
-        - strict: "string.resource.attr_to_be_removed-val"
         - regexp: ".*"
+filter_set_exclude:
+  metrics:
+    default.metric:
+      enabled: true
+    default.metric.to_be_removed:
+      enabled: true
+    metric.input_type:
+      enabled: true
+    optional.metric:
+      enabled: true
+    optional.metric.empty_unit:
+      enabled: true
+  resource_attributes:
+    map.resource.attr:
+      enabled: true
+    optional.resource.attr:
+      enabled: true
+      exclude:
+        - strict: "optional.resource.attr-val"
+    slice.resource.attr:
+      enabled: true
+    string.enum.resource.attr:
+      enabled: true
+      exclude:
+        - strict: "one"
+    string.resource.attr:
+      enabled: true
+      exclude:
+        - strict: "string.resource.attr-val"
+    string.resource.attr_disable_warning:
+      enabled: true
+      exclude:
+        - strict: "string.resource.attr_disable_warning-val"
+    string.resource.attr_remove_warning:
+      enabled: true
+      exclude:
+        - strict: "string.resource.attr_remove_warning-val"
+    string.resource.attr_to_be_removed:
+      enabled: true
+      exclude:
+        - strict: "string.resource.attr_to_be_removed-val"
diff --git a/cmd/mdatagen/templates/metrics_test.go.tmpl b/cmd/mdatagen/templates/metrics_test.go.tmpl
@@ -20,7 +20,8 @@ const (
 	testSetDefault testConfigCollection = iota
 	testSetAll
 	testSetNone
-	testSetFilter
+	testSetFilterInclude
+	testSetFilterExclude
 )
 
 func TestMetricsBuilder(t *testing.T) {
@@ -41,8 +42,12 @@ func TestMetricsBuilder(t *testing.T) {
 			configSet: testSetNone,
 		},
 		{
-			name:       "filter_set",
-			configSet: testSetFilter,
+			name:       "filter_set_include",
+			configSet: testSetFilterInclude,
+		},
+		{
+			name:       "filter_set_exclude",
+			configSet: testSetFilterExclude,
 		},
 	}
 	for _, test := range tests {
@@ -57,7 +62,7 @@ func TestMetricsBuilder(t *testing.T) {
 			expectedWarnings := 0
 			{{- range $name, $metric := .Metrics }}
 			{{- if and $metric.Enabled $metric.Warnings.IfEnabled }}
-			if test.configSet == testSetDefault || test.configSet == testSetAll || test.configSet == testSetFilter {
+			if test.configSet == testSetDefault || test.configSet == testSetAll || test.configSet == testSetFilterInclude || test.configSet == testSetFilterExclude {
 				assert.Equal(t, "[WARNING] `{{ $name }}` should not be enabled: {{ $metric.Warnings.IfEnabled }}", observedLogs.All()[expectedWarnings].Message)
 				expectedWarnings++
 			}
@@ -69,15 +74,15 @@ func TestMetricsBuilder(t *testing.T) {
 			}
 			{{- end }}
 			{{- if $metric.Warnings.IfConfigured }}
-			if test.configSet == testSetAll || test.configSet == testSetNone || test.configSet == testSetFilter {
+			if test.configSet == testSetAll || test.configSet == testSetNone || test.configSet == testSetFilterInclude || test.configSet == testSetFilterExclude {
 				assert.Equal(t, "[WARNING] `{{ $name }}` should not be configured: {{ $metric.Warnings.IfConfigured }}", observedLogs.All()[expectedWarnings].Message)
 				expectedWarnings++
 			}
 			{{- end }}
 			{{- end }}
 			{{- range $name, $attr := .ResourceAttributes }}
 			{{- if and $attr.Enabled $attr.Warnings.IfEnabled }}
-			if test.configSet == testSetDefault || test.configSet == testSetAll || test.configSet == testSetFilter {
+			if test.configSet == testSetDefault || test.configSet == testSetAll || test.configSet == testSetFilterInclude || test.configSet == testSetFilterExclude {
 				assert.Equal(t, "[WARNING] `{{ $name }}` should not be enabled: {{ $attr.Warnings.IfEnabled }}", observedLogs.All()[expectedWarnings].Message)
 				expectedWarnings++
 			}
@@ -89,7 +94,7 @@ func TestMetricsBuilder(t *testing.T) {
 			}
 			{{- end }}
 			{{- if $attr.Warnings.IfConfigured }}
-			if test.configSet == testSetAll || test.configSet == testSetNone || test.configSet == testSetFilter {
+			if test.configSet == testSetAll || test.configSet == testSetNone || test.configSet == testSetFilterInclude || test.configSet == testSetFilterExclude {
 				assert.Equal(t, "[WARNING] `{{ $name }}` should not be configured: {{ $attr.Warnings.IfConfigured }}", observedLogs.All()[expectedWarnings].Message)
 				expectedWarnings++
 			}
@@ -126,7 +131,7 @@ func TestMetricsBuilder(t *testing.T) {
 			{{- end }}
 			metrics := mb.Emit(WithResource(res))
 
-			if test.configSet == testSetNone {
+			if test.configSet == testSetNone || test.configSet == testSetFilterExclude {
 				assert.Equal(t, 0, metrics.ResourceMetrics().Len())
 				return
 			}

diff --git a/cmd/mdatagen/templates/testdata/config.yaml.tmpl b/cmd/mdatagen/templates/testdata/config.yaml.tmpl
@@ -29,7 +29,7 @@ none_set:
       enabled: false
     {{- end }}
   {{- end }}
-filter_set:
+filter_set_include:
   {{- if .Metrics }}
   metrics:
     {{- range $name, $_ := .Metrics }}
@@ -43,10 +43,27 @@ filter_set:
     {{ $name }}:
       enabled: true
       {{- if eq $attr.Type.String "Str" }}
-      exclude:
-        - strict: {{ $attr.TestValue }}
       include:
         - regexp: ".*"
       {{- end }}
     {{- end }}
   {{- end }}
+filter_set_exclude:
+  {{- if .Metrics }}
+  metrics:
+    {{- range $name, $_ := .Metrics }}
+    {{ $name }}:
+      enabled: true
+    {{- end }}
+  {{- end }}
+  {{- if .ResourceAttributes }}
+  resource_attributes:
+    {{- range $name, $attr := .ResourceAttributes }}
+    {{ $name }}:
+      enabled: true
+      {{- if eq $attr.Type.String "Str" }}
+      exclude:
+        - strict: {{ $attr.TestValue }}
+      {{- end }}
+    {{- end }}
+  {{- end }}