From c073e1f3f0fad434d86ac6f63be99dfc0b3fe535 Mon Sep 17 00:00:00 2001
From: Oliver Graff <oliver.e.graff@gmail.com>
Date: Tue, 13 Jun 2017 22:57:32 -0700
Subject: [PATCH] Fix host_test so it will not hang if the SSH connection fails

---
 host_test.go | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/host_test.go b/host_test.go
index 334506e..f98724a 100644
--- a/host_test.go
+++ b/host_test.go
@@ -97,6 +97,7 @@ func TestHostNameCollision(t *testing.T) {
 			return nil
 		})
 		if err != nil {
+			done <- struct{}{}
 			t.Fatal(err)
 		}
 	}()
@@ -203,6 +204,7 @@ func TestHostKick(t *testing.T) {
 			return nil
 		})
 		if err != nil {
+			connected <- struct{}{}
 			close(connected)
 			t.Fatal(err)
 		}
@@ -218,6 +220,7 @@ func TestHostKick(t *testing.T) {
 			return nil
 		})
 		if err != nil {
+			close(done)
 			t.Fatal(err)
 		}
 		close(done)