Add more error types to trigger retries (#25827)

* add more scenarios to trigger retries * add error types to enum
DataDog · May 23, 2024 · d47c6d7 · d47c6d7
1 parent 09a230f
commit d47c6d7
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/test/new-e2e/system-probe/errors.go b/test/new-e2e/system-probe/errors.go
@@ -35,6 +35,8 @@ const (
 	insufficientCapacityError
 	aria2cMissingStatusError
 	ec2StateChangeTimeoutError
+	ioTimeout
+	tcp22ConnectionRefused
 )
 
 type handledError struct {
@@ -74,6 +76,16 @@ var handledErrorsLs = []handledError{
 		metric:      "ec2-timeout-state-change",
 		action:      retryStack | emitMetric,
 	},
+	{
+		errorType:   ioTimeout,
+		errorString: "i/o timeout",
+		action:      retryStack,
+	},
+	{
+		errorType:   tcp22ConnectionRefused,
+		errorString: "failed attempts: dial tcp :22: connect: connection refused",
+		action:      retryStack,
+	},
 }
 
 func errorMetric(errType string) datadogV2.MetricPayload {