benchmark_test.go

package main

import (
	"net/http"
	"net/http/httptest"
	"sync/atomic"
	"testing"
)

func TestBenchmark(t *testing.T) {

	requests := 100
	var received int64

	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Write([]byte("hello"))
		atomic.AddInt64(&received, 1)
	}))
	defer ts.Close()

	config := &Config{
		concurrency:      10,
		requests:         requests,
		method:           "GET",
		executionTimeout: MaxExecutionTimeout,
		url:              ts.URL,
	}

	context := NewContext(config)
	context.SetInt(FieldContentSize, 5)
	benchmark := NewBenchmark(context)

	go benchmark.Run()

	go func() {
		counter := 0
		for record := range benchmark.collector {
			counter++
			if counter == requests || record.Error != nil {
				break
			}
		}
		close(context.stop)
	}()

	context.start.Wait()
	<-context.stop

	if actualReceived := atomic.LoadInt64(&received); int64(requests) != actualReceived {
		t.Fatalf("expected to send %d requests and receive %d responses, but got %d responses", requests, requests, actualReceived)
	}
}