From 4afeaf7991aeecf7da7c269e03e33363b808d7ac Mon Sep 17 00:00:00 2001
From: leigh capili <stealthybox@users.noreply.github.com>
Date: Mon, 7 Oct 2019 20:29:00 -0600
Subject: [PATCH] =?UTF-8?q?Refactor=20SSH=20wait:=20Move=20Conditional.=20?=
 =?UTF-8?q?Extract=20timeout.=20Remove=20Pas=E2=80=A6=20(#474)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Refactor SSH wait: Move Conditional. Extract timeout. Remove Password Auth. Clarify error
---
 cmd/ignite/run/start.go | 64 +++++++++++++++++++----------------------
 e2e/run_test.go         |  2 --
 2 files changed, 29 insertions(+), 37 deletions(-)

diff --git a/cmd/ignite/run/start.go b/cmd/ignite/run/start.go
index 2e6e4f157..4bfe1b16a 100644
--- a/cmd/ignite/run/start.go
+++ b/cmd/ignite/run/start.go
@@ -52,8 +52,11 @@ func Start(so *startOptions) error {
 		return err
 	}
 
-	if err := waitForSSH(so.vm, 10); err != nil {
-		return err
+	// When --ssh is enabled, wait until SSH service started on port 22 at most N seconds
+	if ssh := so.vm.Spec.SSH; ssh != nil && ssh.Generate && len(so.vm.Status.IPAddresses) > 0 {
+		if err := waitForSSH(so.vm, 10, 5); err != nil {
+			return err
+		}
 	}
 
 	// If starting interactively, attach after starting
@@ -64,36 +67,32 @@ func Start(so *startOptions) error {
 }
 
 func dialSuccess(vm *ignite.VM, seconds int) error {
-	// When --ssh is enabled, wait until SSH service started on port 22 at most N seconds
-	ssh := vm.Spec.SSH
-	if ssh != nil && ssh.Generate && len(vm.Status.IPAddresses) > 0 {
-		addr := vm.Status.IPAddresses[0].String() + ":22"
-		perSecond := 10
-		delay := time.Second / time.Duration(perSecond)
-		var err error
-		for i := 0; i < seconds*perSecond; i++ {
-			conn, dialErr := net.DialTimeout("tcp", addr, delay)
-			if conn != nil {
-				defer conn.Close()
-				err = nil
-				break
-			}
-			err = dialErr
-			time.Sleep(delay)
+	addr := vm.Status.IPAddresses[0].String() + ":22"
+	perSecond := 10
+	delay := time.Second / time.Duration(perSecond)
+	var err error
+	for i := 0; i < seconds*perSecond; i++ {
+		conn, dialErr := net.DialTimeout("tcp", addr, delay)
+		if conn != nil {
+			defer conn.Close()
+			err = nil
+			break
 		}
-		if err != nil {
-			if err, ok := err.(*net.OpError); ok && err.Timeout() {
-				return fmt.Errorf("Tried connecting to SSH but timed out %s", err)
-			}
-			return err
+		err = dialErr
+		time.Sleep(delay)
+	}
+	if err != nil {
+		if err, ok := err.(*net.OpError); ok && err.Timeout() {
+			return fmt.Errorf("Tried connecting to SSH but timed out %s", err)
 		}
+		return err
 	}
 
 	return nil
 }
 
-func waitForSSH(vm *ignite.VM, seconds int) error {
-	if err := dialSuccess(vm, seconds); err != nil {
+func waitForSSH(vm *ignite.VM, dialSeconds, sshTimeout int) error {
+	if err := dialSuccess(vm, dialSeconds); err != nil {
 		return err
 	}
 
@@ -110,25 +109,20 @@ func waitForSSH(vm *ignite.VM, seconds int) error {
 	}
 
 	config := &ssh.ClientConfig{
-		User: "user",
-		Auth: []ssh.AuthMethod{
-			ssh.Password("password"),
-		},
 		HostKeyCallback: certCheck.CheckHostKey,
-		Timeout:         5 * time.Second,
+		Timeout:         time.Duration(sshTimeout) * time.Second,
 	}
 
 	addr := vm.Status.IPAddresses[0].String() + ":22"
 	sshConn, err := ssh.Dial("tcp", addr, config)
 	if err != nil {
-		// If error contains "unable to authenticate", it seems able to connect the server
-		errString := err.Error()
-		if strings.Contains(errString, "unable to authenticate") {
+		if strings.Contains(err.Error(), "unable to authenticate") {
+			// we connected to the ssh server and recieved the expected failure
 			return nil
 		}
 		return err
 	}
 
-	sshConn.Close()
-	return fmt.Errorf("timed out checking SSH server")
+	defer sshConn.Close()
+	return fmt.Errorf("waitForSSH: connected successfully with no authentication -- failure was expected")
 }
diff --git a/e2e/run_test.go b/e2e/run_test.go
index f0ca170e0..5435144e4 100644
--- a/e2e/run_test.go
+++ b/e2e/run_test.go
@@ -15,7 +15,6 @@ import (
 	"os/exec"
 	"path"
 	"testing"
-	"time"
 
 	"gotest.tools/assert"
 )
@@ -111,7 +110,6 @@ func runCurl(t *testing.T, vmName, runtime, networkPlugin string) {
 		return
 	}
 
-	time.Sleep(2 * time.Second) // TODO(https://github.com/weaveworks/ignite/issues/423): why is this necessary? Can we work to eliminate this?
 	curlCmd := exec.Command(
 		igniteBin,
 		"--runtime="+runtime,