Function TestAttackSignalTwice

attack_test.go:116–158

(t *testing.T)

}

func TestAttackSignalTwice(t *testing.T) {
	t.Parallel()

	const (
		attackDuration = 10 * time.Second // The attack should never take this long.
	)

	server := httptest.NewServer(
		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {}),
	)
	defer server.Close()

	tr := vegeta.NewStaticTargeter(vegeta.Target{Method: "GET", URL: server.URL})
	atk := vegeta.NewAttacker()
	rate := vegeta.Rate{Freq: 1, Per: time.Second}

	var buf bytes.Buffer
	writer := bufio.NewWriter(&buf)
	enc := vegeta.NewEncoder(writer)
	sig := make(chan os.Signal, 1)
	res := atk.Attack(tr, rate, attackDuration, "")

	var wg sync.WaitGroup
	wg.Add(1)
	go func() {
		defer wg.Done()
		processAttack(atk, res, enc, sig, nil)
	}()

	// Exit as soon as possible.
	sig <- os.Interrupt
	sig <- os.Interrupt
	wg.Wait()
	writer.Flush()

	metrics, err := decodeMetrics(buf)
	if err != nil {
		t.Error(err)
	}
	if got, max := metrics.Duration, time.Second; got > max {
		t.Errorf("attack duration too long. got %+v, max: %+v", got, max)
	}
}