f1cde2acec
* push * init * fix setup * format * fix test * use lane * ok * finalize * fix everything * lint fix: * Update abci/checktx/mempool_parity_check_tx.go Co-authored-by: David Terpay <35130517+davidterpay@users.noreply.github.com> * lint fix * tidy * remove * cleanup --------- Co-authored-by: David Terpay <david.terpay@gmail.com> Co-authored-by: David Terpay <35130517+davidterpay@users.noreply.github.com>
104 lines
2.7 KiB
Go
104 lines
2.7 KiB
Go
package testutils
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
"math/rand"
|
|
"time"
|
|
|
|
"cosmossdk.io/log"
|
|
"cosmossdk.io/math"
|
|
storetypes "cosmossdk.io/store/types"
|
|
"github.com/cosmos/cosmos-sdk/testutil"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
signer_extraction "github.com/skip-mev/block-sdk/v2/adapters/signer_extraction_adapter"
|
|
"github.com/skip-mev/block-sdk/v2/block/base"
|
|
"github.com/skip-mev/block-sdk/v2/lanes/mev"
|
|
testutils "github.com/skip-mev/block-sdk/v2/testutils"
|
|
)
|
|
|
|
type MEVLaneTestSuiteBase struct {
|
|
suite.Suite
|
|
|
|
EncCfg testutils.EncodingConfig
|
|
Config mev.Factory
|
|
Ctx sdk.Context
|
|
Accounts []testutils.Account
|
|
GasTokenDenom string
|
|
}
|
|
|
|
func (s *MEVLaneTestSuiteBase) SetupTest() {
|
|
// Init encoding config
|
|
s.EncCfg = testutils.CreateTestEncodingConfig()
|
|
s.Config = mev.NewDefaultAuctionFactory(s.EncCfg.TxConfig.TxDecoder(), signer_extraction.NewDefaultAdapter())
|
|
testCtx := testutil.DefaultContextWithDB(s.T(), storetypes.NewKVStoreKey("test"), storetypes.NewTransientStoreKey("transient_test"))
|
|
s.Ctx = testCtx.Ctx.WithExecMode(sdk.ExecModePrepareProposal)
|
|
s.Ctx = s.Ctx.WithBlockHeight(1)
|
|
|
|
// Init accounts
|
|
random := rand.New(rand.NewSource(time.Now().Unix()))
|
|
s.Accounts = testutils.RandomAccounts(random, 10)
|
|
s.GasTokenDenom = "stake"
|
|
}
|
|
|
|
func (s *MEVLaneTestSuiteBase) InitLane(
|
|
maxBlockSpace math.LegacyDec,
|
|
expectedExecution map[sdk.Tx]bool,
|
|
matchAll bool,
|
|
) *mev.MEVLane {
|
|
config := base.NewLaneConfig(
|
|
log.NewNopLogger(),
|
|
s.EncCfg.TxConfig.TxEncoder(),
|
|
s.EncCfg.TxConfig.TxDecoder(),
|
|
s.SetUpAnteHandler(expectedExecution),
|
|
signer_extraction.NewDefaultAdapter(),
|
|
maxBlockSpace,
|
|
)
|
|
|
|
factory := mev.NewDefaultAuctionFactory(s.EncCfg.TxConfig.TxDecoder(), signer_extraction.NewDefaultAdapter())
|
|
matchHandler := factory.MatchHandler()
|
|
if matchAll {
|
|
matchHandler = func(_ sdk.Context, _ sdk.Tx) bool {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return mev.NewMEVLane(config, factory, matchHandler)
|
|
}
|
|
|
|
func (s *MEVLaneTestSuiteBase) SetUpAnteHandler(expectedExecution map[sdk.Tx]bool) sdk.AnteHandler {
|
|
txCache := make(map[string]bool)
|
|
for tx, pass := range expectedExecution {
|
|
bz, err := s.EncCfg.TxConfig.TxEncoder()(tx)
|
|
s.Require().NoError(err)
|
|
|
|
hash := sha256.Sum256(bz)
|
|
hashStr := hex.EncodeToString(hash[:])
|
|
txCache[hashStr] = pass
|
|
}
|
|
|
|
anteHandler := func(ctx sdk.Context, tx sdk.Tx, _ bool) (newCtx sdk.Context, err error) {
|
|
bz, err := s.EncCfg.TxConfig.TxEncoder()(tx)
|
|
s.Require().NoError(err)
|
|
|
|
hash := sha256.Sum256(bz)
|
|
hashStr := hex.EncodeToString(hash[:])
|
|
|
|
pass, found := txCache[hashStr]
|
|
if !found {
|
|
return ctx, fmt.Errorf("tx not found")
|
|
}
|
|
|
|
if pass {
|
|
return ctx, nil
|
|
}
|
|
|
|
return ctx, fmt.Errorf("tx failed")
|
|
}
|
|
|
|
return anteHandler
|
|
}
|