mirror of https://github.com/docker/docs.git
65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
package digitalocean
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/docker/machine/libmachine/drivers"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSetConfigFromFlags(t *testing.T) {
|
|
driver := NewDriver("default", "path")
|
|
|
|
checkFlags := &drivers.CheckDriverOptions{
|
|
FlagsValues: map[string]interface{}{
|
|
"digitalocean-access-token": "TOKEN",
|
|
},
|
|
CreateFlags: driver.GetCreateFlags(),
|
|
}
|
|
|
|
err := driver.SetConfigFromFlags(checkFlags)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Empty(t, checkFlags.InvalidFlags)
|
|
}
|
|
|
|
func TestDefaultSSHUserAndPort(t *testing.T) {
|
|
driver := NewDriver("default", "path")
|
|
|
|
checkFlags := &drivers.CheckDriverOptions{
|
|
FlagsValues: map[string]interface{}{
|
|
"digitalocean-access-token": "TOKEN",
|
|
},
|
|
CreateFlags: driver.GetCreateFlags(),
|
|
}
|
|
|
|
err := driver.SetConfigFromFlags(checkFlags)
|
|
assert.NoError(t, err)
|
|
|
|
sshPort, err := driver.GetSSHPort()
|
|
assert.Equal(t, "root", driver.GetSSHUsername())
|
|
assert.Equal(t, 22, sshPort)
|
|
assert.NoError(t, err)
|
|
}
|
|
|
|
func TestCustomSSHUserAndPort(t *testing.T) {
|
|
driver := NewDriver("default", "path")
|
|
|
|
checkFlags := &drivers.CheckDriverOptions{
|
|
FlagsValues: map[string]interface{}{
|
|
"digitalocean-access-token": "TOKEN",
|
|
"digitalocean-ssh-user": "user",
|
|
"digitalocean-ssh-port": 2222,
|
|
},
|
|
CreateFlags: driver.GetCreateFlags(),
|
|
}
|
|
|
|
err := driver.SetConfigFromFlags(checkFlags)
|
|
assert.NoError(t, err)
|
|
|
|
sshPort, err := driver.GetSSHPort()
|
|
assert.Equal(t, "user", driver.GetSSHUsername())
|
|
assert.Equal(t, 2222, sshPort)
|
|
assert.NoError(t, err)
|
|
}
|