From c5cda2e62c75f044b4809b58d85f6654f3e248d4 Mon Sep 17 00:00:00 2001
From: vincent <vincent@ducamps.win>
Date: Sat, 23 Nov 2024 17:59:16 +0100
Subject: [PATCH] fix: case where image got port number and not tag

---
 drivers/docker/driver_test.go | 2 ++
 drivers/docker/utils.go       | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/drivers/docker/driver_test.go b/drivers/docker/driver_test.go
index 99e6e6ef8de..e3b9f6cc91b 100644
--- a/drivers/docker/driver_test.go
+++ b/drivers/docker/driver_test.go
@@ -2860,6 +2860,8 @@ func TestParseDockerImage(t *testing.T) {
 		Repo  string
 		Tag   string
 	}{
+    {"host:5000/library/hello-world", "host:5000/library/hello-world", "latest"},
+    {"host:5000/library/hello-world:1.0", "host:5000/library/hello-world", "1.0"},
 		{"library/hello-world:1.0", "library/hello-world", "1.0"},
 		{"library/hello-world", "library/hello-world", "latest"},
 		{"library/hello-world:latest", "library/hello-world", "latest"},
diff --git a/drivers/docker/utils.go b/drivers/docker/utils.go
index 77a120c5a2f..c329a079f80 100644
--- a/drivers/docker/utils.go
+++ b/drivers/docker/utils.go
@@ -31,6 +31,9 @@ func parseDockerImage(image string) (repo, tag string) {
 	} else if t := repoTag[idx+1:]; !strings.Contains(t, "/") {
 		repo = repoTag[:idx]
 		tag = t
+	} else if t := repoTag[idx+1:]; strings.Contains(t, "/") {
+		repo = image
+		tag = "latest"
 	}
 
 	if tag != "" {