commands/version: Add

Author: rajivr

commands/version.go

@@ -0,0 +1,40 @@
+package commands
+
+import (
+	"fmt"
+
+	"io"
+	"os"
+
+	"github.com/docker/machine/libmachine"
+	"github.com/docker/machine/libmachine/mcndockerclient"
+)
+
+func cmdVersion(c CommandLine, api libmachine.API) error {
+	return printVersion(c, api, os.Stdout)
+}
+
+func printVersion(c CommandLine, api libmachine.API, out io.Writer) error {
+	if len(c.Args()) == 0 {
+		c.ShowVersion()
+		return nil
+	}
+
+	if len(c.Args()) != 1 {
+		return ErrExpectedOneMachine
+	}
+
+	host, err := api.Load(c.Args().First())
+	if err != nil {
+		return err
+	}
+
+	version, err := mcndockerclient.DockerVersion(host)
+	if err != nil {
+		return err
+	}
+
+	fmt.Fprintln(out, version)
+
+	return nil
+}

commands/version_test.go

@@ -0,0 +1,89 @@
+package commands
+
+import (
+	"errors"
+	"testing"
+
+	"bytes"
+
+	"github.com/docker/machine/commands/commandstest"
+	"github.com/docker/machine/libmachine/host"
+	"github.com/docker/machine/libmachine/libmachinetest"
+	"github.com/docker/machine/libmachine/mcndockerclient"
+	"github.com/stretchr/testify/assert"
+)
+
+func TestCmdVersion(t *testing.T) {
+	commandLine := &commandstest.FakeCommandLine{}
+	api := &libmachinetest.FakeAPI{}
+
+	err := cmdVersion(commandLine, api)
+
+	assert.True(t, commandLine.VersionShown)
+	assert.NoError(t, err)
+}
+
+func TestCmdVersionTooManyNames(t *testing.T) {
+	commandLine := &commandstest.FakeCommandLine{
+		CliArgs: []string{"machine1", "machine2"},
+	}
+	api := &libmachinetest.FakeAPI{}
+
+	err := cmdVersion(commandLine, api)
+
+	assert.EqualError(t, err, "Error: Expected one machine name as an argument")
+}
+
+func TestCmdVersionNotFound(t *testing.T) {
+	commandLine := &commandstest.FakeCommandLine{
+		CliArgs: []string{"unknown"},
+	}
+	api := &libmachinetest.FakeAPI{}
+
+	err := cmdVersion(commandLine, api)
+
+	assert.EqualError(t, err, `Host does not exist: "unknown"`)
+}
+
+func TestCmdVersionOnHost(t *testing.T) {
+	defer func(versioner mcndockerclient.DockerVersioner) { mcndockerclient.CurrentDockerVersioner = versioner }(mcndockerclient.CurrentDockerVersioner)
+	mcndockerclient.CurrentDockerVersioner = &mcndockerclient.FakeDockerVersioner{Version: "1.9.1"}
+
+	commandLine := &commandstest.FakeCommandLine{
+		CliArgs: []string{"machine"},
+	}
+	api := &libmachinetest.FakeAPI{
+		Hosts: []*host.Host{
+			{
+				Name: "machine",
+			},
+		},
+	}
+
+	out := &bytes.Buffer{}
+	err := printVersion(commandLine, api, out)
+
+	assert.NoError(t, err)
+	assert.Equal(t, "1.9.1\n", out.String())
+}
+
+func TestCmdVersionFailure(t *testing.T) {
+	defer func(versioner mcndockerclient.DockerVersioner) { mcndockerclient.CurrentDockerVersioner = versioner }(mcndockerclient.CurrentDockerVersioner)
+	mcndockerclient.CurrentDockerVersioner = &mcndockerclient.FakeDockerVersioner{Err: errors.New("connection failure")}
+
+	commandLine := &commandstest.FakeCommandLine{
+		CliArgs: []string{"machine"},
+	}
+	api := &libmachinetest.FakeAPI{
+		Hosts: []*host.Host{
+			{
+				Name: "machine",
+			},
+		},
+	}
+
+	out := &bytes.Buffer{}
+	err := printVersion(commandLine, api, out)
+
+	assert.EqualError(t, err, "connection failure")
+}