Add retry logic to fluent test

aws · Dec 3, 2024 · 7cf5461 · 7cf5461
1 parent 410288f
commit 7cf5461
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/test/fluent/fluent_test.go b/test/fluent/fluent_test.go
@@ -17,7 +17,7 @@ import (
 	"github.com/aws/amazon-cloudwatch-agent-test/util/awsservice"
 )
 
-const logStreamRetry = 20
+const logStreamRetry = 10
 
 // fluent log group with expected log message fields
 var logGroupToKey = map[string][][]string{
@@ -61,10 +61,19 @@ func TestFluentLogs(t *testing.T) {
 		logGroupToKey = logGroupToKeyWindows
 	}
 
+	currRetries := 0
 	now := time.Now()
 	for group, fieldsArr := range logGroupToKey {
 		group = fmt.Sprintf("/aws/containerinsights/%s/%s", env.EKSClusterName, group)
-		if !awsservice.IsLogGroupExists(group) {
+		for currRetries < logStreamRetry {
+			if awsservice.IsLogGroupExists(group) {
+				break
+			} else {
+				currRetries++
+				time.Sleep(time.Duration(currRetries) * time.Second)
+			}
+		}
+		if currRetries >= logStreamRetry {
 			t.Fatalf("fluent log group doesn't exsit: %s", group)
 		}