fix tests

cosmos · Jul 21, 2023 · 6545c84 · 6545c84
1 parent 49cb0ac
commit 6545c84
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/...ation/staking/keeper/determinstic_test.go → ...tion/staking/keeper/deterministic_test.go b/...ation/staking/keeper/determinstic_test.go → ...tion/staking/keeper/deterministic_test.go
@@ -671,7 +671,7 @@ func TestGRPCHistoricalInfo(t *testing.T) {
 
 		assert.NilError(t, f.stakingKeeper.HistoricalInfo.Set(
 			f.ctx,
-			height,
+			stakingtypes.GetHistoricalInfoKeyWithoutPrefix(height),
 			historicalInfo,
 		))
 
@@ -695,7 +695,7 @@ func TestGRPCHistoricalInfo(t *testing.T) {
 
 	assert.NilError(t, f.stakingKeeper.HistoricalInfo.Set(
 		f.ctx,
-		height,
+		stakingtypes.GetHistoricalInfoKeyWithoutPrefix(height),
 		historicalInfo,
 	))
 

diff --git a/tests/integration/staking/keeper/grpc_query_test.go b/tests/integration/staking/keeper/grpc_query_test.go
@@ -29,7 +29,7 @@ func createValidatorAccs(t *testing.T, f *fixture) ([]sdk.AccAddress, []types.Va
 	sortedVals := make([]types.Validator, len(validators))
 	copy(sortedVals, validators)
 	hi := types.NewHistoricalInfo(header, sortedVals, f.stakingKeeper.PowerReduction(f.sdkCtx))
-	assert.NilError(t, f.stakingKeeper.HistoricalInfo.Set(f.sdkCtx, 5, hi))
+	assert.NilError(t, f.stakingKeeper.HistoricalInfo.Set(f.sdkCtx, types.GetHistoricalInfoKeyWithoutPrefix(5), hi))
 
 	return addrs, validators
 }