fasthttp_echo_attacker.go

package ve_perf_tests

import (
	"context"

	"github.com/insolar/assured-ledger/ledger-core/application/testwalletapi/statemachine"
	"github.com/insolar/loaderbot"

	"github.com/insolar/ve-perf-tests/util"
)

type EchoContractTestFastHTTPAttack struct {
	*loaderbot.Runner
	client  *loaderbot.FastHTTPClient
	echoRef string
}

func (a *EchoContractTestFastHTTPAttack) Setup(cfg loaderbot.RunnerConfig) error {
	a.client = loaderbot.NewLoggingFastHTTPClient(cfg.DumpTransport)
	return nil
}
func (a *EchoContractTestFastHTTPAttack) Do(_ context.Context) loaderbot.DoResult {
	url := a.Cfg.TargetUrl + util.WalletGetBalancePath
	err := util.GetWalletBalanceFast(a.client, url, statemachine.BuiltinTestAPIEcho)
	if err != nil {
		return loaderbot.DoResult{
			Error:        err.Error(),
			RequestLabel: a.Name,
		}
	}

	return loaderbot.DoResult{
		RequestLabel: a.Name,
	}
}
func (a *EchoContractTestFastHTTPAttack) Clone(r *loaderbot.Runner) loaderbot.Attack {
	return &EchoContractTestFastHTTPAttack{Runner: r}
}

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