Function TestVersionCommand

cmd/compose/version_test.go:31–77

(t *testing.T)

)

func TestVersionCommand(t *testing.T) {
	originalVersion := internal.Version
	defer func() {
		internal.Version = originalVersion
	}()
	internal.Version = "v9.9.9-test"

	tests := []struct {
		name string
		args []string
		want string
	}{
		{
			name: "default",
			args: []string{},
			want: "Docker Compose version v9.9.9-test\n",
		},
		{
			name: "short flag",
			args: []string{"--short"},
			want: "9.9.9-test\n",
		},
		{
			name: "json flag",
			args: []string{"--format", "json"},
			want: `{"version":"v9.9.9-test"}` + "\n",
		},
	}

	for _, test := range tests {
		t.Run(test.name, func(t *testing.T) {
			ctrl := gomock.NewController(t)
			defer ctrl.Finish()

			buf := new(bytes.Buffer)
			cli := mocks.NewMockCli(ctrl)
			cli.EXPECT().Out().Return(streams.NewOut(buf)).AnyTimes()

			cmd := versionCommand(cli)
			cmd.SetArgs(test.args)
			err := cmd.Execute()
			assert.NilError(t, err)

			assert.Equal(t, test.want, buf.String())
		})
	}
}