package secret import ( "bytes" "fmt" "io/ioutil" "testing" "time" "github.com/docker/cli/cli/internal/test" "github.com/docker/docker/api/types/swarm" "github.com/pkg/errors" // Import builders to get the builder function as package function . "github.com/docker/cli/cli/internal/test/builders" "github.com/docker/docker/pkg/testutil" "github.com/docker/docker/pkg/testutil/golden" "github.com/stretchr/testify/assert" ) func TestSecretInspectErrors(t *testing.T) { testCases := []struct { args []string flags map[string]string secretInspectFunc func(secretID string) (swarm.Secret, []byte, error) expectedError string }{ { expectedError: "requires at least 1 argument", }, { args: []string{"foo"}, secretInspectFunc: func(secretID string) (swarm.Secret, []byte, error) { return swarm.Secret{}, nil, errors.Errorf("error while inspecting the secret") }, expectedError: "error while inspecting the secret", }, { args: []string{"foo"}, flags: map[string]string{ "format": "{{invalid format}}", }, expectedError: "Template parsing error", }, { args: []string{"foo", "bar"}, secretInspectFunc: func(secretID string) (swarm.Secret, []byte, error) { if secretID == "foo" { return *Secret(SecretName("foo")), nil, nil } return swarm.Secret{}, nil, errors.Errorf("error while inspecting the secret") }, expectedError: "error while inspecting the secret", }, } for _, tc := range testCases { buf := new(bytes.Buffer) cmd := newSecretInspectCommand( test.NewFakeCliWithOutput(&fakeClient{ secretInspectFunc: tc.secretInspectFunc, }, buf), ) cmd.SetArgs(tc.args) for key, value := range tc.flags { cmd.Flags().Set(key, value) } cmd.SetOutput(ioutil.Discard) testutil.ErrorContains(t, cmd.Execute(), tc.expectedError) } } func TestSecretInspectWithoutFormat(t *testing.T) { testCases := []struct { name string args []string secretInspectFunc func(secretID string) (swarm.Secret, []byte, error) }{ { name: "single-secret", args: []string{"foo"}, secretInspectFunc: func(name string) (swarm.Secret, []byte, error) { if name != "foo" { return swarm.Secret{}, nil, errors.Errorf("Invalid name, expected %s, got %s", "foo", name) } return *Secret(SecretID("ID-foo"), SecretName("foo")), nil, nil }, }, { name: "multiple-secrets-with-labels", args: []string{"foo", "bar"}, secretInspectFunc: func(name string) (swarm.Secret, []byte, error) { return *Secret(SecretID("ID-"+name), SecretName(name), SecretLabels(map[string]string{ "label1": "label-foo", })), nil, nil }, }, } for _, tc := range testCases { buf := new(bytes.Buffer) cmd := newSecretInspectCommand( test.NewFakeCliWithOutput(&fakeClient{ secretInspectFunc: tc.secretInspectFunc, }, buf), ) cmd.SetArgs(tc.args) assert.NoError(t, cmd.Execute()) actual := buf.String() expected := golden.Get(t, []byte(actual), fmt.Sprintf("secret-inspect-without-format.%s.golden", tc.name)) testutil.EqualNormalizedString(t, testutil.RemoveSpace, actual, string(expected)) } } func TestSecretInspectWithFormat(t *testing.T) { secretInspectFunc := func(name string) (swarm.Secret, []byte, error) { return *Secret(SecretName("foo"), SecretLabels(map[string]string{ "label1": "label-foo", })), nil, nil } testCases := []struct { name string format string args []string secretInspectFunc func(name string) (swarm.Secret, []byte, error) }{ { name: "simple-template", format: "{{.Spec.Name}}", args: []string{"foo"}, secretInspectFunc: secretInspectFunc, }, { name: "json-template", format: "{{json .Spec.Labels}}", args: []string{"foo"}, secretInspectFunc: secretInspectFunc, }, } for _, tc := range testCases { buf := new(bytes.Buffer) cmd := newSecretInspectCommand( test.NewFakeCliWithOutput(&fakeClient{ secretInspectFunc: tc.secretInspectFunc, }, buf), ) cmd.SetArgs(tc.args) cmd.Flags().Set("format", tc.format) assert.NoError(t, cmd.Execute()) actual := buf.String() expected := golden.Get(t, []byte(actual), fmt.Sprintf("secret-inspect-with-format.%s.golden", tc.name)) testutil.EqualNormalizedString(t, testutil.RemoveSpace, actual, string(expected)) } } func TestSecretInspectPretty(t *testing.T) { testCases := []struct { name string secretInspectFunc func(string) (swarm.Secret, []byte, error) }{ { name: "simple", secretInspectFunc: func(id string) (swarm.Secret, []byte, error) { return *Secret( SecretLabels(map[string]string{ "lbl1": "value1", }), SecretID("secretID"), SecretName("secretName"), SecretCreatedAt(time.Time{}), SecretUpdatedAt(time.Time{}), ), []byte{}, nil }, }, } for _, tc := range testCases { buf := new(bytes.Buffer) cmd := newSecretInspectCommand( test.NewFakeCliWithOutput(&fakeClient{ secretInspectFunc: tc.secretInspectFunc, }, buf)) cmd.SetArgs([]string{"secretID"}) cmd.Flags().Set("pretty", "true") assert.NoError(t, cmd.Execute()) actual := buf.String() expected := golden.Get(t, []byte(actual), fmt.Sprintf("secret-inspect-pretty.%s.golden", tc.name)) testutil.EqualNormalizedString(t, testutil.RemoveSpace, actual, string(expected)) } }