catch nil RewardCurrentBlockEmission and nan TotalSumPreviousTopicWei…

…ghts at genesis
allora-network · Nov 2, 2024 · 6e211de · 6e211de
1 parent 38bec6c
commit 6e211de
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/x/emissions/keeper/genesis.go b/x/emissions/keeper/genesis.go
@@ -882,6 +882,9 @@ func (k *Keeper) InitGenesis(ctx context.Context, data *types.GenesisState) erro
 	}
 
 	// TotalSumPreviousTopicWeights
+	if data.TotalSumPreviousTopicWeights.IsNaN() {
+		return errors.Wrap(nil, "error TotalSumPreviousTopicWeights is NaN")
+	}
 	if data.TotalSumPreviousTopicWeights.Gt(alloraMath.ZeroDec()) {
 		if err := k.SetTotalSumPreviousTopicWeights(ctx, data.TotalSumPreviousTopicWeights); err != nil {
 			return errors.Wrap(err, "error setting TotalSumPreviousTopicWeights")
@@ -893,6 +896,9 @@ func (k *Keeper) InitGenesis(ctx context.Context, data *types.GenesisState) erro
 	}
 
 	// RewardsCurrentBlockEmission cosmossdk_io_math.Int
+	if data.RewardCurrentBlockEmission.IsNil() {
+		return errors.Wrap(nil, "error RewardCurrentBlockEmission is nil")
+	}
 	if data.RewardCurrentBlockEmission.GT(cosmosMath.ZeroInt()) {
 		if err := k.SetRewardCurrentBlockEmission(ctx, data.RewardCurrentBlockEmission); err != nil {
 			return errors.Wrap(err, "error setting RewardCurrentBlockEmission")

diff --git a/x/emissions/keeper/genesis_test.go b/x/emissions/keeper/genesis_test.go
@@ -8,9 +8,15 @@ import (
 func (s *KeeperTestSuite) TestImportExportGenesisNoError() {
 	err := s.emissionsKeeper.SetTopicStake(s.ctx, 2, cosmossdk_io_math.OneInt())
 	s.Require().NoError(err)
+	intVal := int64(1234567890)
+	err = s.emissionsKeeper.SetRewardCurrentBlockEmission(s.ctx, cosmossdk_io_math.NewInt(intVal))
+	s.Require().NoError(err)
 	genesisState, err := s.emissionsKeeper.ExportGenesis(s.ctx)
 	s.Require().NoError(err)
 
 	err = s.emissionsKeeper.InitGenesis(s.ctx, genesisState)
 	s.Require().NoError(err)
+	rewardCurrentBlockEmission, err := s.emissionsKeeper.GetRewardCurrentBlockEmission(s.ctx)
+	s.Require().NoError(err)
+	s.Require().Equal(rewardCurrentBlockEmission.String(), cosmossdk_io_math.NewInt(intVal).String())
 }