From fe2bd64ff7dae0d051512f61a30087a8feeee0d9 Mon Sep 17 00:00:00 2001
From: Shubharanshu Mahapatra <shubhum@amazon.com>
Date: Wed, 27 Mar 2024 06:36:35 -0700
Subject: [PATCH] fix: image pull retry logic

Signed-off-by: Shubharanshu Mahapatra <shubhum@amazon.com>
---
 tests/tests.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/tests.go b/tests/tests.go
index 4d3c58e..65a5d17 100644
--- a/tests/tests.go
+++ b/tests/tests.go
@@ -108,7 +108,7 @@ func SetupLocalRegistry(o *option.Option) {
 		// retry pull for 3 times
 		var session *gexec.Session
 		for i := 0; i < retryPull; i++ {
-			session = command.New(o, "pull", ref).WithTimeoutInSeconds(30).WithoutSuccessfulExit().Run()
+			session = command.New(o, "pull", ref).WithTimeoutInSeconds(30).WithoutCheckingExitCode().Run()
 			if session.ExitCode() == 0 {
 				break
 			}