Fixed issue where query only exported to file
This commit is contained in:
@@ -90,11 +90,11 @@ const (
|
||||
|
||||
func GetTable(userInput string) Table {
|
||||
switch strings.ToLower(userInput) {
|
||||
case "results":
|
||||
case "dehashed", "results":
|
||||
return ResultsTable
|
||||
case "runs":
|
||||
return RunsTable
|
||||
case "creds":
|
||||
case "users", "creds":
|
||||
return CredsTable
|
||||
case "whois":
|
||||
return WhoIsTable
|
||||
|
||||
@@ -0,0 +1,18 @@
|
||||
package sqlite
|
||||
|
||||
import "testing"
|
||||
|
||||
func TestGetTableAcceptsDisplayedTableNames(t *testing.T) {
|
||||
tests := map[string]Table{
|
||||
"dehashed": ResultsTable,
|
||||
"results": ResultsTable,
|
||||
"users": CredsTable,
|
||||
"creds": CredsTable,
|
||||
}
|
||||
|
||||
for input, want := range tests {
|
||||
if got := GetTable(input); got != want {
|
||||
t.Fatalf("GetTable(%q) = %v, want %v", input, got, want)
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user