53 lines
976 B
Go
53 lines
976 B
Go
package formatter
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/gotestyourself/gotestyourself/assert"
|
|
)
|
|
|
|
func TestCheckpointContextFormatWrite(t *testing.T) {
|
|
cases := []struct {
|
|
context Context
|
|
expected string
|
|
}{
|
|
{
|
|
Context{Format: NewCheckpointFormat(defaultCheckpointFormat)},
|
|
`CHECKPOINT NAME
|
|
checkpoint-1
|
|
checkpoint-2
|
|
checkpoint-3
|
|
`,
|
|
},
|
|
{
|
|
Context{Format: NewCheckpointFormat("{{.Name}}")},
|
|
`checkpoint-1
|
|
checkpoint-2
|
|
checkpoint-3
|
|
`,
|
|
},
|
|
{
|
|
Context{Format: NewCheckpointFormat("{{.Name}}:")},
|
|
`checkpoint-1:
|
|
checkpoint-2:
|
|
checkpoint-3:
|
|
`,
|
|
},
|
|
}
|
|
|
|
checkpoints := []types.Checkpoint{
|
|
{Name: "checkpoint-1"},
|
|
{Name: "checkpoint-2"},
|
|
{Name: "checkpoint-3"},
|
|
}
|
|
for _, testcase := range cases {
|
|
out := bytes.NewBufferString("")
|
|
testcase.context.Output = out
|
|
err := CheckpointWrite(testcase.context, checkpoints)
|
|
assert.NilError(t, err)
|
|
assert.Equal(t, out.String(), testcase.expected)
|
|
}
|
|
}
|