bettercap/session/session_test.go
eenblam 3c3ed30001 Add tests for Session.ParseCommands()
Small tests, but can be expanded to table of cases
2018-08-31 14:09:54 -07:00

38 lines
1 KiB
Go

package session
import (
"fmt"
"testing"
)
func TestParseCommands(t *testing.T) {
//commands := ParseCommands("wifi.recon on; asdf; \"asdf;\" asdf")
t.Run("handles a semicolon as a delimiter", func(t *testing.T) {
first := "wifi.recon on"
second := "wifi.ap"
cmd := fmt.Sprintf("%s; %s", first, second)
commands := ParseCommands(cmd)
if l := len(commands); l != 2 {
t.Fatalf("Expected 2 commands, got %d", l)
}
if got := commands[0]; got != first {
t.Fatalf("expected %s got %s", first, got)
}
if got := commands[1]; got != second {
t.Fatalf("expected %s got %s", second, got)
}
})
t.Run("handles semicolon inside quotes", func(t *testing.T) {
cmd := "set ticker.commands \"clear; net.show\""
commands := ParseCommands(cmd)
if l := len(commands); l != 1 {
t.Fatalf("expected 1 command, got %d", l)
}
// Expect double-quotes stripped
expected := "set ticker.commands clear; net.show"
if got := commands[0]; got != expected {
fmt.Println(got)
t.Fatalf("expected %s got %s", cmd, got)
}
})
}