commands/ssh: Add

Author: rajivr

commands/ssh.go

@@ -0,0 +1,51 @@
+package commands
+
+import (
+	"fmt"
+
+	"github.com/docker/machine/libmachine"
+	"github.com/docker/machine/libmachine/state"
+)
+
+type errStateInvalidForSSH struct {
+	HostName string
+}
+
+func (e errStateInvalidForSSH) Error() string {
+	return fmt.Sprintf("Error: Cannot run SSH command: Host %q is not running", e.HostName)
+}
+
+func cmdSSH(c CommandLine, api libmachine.API) error {
+	// Check for help flag -- Needed due to SkipFlagParsing
+	firstArg := c.Args().First()
+	if firstArg == "-help" || firstArg == "--help" || firstArg == "-h" {
+		c.ShowHelp()
+		return nil
+	}
+
+	target, err := targetHost(c, api)
+	if err != nil {
+		return err
+	}
+
+	host, err := api.Load(target)
+	if err != nil {
+		return err
+	}
+
+	currentState, err := host.Driver.GetState()
+	if err != nil {
+		return err
+	}
+
+	if currentState != state.Running {
+		return errStateInvalidForSSH{host.Name}
+	}
+
+	client, err := host.CreateSSHClient()
+	if err != nil {
+		return err
+	}
+
+	return client.Shell(c.Args().Tail()...)
+}

commands/ssh_test.go

@@ -0,0 +1,111 @@
+package commands
+
+import (
+	"testing"
+
+	"github.com/docker/machine/commands/commandstest"
+	"github.com/docker/machine/drivers/fakedriver"
+	"github.com/docker/machine/libmachine"
+	"github.com/docker/machine/libmachine/drivers"
+	"github.com/docker/machine/libmachine/host"
+	"github.com/docker/machine/libmachine/libmachinetest"
+	"github.com/docker/machine/libmachine/ssh"
+	"github.com/docker/machine/libmachine/ssh/sshtest"
+	"github.com/docker/machine/libmachine/state"
+	"github.com/stretchr/testify/assert"
+)
+
+type FakeSSHClientCreator struct {
+	client ssh.Client
+}
+
+func (fsc *FakeSSHClientCreator) CreateSSHClient(d drivers.Driver) (ssh.Client, error) {
+	if fsc.client == nil {
+		fsc.client = &sshtest.FakeClient{}
+	}
+	return fsc.client, nil
+}
+
+func TestCmdSSH(t *testing.T) {
+	testCases := []struct {
+		commandLine   CommandLine
+		api           libmachine.API
+		expectedErr   error
+		helpShown     bool
+		clientCreator host.SSHClientCreator
+		expectedShell []string
+	}{
+		{
+			commandLine: &commandstest.FakeCommandLine{
+				CliArgs: []string{"-h"},
+			},
+			api:         &libmachinetest.FakeAPI{},
+			expectedErr: nil,
+			helpShown:   true,
+		},
+		{
+			commandLine: &commandstest.FakeCommandLine{
+				CliArgs: []string{"--help"},
+			},
+			api:         &libmachinetest.FakeAPI{},
+			expectedErr: nil,
+			helpShown:   true,
+		},
+		{
+			commandLine: &commandstest.FakeCommandLine{
+				CliArgs: []string{},
+			},
+			api:         &libmachinetest.FakeAPI{},
+			expectedErr: ErrNoDefault,
+		},
+		{
+			commandLine: &commandstest.FakeCommandLine{
+				CliArgs: []string{"default", "df", "-h"},
+			},
+			api: &libmachinetest.FakeAPI{
+				Hosts: []*host.Host{
+					{
+						Name: "default",
+						Driver: &fakedriver.Driver{
+							MockState: state.Running,
+						},
+					},
+				},
+			},
+			expectedErr:   nil,
+			clientCreator: &FakeSSHClientCreator{},
+			expectedShell: []string{"df", "-h"},
+		},
+		{
+			commandLine: &commandstest.FakeCommandLine{
+				CliArgs: []string{"default"},
+			},
+			api: &libmachinetest.FakeAPI{
+				Hosts: []*host.Host{
+					{
+						Name: "default",
+						Driver: &fakedriver.Driver{
+							MockState: state.Stopped,
+						},
+					},
+				},
+			},
+			expectedErr: errStateInvalidForSSH{"default"},
+		},
+	}
+
+	for _, tc := range testCases {
+		host.SetSSHClientCreator(tc.clientCreator)
+
+		err := cmdSSH(tc.commandLine, tc.api)
+		assert.Equal(t, err, tc.expectedErr)
+
+		if fcl, ok := tc.commandLine.(*commandstest.FakeCommandLine); ok {
+			assert.Equal(t, tc.helpShown, fcl.HelpShown)
+		}
+
+		if fcc, ok := tc.clientCreator.(*FakeSSHClientCreator); ok {
+			assert.Equal(t, tc.expectedShell, fcc.client.(*sshtest.FakeClient).ActivatedShell)
+		}
+	}
+}