fix: case where image got port number and not tag

hashicorp · Nov 23, 2024 · c5cda2e · c5cda2e
1 parent f7a4ded
commit c5cda2e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git 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
@@ -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 != "" {