Skip to content

Commit 006292b

Browse files
committed
cli/command/formatter: use ContainerState consts
Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
1 parent e03712f commit 006292b

File tree

2 files changed

+13
-8
lines changed

2 files changed

+13
-8
lines changed

cli/command/formatter/container_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -352,8 +352,8 @@ size: 0B
352352
}
353353

354354
containers := []container.Summary{
355-
{ID: "containerID1", Names: []string{"/foobar_baz"}, Image: "ubuntu", Created: unixTime, State: "running"},
356-
{ID: "containerID2", Names: []string{"/foobar_bar"}, Image: "ubuntu", Created: unixTime, State: "running"},
355+
{ID: "containerID1", Names: []string{"/foobar_baz"}, Image: "ubuntu", Created: unixTime, State: container.StateRunning},
356+
{ID: "containerID2", Names: []string{"/foobar_bar"}, Image: "ubuntu", Created: unixTime, State: container.StateRunning},
357357
}
358358

359359
for _, tc := range cases {
@@ -434,8 +434,8 @@ func TestContainerContextWriteWithNoContainers(t *testing.T) {
434434
func TestContainerContextWriteJSON(t *testing.T) {
435435
unix := time.Now().Add(-65 * time.Second).Unix()
436436
containers := []container.Summary{
437-
{ID: "containerID1", Names: []string{"/foobar_baz"}, Image: "ubuntu", Created: unix, State: "running"},
438-
{ID: "containerID2", Names: []string{"/foobar_bar"}, Image: "ubuntu", Created: unix, State: "running"},
437+
{ID: "containerID1", Names: []string{"/foobar_baz"}, Image: "ubuntu", Created: unix, State: container.StateRunning},
438+
{ID: "containerID2", Names: []string{"/foobar_bar"}, Image: "ubuntu", Created: unix, State: container.StateRunning},
439439
}
440440
expectedCreated := time.Unix(unix, 0).String()
441441
expectedJSONs := []map[string]any{

cli/command/formatter/disk_usage.go

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@ import (
44
"bytes"
55
"fmt"
66
"strconv"
7-
"strings"
87
"text/template"
98

109
"github.com/distribution/reference"
@@ -330,9 +329,15 @@ func (c *diskUsageContainersContext) TotalCount() string {
330329
}
331330

332331
func (*diskUsageContainersContext) isActive(ctr container.Summary) bool {
333-
return strings.Contains(ctr.State, "running") ||
334-
strings.Contains(ctr.State, "paused") ||
335-
strings.Contains(ctr.State, "restarting")
332+
switch ctr.State {
333+
case container.StateRunning, container.StatePaused, container.StateRestarting:
334+
return true
335+
case container.StateCreated, container.StateRemoving, container.StateExited, container.StateDead:
336+
return false
337+
default:
338+
// Unknown state (should never happen).
339+
return false
340+
}
336341
}
337342

338343
func (c *diskUsageContainersContext) Active() string {

0 commit comments

Comments
 (0)