DockerCLI/cli/command/system/version_test.go

package system

import (
	"fmt"
	"strings"
	"testing"

	"github.com/docker/cli/cli/command"

	"github.com/docker/cli/internal/test"
	"github.com/docker/docker/api"
	"github.com/docker/docker/api/types"
	"github.com/stretchr/testify/assert"
	"golang.org/x/net/context"
)

func TestVersionWithoutServer(t *testing.T) {
	cli := test.NewFakeCli(&fakeClient{
		serverVersion: func(ctx context.Context) (types.Version, error) {
			return types.Version{}, fmt.Errorf("no server")
		},
	})
	cmd := NewVersionCommand(cli)
	cmd.SetOutput(cli.Err())
	assert.Error(t, cmd.Execute())
	assert.Contains(t, cleanTabs(cli.OutBuffer().String()), "Client:")
	assert.NotContains(t, cleanTabs(cli.OutBuffer().String()), "Server:")
}

func fakeServerVersion(ctx context.Context) (types.Version, error) {
	return types.Version{
		Version:    "docker-dev",
		APIVersion: api.DefaultVersion,
	}, nil
}

func TestVersionWithOrchestrator(t *testing.T) {
	cli := test.NewFakeCli(&fakeClient{serverVersion: fakeServerVersion})
	cli.SetClientInfo(func() command.ClientInfo { return command.ClientInfo{Orchestrator: "swarm"} })
	cmd := NewVersionCommand(cli)
	assert.NoError(t, cmd.Execute())
	assert.Contains(t, cleanTabs(cli.OutBuffer().String()), "Orchestrator: swarm")
}

func cleanTabs(line string) string {
	return strings.Join(strings.Fields(line), " ")
}