package swap_test import ( "testing" "github.com/kava-labs/kava/x/swap/testutil" "github.com/kava-labs/kava/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)) }