fix nits

cosmos · May 5, 2024 · d99ba32 · d99ba32
1 parent 3100806
commit d99ba32
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/x/protocolpool/keeper/genesis.go b/x/protocolpool/keeper/genesis.go
@@ -30,8 +30,8 @@ func (k Keeper) InitGenesis(ctx context.Context, data *types.GenesisState) error
 		if budget.LastClaimedAt == nil || budget.LastClaimedAt.IsZero() {
 			budget.LastClaimedAt = &currentTime
 		}
-		// ignore budgets with period <= 0
-		if budget.Period != nil && budget.Period.Seconds() <= 0 {
+		// ignore budgets with period <= 0 || nil
+		if budget.Period == nil || (budget.Period != nil && budget.Period.Seconds() <= 0) {
 			continue
 		}
 

diff --git a/x/protocolpool/keeper/msg_server_test.go b/x/protocolpool/keeper/msg_server_test.go
@@ -242,7 +242,6 @@ func (suite *KeeperTestSuite) TestMsgClaimBudget() {
 				hinfo := suite.environment.HeaderService.HeaderInfo(suite.ctx)
 				hinfo.Time = hinfo.Time.Add(240 * time.Hour)
 				suite.ctx = suite.ctx.WithHeaderInfo(hinfo)
-
 			},
 			recipientAddress: recipientAddr,
 			claimableFunds:   sdk.NewInt64Coin("foo", 100), // claiming the whole budget, 2 * 50foo = 100foo