From 0e7d422e5f4dbfc2ef753ee9120b01a7f7640059 Mon Sep 17 00:00:00 2001
From: Sebastiaan van Stijn <github@gone.nl>
Date: Sun, 24 Aug 2025 14:57:36 +0200
Subject: [PATCH] cli/command/formatter: TestContainerPsContext: add test-cases

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
---
 cli/command/formatter/container_test.go | 45 ++++++++++++++++++++++---
 1 file changed, 41 insertions(+), 4 deletions(-)

diff --git a/cli/command/formatter/container_test.go b/cli/command/formatter/container_test.go
index 96626d29f0..b5a96950c1 100644
--- a/cli/command/formatter/container_test.go
+++ b/cli/command/formatter/container_test.go
@@ -54,14 +54,51 @@ func TestContainerPsContext(t *testing.T) {
 			call:      ctx.Image,
 		},
 		{
-			container: container.Summary{Image: "verylongimagename"},
+			container: container.Summary{Image: "ubuntu:latest"},
 			trunc:     true,
-			expValue:  "verylongimagename",
+			expValue:  "ubuntu:latest",
 			call:      ctx.Image,
 		},
 		{
-			container: container.Summary{Image: "verylongimagename"},
-			expValue:  "verylongimagename",
+			container: container.Summary{Image: "docker.io/library/ubuntu"},
+			trunc:     true,
+			expValue:  "ubuntu",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "docker.io/library/ubuntu:latest"},
+			trunc:     true,
+			expValue:  "ubuntu:latest",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "ubuntu:latest@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu:latest",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "ubuntu@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "docker.io/library/ubuntu@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "docker.io/library/ubuntu:latest@sha256:a5a665ff33eced1e0803148700880edab4269067ed77e27737a708d0d293fbf5"},
+			trunc:     true,
+			expValue:  "ubuntu:latest",
+			call:      ctx.Image,
+		},
+		{
+			container: container.Summary{Image: "verylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagename"},
+			trunc:     true,
+			expValue:  "verylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagenameverylongimagename",
 			call:      ctx.Image,
 		},
 		{