fix(tests): fixed rewards tests

milkyway-labs · Nov 12, 2024 · 334f931 · 334f931
1 parent 5e428ff
commit 334f931
Show file tree

Hide file tree

Showing 16 changed files with 720 additions and 443 deletions.
diff --git a/app/testutil/test_suite.go b/app/testutil/test_suite.go
diff --git a/testutils/storetesting/keeper.go b/testutils/storetesting/keeper.go
@@ -1,6 +1,7 @@
 package storetesting
 
 import (
+	"slices"
 	"testing"
 
 	"cosmossdk.io/log"
@@ -44,7 +45,10 @@ func NewBaseKeeperTestData(t *testing.T, keys []string) BaseKeeperTestData {
 	var data BaseKeeperTestData
 
 	// Define store keys
-	data.Keys = storetypes.NewKVStoreKeys(keys...)
+	storeKeys := append(keys, []string{authtypes.StoreKey, banktypes.StoreKey}...)
+	slices.Sort(storeKeys)
+	storeKeys = slices.Compact(storeKeys)
+	data.Keys = storetypes.NewKVStoreKeys(storeKeys...)
 
 	// Setup the context
 	data.Context = BuildContext(data.Keys, nil, nil)

diff --git a/x/assets/keeper/assets_test.go b/x/assets/keeper/assets_test.go
@@ -43,7 +43,7 @@ func (suite *KeeperTestSuite) TestKeeper_SetAsset() {
 		suite.Run(tc.name, func() {
 			suite.SetupTest()
 
-			ctx, _ := suite.Ctx.CacheContext()
+			ctx, _ := suite.ctx.CacheContext()
 			if tc.store != nil {
 				tc.store(ctx)
 			}
@@ -91,7 +91,7 @@ func (suite *KeeperTestSuite) TestKeeper_GetAsset() {
 		suite.Run(tc.name, func() {
 			suite.SetupTest()
 
-			ctx, _ := suite.Ctx.CacheContext()
+			ctx, _ := suite.ctx.CacheContext()
 			if tc.store != nil {
 				tc.store(ctx)
 			}
@@ -139,7 +139,7 @@ func (suite *KeeperTestSuite) TestRemoveAsset() {
 		suite.Run(tc.name, func() {
 			suite.SetupTest()
 
-			ctx, _ := suite.Ctx.CacheContext()
+			ctx, _ := suite.ctx.CacheContext()
 			if tc.store != nil {
 				tc.store(ctx)
 			}

diff --git a/x/assets/keeper/genesis_test.go b/x/assets/keeper/genesis_test.go
@@ -40,7 +40,7 @@ func (suite *KeeperTestSuite) TestExportGenesis() {
 		suite.Run(tc.name, func() {
 			suite.SetupTest()
 
-			ctx, _ := suite.Ctx.CacheContext()
+			ctx, _ := suite.ctx.CacheContext()
 			if tc.store != nil {
 				tc.store(ctx)
 			}
@@ -94,7 +94,7 @@ func (suite *KeeperTestSuite) TestInitGenesis() {
 		suite.Run(tc.name, func() {
 			suite.SetupTest()
 
-			ctx, _ := suite.Ctx.CacheContext()
+			ctx, _ := suite.ctx.CacheContext()
 			err := suite.keeper.InitGenesis(ctx, tc.genesis)
 			if tc.shouldErr {
 				suite.Require().Error(err)

diff --git a/x/assets/keeper/grpc_query_test.go b/x/assets/keeper/grpc_query_test.go
@@ -87,7 +87,7 @@ func (suite *KeeperTestSuite) TestQuerier_Assets() {
 		suite.Run(tc.name, func() {
 			suite.SetupTest()
 
-			ctx, _ := suite.Ctx.CacheContext()
+			ctx, _ := suite.ctx.CacheContext()
 			if tc.store != nil {
 				tc.store(ctx)
 			}
@@ -140,7 +140,7 @@ func (suite *KeeperTestSuite) TestQuerier_Asset() {
 		suite.Run(tc.name, func() {
 			suite.SetupTest()
 
-			ctx, _ := suite.Ctx.CacheContext()
+			ctx, _ := suite.ctx.CacheContext()
 			if tc.store != nil {
 				tc.store(ctx)
 			}