mirror of https://github.com/docker/cli.git
120 lines
2.9 KiB
Go
120 lines
2.9 KiB
Go
package formatter
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"gotest.tools/v3/assert"
|
|
"gotest.tools/v3/golden"
|
|
)
|
|
|
|
func TestDiskUsageContextFormatWrite(t *testing.T) {
|
|
cases := []struct {
|
|
context DiskUsageContext
|
|
expected string
|
|
}{
|
|
// Check default output format (verbose and non-verbose mode) for table headers
|
|
{
|
|
DiskUsageContext{
|
|
Context: Context{
|
|
Format: NewDiskUsageFormat("table", false),
|
|
},
|
|
Verbose: false,
|
|
},
|
|
`TYPE TOTAL ACTIVE SIZE RECLAIMABLE
|
|
Images 0 0 0B 0B
|
|
Containers 0 0 0B 0B
|
|
Local Volumes 0 0 0B 0B
|
|
Build Cache 0 0 0B 0B
|
|
`,
|
|
},
|
|
{
|
|
DiskUsageContext{Verbose: true, Context: Context{Format: NewDiskUsageFormat("table", true)}},
|
|
`Images space usage:
|
|
|
|
REPOSITORY TAG IMAGE ID CREATED SIZE SHARED SIZE UNIQUE SIZE CONTAINERS
|
|
|
|
Containers space usage:
|
|
|
|
CONTAINER ID IMAGE COMMAND LOCAL VOLUMES SIZE CREATED STATUS NAMES
|
|
|
|
Local Volumes space usage:
|
|
|
|
VOLUME NAME LINKS SIZE
|
|
|
|
Build cache usage: 0B
|
|
|
|
CACHE ID CACHE TYPE SIZE CREATED LAST USED USAGE SHARED
|
|
`,
|
|
},
|
|
{
|
|
DiskUsageContext{Verbose: true, Context: Context{Format: NewDiskUsageFormat("raw", true)}},
|
|
``,
|
|
},
|
|
{
|
|
DiskUsageContext{Verbose: true, Context: Context{Format: NewDiskUsageFormat("{{json .}}", true)}},
|
|
`{"Images":[],"Containers":[],"Volumes":[],"BuildCache":[]}`,
|
|
},
|
|
// Errors
|
|
{
|
|
DiskUsageContext{
|
|
Context: Context{
|
|
Format: "{{InvalidFunction}}",
|
|
},
|
|
},
|
|
`template parsing error: template: :1: function "InvalidFunction" not defined`,
|
|
},
|
|
{
|
|
DiskUsageContext{
|
|
Context: Context{
|
|
Format: "{{nil}}",
|
|
},
|
|
},
|
|
`template parsing error: template: :1:2: executing "" at <nil>: nil is not a command`,
|
|
},
|
|
// Table Format
|
|
{
|
|
DiskUsageContext{
|
|
Context: Context{
|
|
Format: NewDiskUsageFormat("table", false),
|
|
},
|
|
},
|
|
`TYPE TOTAL ACTIVE SIZE RECLAIMABLE
|
|
Images 0 0 0B 0B
|
|
Containers 0 0 0B 0B
|
|
Local Volumes 0 0 0B 0B
|
|
Build Cache 0 0 0B 0B
|
|
`,
|
|
},
|
|
{
|
|
DiskUsageContext{
|
|
Context: Context{
|
|
Format: NewDiskUsageFormat("table {{.Type}}\t{{.Active}}", false),
|
|
},
|
|
},
|
|
string(golden.Get(t, "disk-usage-context-write-custom.golden")),
|
|
},
|
|
// Raw Format
|
|
{
|
|
DiskUsageContext{
|
|
Context: Context{
|
|
Format: NewDiskUsageFormat("raw", false),
|
|
},
|
|
},
|
|
string(golden.Get(t, "disk-usage-raw-format.golden")),
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
t.Run(string(tc.context.Format), func(t *testing.T) {
|
|
var out bytes.Buffer
|
|
tc.context.Output = &out
|
|
if err := tc.context.Write(); err != nil {
|
|
assert.Error(t, err, tc.expected)
|
|
} else {
|
|
assert.Equal(t, out.String(), tc.expected)
|
|
}
|
|
})
|
|
}
|
|
}
|