fix: used parameter of voting period to update gov genesis

babylonchain · Jul 25, 2024 · 5019776 · 5019776
1 parent c5d6f45
commit 5019776
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/test/e2e/initialization/config.go b/test/e2e/initialization/config.go
@@ -218,7 +218,7 @@ func initGenesis(chain *internalChain, votingPeriod, expeditedVotingPeriod time.
 		return err
 	}
 
-	err = updateModuleGenesis(appGenState, govtypes.ModuleName, &govv1.GenesisState{}, updateGovGenesis)
+	err = updateModuleGenesis(appGenState, govtypes.ModuleName, &govv1.GenesisState{}, updateGovGenesis(votingPeriod, expeditedVotingPeriod))
 	if err != nil {
 		return err
 	}
@@ -295,11 +295,12 @@ func updateBankGenesis(bankGenState *banktypes.GenesisState) {
 	})
 }
 
-func updateGovGenesis(govGenState *govv1.GenesisState) {
-	govGenState.Params.MinDeposit = sdk.NewCoins(sdk.NewCoin(BabylonDenom, sdkmath.NewInt(100)))
-
-	votingPeriod := time.Duration(time.Second * 10)
-	govGenState.Params.VotingPeriod = &votingPeriod
+func updateGovGenesis(votingPeriod, expeditedVotingPeriod time.Duration) func(govGenState *govv1.GenesisState) {
+	return func(govGenState *govv1.GenesisState) {
+		govGenState.Params.MinDeposit = sdk.NewCoins(sdk.NewCoin(BabylonDenom, sdkmath.NewInt(100)))
+		govGenState.Params.VotingPeriod = &votingPeriod
+		govGenState.Params.ExpeditedVotingPeriod = &expeditedVotingPeriod
+	}
 }
 
 func updateMintGenesis(mintGenState *minttypes.GenesisState) {