0g-chain/x/swap/module_test.go
2024-05-01 11:17:24 +08:00

41 lines
932 B
Go

package swap_test
import (
"testing"
"github.com/0glabs/0g-chain/x/swap/testutil"
"github.com/0glabs/0g-chain/x/swap/types"
crisiskeeper "github.com/cosmos/cosmos-sdk/x/crisis/keeper"
"github.com/stretchr/testify/suite"
)
type moduleTestSuite struct {
testutil.Suite
crisisKeeper crisiskeeper.Keeper
}
func (suite *moduleTestSuite) SetupTest() {
suite.Suite.SetupTest()
suite.crisisKeeper = suite.App.GetCrisisKeeper()
}
func (suite *moduleTestSuite) TestRegisterInvariants() {
swapRoutes := []string{}
for _, route := range suite.crisisKeeper.Routes() {
if route.ModuleName == types.ModuleName {
swapRoutes = append(swapRoutes, route.Route)
}
}
suite.Contains(swapRoutes, "pool-records")
suite.Contains(swapRoutes, "share-records")
suite.Contains(swapRoutes, "pool-reserves")
suite.Contains(swapRoutes, "pool-shares")
}
func TestModuleTestSuite(t *testing.T) {
suite.Run(t, new(moduleTestSuite))
}