(t *testing.T)
| 449 | } |
| 450 | |
| 451 | func TestMainEmptyDbParam(t *testing.T) { |
| 452 | log.SetOutput(&bytes.Buffer{}) |
| 453 | |
| 454 | err := cmd([]string{ |
| 455 | "--query", |
| 456 | "SELECT * FROM ip_dns WHERE dns = ?", |
| 457 | }) |
| 458 | if err == nil { |
| 459 | t.Fatal(`Should throw an error`) |
| 460 | } |
| 461 | if !strings.Contains(err.Error(), "Must provide --db param") { |
| 462 | t.Fatalf(`Should throw a "Must provide --db param" error: %v`, err) |
| 463 | } |
| 464 | } |
| 465 | |
| 466 | func TestMainEmptyQueryParam(t *testing.T) { |
| 467 | log.SetOutput(&bytes.Buffer{}) |