test: use chaintestutil (#296)

* rename * fix * format * workflow * add basic test * add targetss * use commit * workflow * basic setup * add keeper * fix * fix? * update * update * update * fix * fix * go work inti
2023-12-13 09:01:14 -05:00
parent 7e279c5daa
commit b48073d3eb
13 changed files with 586 additions and 197 deletions
@@ -0,0 +1,80 @@
+// Package networksuite provides a base test suite for tests that need a local network instance
+package networksuite
+
+import (
+	"math/rand"
+
+	"cosmossdk.io/math"
+	"github.com/cosmos/gogoproto/proto"
+	"github.com/skip-mev/chaintestutil/network"
+	"github.com/skip-mev/chaintestutil/sample"
+	"github.com/stretchr/testify/require"
+	"github.com/stretchr/testify/suite"
+
+	"github.com/skip-mev/block-sdk/lanes/base"
+	"github.com/skip-mev/block-sdk/lanes/free"
+	"github.com/skip-mev/block-sdk/lanes/mev"
+	"github.com/skip-mev/block-sdk/tests/app"
+	auctiontypes "github.com/skip-mev/block-sdk/x/auction/types"
+	blocksdktypes "github.com/skip-mev/block-sdk/x/blocksdk/types"
+)
+
+// NetworkTestSuite is a test suite for query tests that initializes a network instance.
+type NetworkTestSuite struct {
+	suite.Suite
+
+	Network       *network.Network
+	AuctionState  auctiontypes.GenesisState
+	BlockSDKState blocksdktypes.GenesisState
+}
+
+// SetupSuite setups the local network with a genesis state.
+func (nts *NetworkTestSuite) SetupSuite() {
+	var (
+		r   = sample.Rand()
+		cfg = network.NewConfig(app.AppConfig)
+	)
+
+	updateGenesisConfigState := func(moduleName string, moduleState proto.Message) {
+		buf, err := cfg.Codec.MarshalJSON(moduleState)
+		require.NoError(nts.T(), err)
+		cfg.GenesisState[moduleName] = buf
+	}
+
+	// initialize genesis
+	require.NoError(nts.T(), cfg.Codec.UnmarshalJSON(cfg.GenesisState[auctiontypes.ModuleName], &nts.AuctionState))
+	nts.AuctionState = populateAuction(r, nts.AuctionState)
+	updateGenesisConfigState(auctiontypes.ModuleName, &nts.AuctionState)
+
+	nts.BlockSDKState = populateBlockSDK(r, nts.BlockSDKState)
+	updateGenesisConfigState(blocksdktypes.ModuleName, &nts.BlockSDKState)
+
+	nts.Network = network.New(nts.T(), cfg)
+}
+
+func populateAuction(_ *rand.Rand, auctionState auctiontypes.GenesisState) auctiontypes.GenesisState {
+	// TODO intercept and populate state randomly if desired
+	return auctionState
+}
+
+func populateBlockSDK(_ *rand.Rand, bsdkState blocksdktypes.GenesisState) blocksdktypes.GenesisState {
+	bsdkState.Lanes = blocksdktypes.Lanes{
+		{
+			Id:            mev.LaneName,
+			MaxBlockSpace: math.LegacyMustNewDecFromStr("0.2"),
+			Order:         0,
+		},
+		{
+			Id:            free.LaneName,
+			MaxBlockSpace: math.LegacyMustNewDecFromStr("0.2"),
+			Order:         1,
+		},
+		{
+			Id:            base.LaneName,
+			MaxBlockSpace: math.LegacyMustNewDecFromStr("0.6"),
+			Order:         2,
+		},
+	}
+
+	return bsdkState
+}