From 636ec68c85ee167f059c9f9ca5cad23aad4d0628 Mon Sep 17 00:00:00 2001
From: Aaron <aaron.ounn@gmail.com>
Date: Sat, 6 Aug 2016 20:38:46 +0300
Subject: [PATCH] Create function to advance scanner tokens

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

diff --git a/host_test.go b/host_test.go
index d29a875..ca8c0bf 100644
--- a/host_test.go
+++ b/host_test.go
@@ -15,6 +15,15 @@ import (
 	"golang.org/x/crypto/ssh"
 )
 
+func nextScanToken(scanner *bufio.Scanner, i int) *bufio.Scanner {
+	count := 0
+	for count < i {
+		scanner.Scan()
+		count++
+	}
+	return scanner
+}
+
 func stripPrompt(s string) string {
 	pos := strings.LastIndex(s, "\033[K")
 	if pos < 0 {
@@ -107,9 +116,7 @@ func TestHostNameCollision(t *testing.T) {
 		scanner := bufio.NewScanner(r)
 
 		// Consume the initial buffer
-		scanner.Scan()
-		scanner.Scan()
-		scanner.Scan()
+		nextScanToken(scanner, 3)
 
 		actual := scanner.Text()
 		if !strings.HasPrefix(actual, "[Guest1] ") {