fasthttp_attacker_mock.go

package loaderbot

import (
	"context"

	"github.com/valyala/fasthttp"
)

type FastHTTPAttackerExample struct {
	*Runner
}

func (a *FastHTTPAttackerExample) Clone(r *Runner) Attack {
	return &FastHTTPAttackerExample{Runner: r}
}

func (a *FastHTTPAttackerExample) Setup(c RunnerConfig) error {
	return nil
}

func (a *FastHTTPAttackerExample) Do(_ context.Context) DoResult {
	req := fasthttp.AcquireRequest()
	req.SetRequestURI(a.Cfg.TargetUrl)
	resp := fasthttp.AcquireResponse()
	defer fasthttp.ReleaseRequest(req)
	defer fasthttp.ReleaseResponse(resp)
	err := a.FastHTTPClient.Do(req, resp)
	if resp.StatusCode() >= 400 {
		return DoResult{
			Error: "request failed",
		}
	}
	if err != nil {
		return DoResult{
			Error: err.Error(),
		}
	}
	return DoResult{RequestLabel: a.Name}
}

func (a *FastHTTPAttackerExample) Teardown() error {
	return nil
}