From dbd0197a4d80b2ba6883f374853e1a23f3e45b7e Mon Sep 17 00:00:00 2001
From: Sajiyah Salat <109643863+Sajiyah-Salat@users.noreply.github.com>
Date: Tue, 15 Aug 2023 18:34:45 +0530
Subject: [PATCH] :seedling: Image name edge cases covered (#3514)

imagename edge case cover
---
 test/e2e/utils/test_context.go | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/test/e2e/utils/test_context.go b/test/e2e/utils/test_context.go
index c10b8e138b9..6ee508eb74c 100644
--- a/test/e2e/utils/test_context.go
+++ b/test/e2e/utils/test_context.go
@@ -243,10 +243,16 @@ func (t *TestContext) Destroy() {
 	//nolint:gosec
 	// if image name is not present or not provided skip execution of docker command
 	if t.ImageName != "" {
-		cmd := exec.Command("docker", "rmi", "-f", t.ImageName)
-		if _, err := t.Run(cmd); err != nil {
-			warnError(err)
+		// Check white space from image name
+		if len(strings.TrimSpace(t.ImageName)) == 0 {
+			fmt.Println("Image not set, skip cleaning up of docker image")
+		} else {
+			cmd := exec.Command("docker", "rmi", "-f", t.ImageName)
+			if _, err := t.Run(cmd); err != nil {
+				warnError(err)
+			}
 		}
+
 	}
 	if err := os.RemoveAll(t.Dir); err != nil {
 		warnError(err)