75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
package main
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"log"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
// Discard default log output
|
|
log.SetOutput(ioutil.Discard)
|
|
}
|
|
|
|
func TestRateValidation(t *testing.T) {
|
|
rate, duration, targetsf, ordering, output := defaultArguments()
|
|
rate = 0
|
|
|
|
err := attack(rate, duration, targetsf, ordering, output)
|
|
if err == nil || (err != nil && !strings.HasPrefix(err.Error(), errRatePrefix)) {
|
|
t.Errorf("Rate 0 shouldn't be valid: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestDurationValidation(t *testing.T) {
|
|
rate, duration, targetsf, ordering, output := defaultArguments()
|
|
duration = 0
|
|
|
|
err := attack(rate, duration, targetsf, ordering, output)
|
|
if err == nil || (err != nil && !strings.HasPrefix(err.Error(), errDurationPrefix)) {
|
|
t.Errorf("Duration 0 shouldn't be valid: %s", err)
|
|
}
|
|
}
|
|
|
|
func TestTargetsValidation(t *testing.T) {
|
|
rate, duration, goodFile, ordering, output := defaultArguments()
|
|
|
|
// Good case
|
|
err := attack(rate, duration, goodFile, ordering, output)
|
|
if err != nil {
|
|
t.Errorf("Targets file `%s` should be valid: %s", goodFile, err)
|
|
}
|
|
|
|
// Bad case
|
|
badFile := "randomInexistingFile12345.txt"
|
|
err = attack(rate, duration, badFile, ordering, output)
|
|
if err == nil || (err != nil && !strings.HasPrefix(err.Error(), errTargetsFilePrefix)) {
|
|
t.Errorf("Targets file `%s` shouldn't be valid: %s", badFile, err)
|
|
}
|
|
}
|
|
|
|
func TestOrderingValidation(t *testing.T) {
|
|
rate, duration, targetsf, _, output := defaultArguments()
|
|
|
|
// Good cases
|
|
for _, ordering := range []string{"random", "sequential"} {
|
|
err := attack(rate, duration, targetsf, ordering, output)
|
|
if err != nil {
|
|
t.Errorf("Ordering `%s` should be valid: %s", ordering, err)
|
|
}
|
|
}
|
|
|
|
// Bad case
|
|
badOrdering := "lolcat"
|
|
err := attack(rate, duration, targetsf, badOrdering, output)
|
|
if err == nil || (err != nil && !strings.HasPrefix(err.Error(), errOrderingPrefix)) {
|
|
t.Errorf("Ordering `%s` shouldn't be valid: %s", badOrdering, err)
|
|
}
|
|
}
|
|
|
|
func defaultArguments() (uint64, time.Duration, string, string, string) {
|
|
return uint64(1000), 5 * time.Millisecond, ".targets.txt", "random", "/dev/null"
|
|
}
|