@@ -15,13 +15,12 @@ import (
15
15
"text/template"
16
16
"time"
17
17
18
- "go.opentelemetry.io/collector/pdata/plog"
19
- "go.opentelemetry.io/collector/pdata/pmetric"
20
- "gopkg.in/yaml.v3"
21
-
22
18
"github.com/stretchr/testify/assert"
23
19
"github.com/stretchr/testify/require"
24
20
"go.opentelemetry.io/collector/consumer/consumertest"
21
+ "go.opentelemetry.io/collector/pdata/plog"
22
+ "go.opentelemetry.io/collector/pdata/pmetric"
23
+ "gopkg.in/yaml.v3"
25
24
"helm.sh/helm/v3/pkg/action"
26
25
"helm.sh/helm/v3/pkg/kube"
27
26
corev1 "k8s.io/api/core/v1"
@@ -258,8 +257,15 @@ func testIndexSwitch(t *testing.T) {
258
257
var indices []string
259
258
logs := agentLogsConsumer .AllLogs ()
260
259
sourcetypes , indices = getLogsIndexAndSourceType (logs )
261
- assert .True (t , len (sourcetypes ) > 1 ) // we are also receiving logs from other kind containers
262
- assert .Contains (t , sourcetypes , "kube:container:kindnet-cni" )
260
+ assert .Greater (t , len (sourcetypes ), 1 ) // we are receiving logs from different containers
261
+ // check sourcetypes have same prefix
262
+ prefix := "kube:container:"
263
+ for _ , element := range sourcetypes {
264
+ if ! strings .HasPrefix (element , prefix ) {
265
+ t .Errorf ("Element does not start with the prefix %q: %s" , prefix , element )
266
+ }
267
+ }
268
+ assert .NotContains (t , sourcetypes , nonDefaultSourcetype )
263
269
assert .True (t , len (indices ) == 1 )
264
270
assert .True (t , indices [0 ] == logsIndex )
265
271
@@ -278,7 +284,6 @@ func testIndexSwitch(t *testing.T) {
278
284
internal .ResetLogsSink (t , agentLogsConsumer )
279
285
internal .ResetMetricsSink (t , hecMetricsConsumer )
280
286
281
- internal .WaitForMetrics (t , 3 , hecMetricsConsumer )
282
287
internal .WaitForLogs (t , 3 , agentLogsConsumer )
283
288
logs = agentLogsConsumer .AllLogs ()
284
289
sourcetypes , indices = getLogsIndexAndSourceType (logs )
@@ -287,6 +292,8 @@ func testIndexSwitch(t *testing.T) {
287
292
assert .Contains (t , sourcetypes , nonDefaultSourcetype )
288
293
assert .True (t , len (indices ) == 1 )
289
294
assert .True (t , len (sourcetypes ) == 1 )
295
+
296
+ internal .WaitForMetrics (t , 3 , hecMetricsConsumer )
290
297
mIndices = getMetricsIndex (hecMetricsConsumer .AllMetrics ())
291
298
assert .True (t , len (mIndices ) == 1 )
292
299
assert .True (t , mIndices [0 ] == newMetricsIndex )
@@ -307,6 +314,7 @@ func testClusterReceiverEnabledOrDisabled(t *testing.T) {
307
314
logsObjectsHecEndpoint := fmt .Sprintf ("http://%s:%d/services/collector" , hostEp , internal .HECObjectsReceiverPort )
308
315
309
316
t .Run ("check cluster receiver enabled" , func (t * testing.T ) {
317
+ internal .ResetLogsSink (t , logsObjectsConsumer )
310
318
replacements := map [string ]interface {}{
311
319
"ClusterReceiverEnabled" : false ,
312
320
"LogObjectsHecEndpoint" : logsObjectsHecEndpoint ,
0 commit comments