golang框架在高并发场景中的自动化测试

package main

import (
    "context"
    "fmt"
    "sync"
    "testing"
    "time"

    "github.com/go-resty/resty/v2"
)

func createRecord(ctx context.Context, httpClient *resty.Client, records chan<- []byte, wg *sync.WaitGroup) {
    defer wg.Done()

    req := httpClient.R().SetBody("")
    resp, err := req.Post("/api/v1/records")
    if err != nil {
        fmt.Println(err)
        return
    }

    records <- resp.Body()
}

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

    httpClient := resty.New()
    records := make(chan []byte, 100)
    wg := new(sync.WaitGroup)

    const concurrency = 100
    for i := 0; i < concurrency; i++ {
        wg.Add(1)
        go createRecord(context.Background(), httpClient, records, wg)
    }

    timeout := time.After(3 * time.Second)
    ok := false
    numRequests := 0
loop:
    for {
        select {
        case <-timeout:
            ok = true
            break loop
        case record := <-records:
            numRequests++
            fmt.Println(record)
        }
    }

    close(records)
    wg.Wait()

    if !ok {
        t.Errorf("Took too long. Only %d out of %d requests processed.",
            numRequests, concurrency)
    }

    if numRequests != concurrency {
        t.Errorf("Incorrect number of requests. Got %d, expected %d.",
            numRequests, concurrency)
    }
}