Extract main logic into attack function and test it

main_test.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"bytes"
+	"net/http"
+	"net/http/httptest"
+	"sync/atomic"
+	"testing"
+	"time"
+)
+
+func TestAttackRate(t *testing.T) {
+	hitCount := uint64(0)
+	server := httptest.NewServer(
+		http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+			atomic.AddUint64(&hitCount, 1)
+		}),
+	)
+	targets, err := NewTargets(bytes.NewBufferString("GET " + server.URL + "\n"))
+	if err != nil {
+		t.Fatal(err)
+	}
+	attack(targets, "random", 50, 1*time.Second, NewTextReporter())
+	if hits := atomic.LoadUint64(&hitCount); hits != 50 {
+		t.Fatalf("Wrong number of hits: want %d, got %d\n", 50, hits)
+	}
+}