Uses net.OpError for network error testing

hashicorp · Sep 15, 2021 · a49aa13 · a49aa13
1 parent de4c91c
commit a49aa13
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/session_test.go b/session_test.go
@@ -4,6 +4,7 @@ import (
 	"errors"
 	"fmt"
 	"io/ioutil"
+	"net"
 	"os"
 	"reflect"
 	"runtime"
@@ -1239,42 +1240,42 @@ func TestSessionRetryHandlers(t *testing.T) {
 		{
 			Description:              "send request no such host failed under MaxNetworkRetryCount",
 			RetryCount:               MaxNetworkRetryCount - 1,
-			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", errors.New("no such host")),
+			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", &net.OpError{Op: "dial", Err: errors.New("no such host")}),
 			ExpectedRetryableValue:   true,
 			ExpectRetryToBeAttempted: true,
 		},
 		{
 			Description:              "send request no such host failed over MaxNetworkRetryCount",
 			RetryCount:               MaxNetworkRetryCount,
-			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", errors.New("no such host")),
+			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", &net.OpError{Op: "dial", Err: errors.New("no such host")}),
 			ExpectedRetryableValue:   false,
 			ExpectRetryToBeAttempted: false,
 		},
 		{
 			Description:              "send request connection refused failed under MaxNetworkRetryCount",
 			RetryCount:               MaxNetworkRetryCount - 1,
-			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", errors.New("connection refused")),
+			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", &net.OpError{Op: "dial", Err: errors.New("connection refused")}),
 			ExpectedRetryableValue:   true,
 			ExpectRetryToBeAttempted: true,
 		},
 		{
 			Description:              "send request connection refused failed over MaxNetworkRetryCount",
 			RetryCount:               MaxNetworkRetryCount,
-			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", errors.New("connection refused")),
+			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", &net.OpError{Op: "dial", Err: errors.New("connection refused")}),
 			ExpectedRetryableValue:   false,
 			ExpectRetryToBeAttempted: false,
 		},
 		{
 			Description:              "send request other error failed under MaxNetworkRetryCount",
 			RetryCount:               MaxNetworkRetryCount - 1,
-			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", errors.New("other error")),
+			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", &net.OpError{Op: "dial", Err: errors.New("other error")}),
 			ExpectedRetryableValue:   true,
 			ExpectRetryToBeAttempted: true,
 		},
 		{
 			Description:              "send request other error failed over MaxNetworkRetryCount",
 			RetryCount:               MaxNetworkRetryCount,
-			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", errors.New("other error")),
+			Error:                    awserr.New(request.ErrCodeRequestError, "send request failed", &net.OpError{Op: "dial", Err: errors.New("other error")}),
 			ExpectedRetryableValue:   true,
 			ExpectRetryToBeAttempted: true,
 		},