docs/commands/ssh_test.go

112 lines
2.6 KiB
Go

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)
}
}
}