catch case where duration is longer than configured

allcloud-io · Jul 3, 2023 · 6f10162 · 6f10162
1 parent 4b4280f
commit 6f10162
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aws/sts.go b/aws/sts.go
@@ -45,7 +45,7 @@ func AssumeSAMLRole(PrincipalArn, RoleArn, SAMLAssertion, awsRegion string, dura
 		var ae smithy.APIError
 		if errors.As(err, &ae) {
 			// Check if error indicates exceeded duration, no structured error exists so check error message content.
-			if strings.Contains(ae.ErrorMessage(), "'durationSeconds' failed to satisfy constraint") {
+			if strings.Contains(ae.ErrorMessage(), "'durationSeconds' failed to satisfy constraint") || ae.ErrorMessage() == ErrInvalidSessionDuration {
 				// Return a custom error to allow the caller to retry etc.
 				// TODO Return a custom error type instead of a special value:
 				// https://dave.cheney.net/2014/12/24/inspecting-errors