modify UT for collector_fields test

SumoLogic · Dec 10, 2024 · a7ff13d · a7ff13d
1 parent 4cc054c
commit a7ff13d
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/pkg/configprovider/globprovider/provider_test.go b/pkg/configprovider/globprovider/provider_test.go
@@ -176,6 +176,10 @@ func TestRemotelyManagedMergeFlow(t *testing.T) {
 						"collector_fields": map[string]interface{}{
 							"zone": "eu",
 						},
+						"collector_fields1": map[string]interface{}{
+							"cluster": "cluster-1",
+							"zone": "eu",
+						},
 					},
 				},
 				"processor": "someprocessor",
@@ -198,6 +202,10 @@ func TestLocallyManagedMergeFlow(t *testing.T) {
 							"cluster": "cluster-1",
 							"zone": "eu",
 						},
+						"collector_fields1": map[string]interface{}{
+							"cluster": "cluster-1",
+							"zone": "eu",
+						},
 					},
 				},
 				"processor": "someprocessor",

diff --git a/pkg/configprovider/globprovider/testdata/mergefunc/a_srcconf.yaml b/pkg/configprovider/globprovider/testdata/mergefunc/a_srcconf.yaml
@@ -3,4 +3,6 @@ extensions:
     childKey: "value"
     collector_fields:
       cluster: "cluster-1"
+    collector_fields1:
+      cluster: "cluster-1"
 processor: "someprocessor"
diff --git a/pkg/configprovider/globprovider/testdata/mergefunc/mergeconf.yaml b/pkg/configprovider/globprovider/testdata/mergefunc/mergeconf.yaml
@@ -3,3 +3,5 @@ extensions:
     childKey: "value"
     collector_fields:
       zone: "eu"
+    collector_fields1:
+      zone: "eu"
diff --git a/pkg/configprovider/opampprovider/provider_test.go b/pkg/configprovider/opampprovider/provider_test.go
@@ -144,7 +144,9 @@ func TestRemotelyManagedFlowDisabled(t *testing.T) {
 	exp := confmap.NewFromStringMap(map[string]any{
 		"extensions::sumologic::childKey":                   "value",
 		"extensions::sumologic::collector_fields::cluster":   "cluster-1",
-                "extensions::sumologic::collector_fields::zone":      "eu",
+        "extensions::sumologic::collector_fields::zone":      "eu",
+        "extensions::sumologic::collector_fields1::cluster":   "cluster-1",
+        "extensions::sumologic::collector_fields1::zone":      "eu",
 		"processor":                                         "someprocessor",
 		"extensions::opamp::remote_configuration_directory": "../globprovider/testdata/mergefunc",
 		"extensions::opamp::endpoint":                       "wss://example.com/v1/opamp",
@@ -179,6 +181,8 @@ func TestRemotelyManagedFlowEnabled(t *testing.T) {
 	exp := confmap.NewFromStringMap(map[string]any{
 		"extensions::sumologic::childKey":                   "value",
 		"extensions::sumologic::collector_fields::zone":      "eu",
+		"extensions::sumologic::collector_fields1::cluster":   "cluster-1",
+        "extensions::sumologic::collector_fields1::zone":      "eu",
 		"processor":                                         "someprocessor",
 		"extensions::opamp::remote_configuration_directory": "../globprovider/testdata/mergefunc",
 		"extensions::opamp::endpoint":                       "wss://example.com/v1/opamp",