TheCloud/ssh-chat
1
0
Fork 0
mirror of https://github.com/shazow/ssh-chat.git synced 2025-09-22 10:50:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

Create function to advance scanner tokens

Browse Source
This commit is contained in:
Aaron 2016-08-06 20:38:46 +03:00
parent a4671ac262
commit 636ec68c85
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
host_test.go
View File

@ -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] ") {