cli/command/formatter: TestContainerPsContext: add test-cases

Signed-off-by: Sebastiaan van Stijn <[email protected]>

Author: thaJeztah

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,
 		},
 		{