[test] Add load Framework (#1991)

Signed-off-by: rodrigo <[email protected]>
Co-authored-by: Aaron Buchwald <[email protected]>

Author: RodrigoVillar

cli/spam.go

@@ -31,9 +31,6 @@ var (
 	validityWindow        int64
 	hideTxs               bool
 	checkAllChains        bool
-	spamDefaults          bool
-	spamKey               string
-	clusterInfo           string
 	prometheusBaseURI     string
 	prometheusOpenBrowser bool
 	prometheusFile        string
@@ -53,7 +50,6 @@ func init() {
 		keyCmd,
 		chainCmd,
 		actionCmd,
-		spamCmd,
 		prometheusCmd,
 	)
 	rootCmd.PersistentFlags().StringVar(
@@ -152,32 +148,6 @@ func init() {
 		transferCmd,
 	)
 
-	runSpamCmd.PersistentFlags().BoolVar(
-		&spamDefaults,
-		"defaults",
-		false,
-		"use default spam parameters",
-	)
-
-	runSpamCmd.PersistentFlags().StringVar(
-		&clusterInfo,
-		"cluster-info",
-		"",
-		"output from avalanche-cli with cluster info",
-	)
-
-	runSpamCmd.PersistentFlags().StringVar(
-		&spamKey,
-		"key",
-		"",
-		"private key used to distribute funds",
-	)
-
-	// spam
-	spamCmd.AddCommand(
-		runSpamCmd,
-	)
-
 	// prometheus
 	generatePrometheusCmd.PersistentFlags().StringVar(
 		&prometheusBaseURI,

examples/morpheusvm/cmd/morpheus-cli/cmd/root.go

@@ -9,7 +9,6 @@ require (
 	github.com/onsi/ginkgo/v2 v2.13.1
 	github.com/spf13/cobra v1.8.1
 	github.com/stretchr/testify v1.10.0
-	golang.org/x/exp v0.0.0-20241215155358-4a5509556b9e
 )
 
 require (
@@ -131,6 +130,7 @@ require (
 	go.uber.org/multierr v1.11.0 // indirect
 	go.uber.org/zap v1.26.0 // indirect
 	golang.org/x/crypto v0.35.0 // indirect
+	golang.org/x/exp v0.0.0-20241215155358-4a5509556b9e // indirect
 	golang.org/x/net v0.36.0 // indirect
 	golang.org/x/oauth2 v0.21.0 // indirect
 	golang.org/x/sync v0.11.0 // indirect

examples/morpheusvm/cmd/morpheus-cli/cmd/spam.go

@@ -0,0 +1,67 @@
+// Copyright (C) 2024, Ava Labs, Inc. All rights reserved.
+// See the file LICENSE for licensing terms.
+
+package load
+
+import (
+	"context"
+	"encoding/binary"
+	"errors"
+	"time"
+
+	"github.com/ava-labs/hypersdk/chain"
+	"github.com/ava-labs/hypersdk/examples/morpheusvm/actions"
+	"github.com/ava-labs/hypersdk/fees"
+	"github.com/ava-labs/hypersdk/load"
+)
+
+var (
+	ErrTxGeneratorFundsExhausted = errors.New("tx generator funds exhausted")
+
+	_ load.TxGenerator[*chain.Transaction] = (*TxGenerator)(nil)
+)
+
+type TxGenerator struct {
+	authFactory chain.AuthFactory
+	currBalance uint64
+	ruleFactory chain.RuleFactory
+	unitPrices  fees.Dimensions
+}
+
+func NewTxGenerator(
+	authFactory chain.AuthFactory,
+	ruleFactory chain.RuleFactory,
+	currBalance uint64,
+	unitPrices fees.Dimensions,
+) *TxGenerator {
+	return &TxGenerator{
+		authFactory: authFactory,
+		ruleFactory: ruleFactory,
+		currBalance: currBalance,
+		unitPrices:  unitPrices,
+	}
+}
+
+func (t *TxGenerator) GenerateTx(context.Context) (*chain.Transaction, error) {
+	tx, err := chain.GenerateTransaction(
+		t.ruleFactory,
+		t.unitPrices,
+		time.Now().UnixMilli(),
+		[]chain.Action{
+			&actions.Transfer{
+				To:    t.authFactory.Address(),
+				Value: 1,
+				Memo:  binary.BigEndian.AppendUint64(nil, t.currBalance),
+			},
+		},
+		t.authFactory,
+	)
+	if err != nil {
+		return nil, err
+	}
+	if tx.MaxFee() > t.currBalance {
+		return nil, ErrTxGeneratorFundsExhausted
+	}
+	t.currBalance -= tx.MaxFee()
+	return tx, nil
+}