Test the SSH driver

Author: mlafeldt

driver/ssh/driver.go

@@ -10,9 +10,9 @@ import (
 
 // Driver is a driver based on SSH.
 type Driver struct {
-	host        string
-	sshClient   *openssh.Client
-	rsyncClient *rsync.Client
+	Host        string
+	SSHClient   *openssh.Client
+	RsyncClient *rsync.Client
 }
 
 // NewDriver creates a new SSH driver that communicates with the given host.
@@ -31,15 +31,15 @@ func NewDriver(host string) (*Driver, error) {
 
 // RunCommand runs the specified command on the host.
 func (drv Driver) RunCommand(args []string) error {
-	return drv.sshClient.RunCommand(args)
+	return drv.SSHClient.RunCommand(args)
 }
 
 // Upload copies files to the host.
 func (drv Driver) Upload(dst string, src ...string) error {
-	return drv.rsyncClient.Copy(dst, src...)
+	return drv.RsyncClient.Copy(dst, src...)
 }
 
 // String returns the driver's name.
 func (drv Driver) String() string {
-	return fmt.Sprintf("SSH driver (host: %s)", drv.sshClient.Host)
+	return fmt.Sprintf("SSH driver (host: %s)", drv.SSHClient.Host)
 }

driver/ssh/driver_test.go

@@ -12,10 +12,17 @@ func TestDriverInterface(t *testing.T) {
 	assert.Implements(t, (*driver.Driver)(nil), new(Driver))
 }
 
-func TestString(t *testing.T) {
+func TestNewDriver(t *testing.T) {
 	drv, err := NewDriver("some-user@some-host:1234")
-	assert.NoError(t, err)
-	assert.Equal(t, "SSH driver (host: some-host)", drv.String())
+	if assert.NoError(t, err) {
+		assert.Equal(t, "some-host", drv.SSHClient.Host)
+		assert.Equal(t, 1234, drv.SSHClient.Port)
+		assert.Equal(t, "some-user", drv.SSHClient.User)
+		assert.Equal(t, "some-host", drv.RsyncClient.RemoteHost)
+	}
 }
 
-// TODO: test RunCommand machinery
+func TestString(t *testing.T) {
+	drv, _ := NewDriver("some-user@some-host:1234")
+	assert.Equal(t, "SSH driver (host: some-host)", drv.String())
+}