64 lines
1.7 KiB
Go
64 lines
1.7 KiB
Go
![]() |
package formatter
|
||
|
|
||
|
import (
|
||
|
"bytes"
|
||
|
"testing"
|
||
|
"time"
|
||
|
|
||
|
"github.com/docker/docker/api/types/swarm"
|
||
|
"github.com/docker/docker/pkg/testutil/assert"
|
||
|
)
|
||
|
|
||
|
func TestSecretContextFormatWrite(t *testing.T) {
|
||
|
// Check default output format (verbose and non-verbose mode) for table headers
|
||
|
cases := []struct {
|
||
|
context Context
|
||
|
expected string
|
||
|
}{
|
||
|
// Errors
|
||
|
{
|
||
|
Context{Format: "{{InvalidFunction}}"},
|
||
|
`Template parsing error: template: :1: function "InvalidFunction" not defined
|
||
|
`,
|
||
|
},
|
||
|
{
|
||
|
Context{Format: "{{nil}}"},
|
||
|
`Template parsing error: template: :1:2: executing "" at <nil>: nil is not a command
|
||
|
`,
|
||
|
},
|
||
|
// Table format
|
||
|
{Context{Format: NewSecretFormat("table", false)},
|
||
|
`ID NAME CREATED UPDATED
|
||
|
1 passwords Less than a second ago Less than a second ago
|
||
|
2 id_rsa Less than a second ago Less than a second ago
|
||
|
`},
|
||
|
{Context{Format: NewSecretFormat("table {{.Name}}", true)},
|
||
|
`NAME
|
||
|
passwords
|
||
|
id_rsa
|
||
|
`},
|
||
|
{Context{Format: NewSecretFormat("{{.ID}}-{{.Name}}", false)},
|
||
|
`1-passwords
|
||
|
2-id_rsa
|
||
|
`},
|
||
|
}
|
||
|
|
||
|
secrets := []swarm.Secret{
|
||
|
{ID: "1",
|
||
|
Meta: swarm.Meta{CreatedAt: time.Now(), UpdatedAt: time.Now()},
|
||
|
Spec: swarm.SecretSpec{Annotations: swarm.Annotations{Name: "passwords"}}},
|
||
|
{ID: "2",
|
||
|
Meta: swarm.Meta{CreatedAt: time.Now(), UpdatedAt: time.Now()},
|
||
|
Spec: swarm.SecretSpec{Annotations: swarm.Annotations{Name: "id_rsa"}}},
|
||
|
}
|
||
|
for _, testcase := range cases {
|
||
|
out := bytes.NewBufferString("")
|
||
|
testcase.context.Output = out
|
||
|
if err := SecretWrite(testcase.context, secrets); err != nil {
|
||
|
assert.Error(t, err, testcase.expected)
|
||
|
} else {
|
||
|
assert.Equal(t, out.String(), testcase.expected)
|
||
|
}
|
||
|
}
|
||
|
}
|