mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-25 07:45:18 +00:00
229 lines
8.5 KiB
Go
229 lines
8.5 KiB
Go
package committee_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
abci "github.com/cometbft/cometbft/abci/types"
|
|
tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
govv1beta1 "github.com/cosmos/cosmos-sdk/x/gov/types/v1beta1"
|
|
|
|
"github.com/0glabs/0g-chain/app"
|
|
// "github.com/0glabs/0g-chain/x/cdp"
|
|
// cdptypes "github.com/0glabs/0g-chain/x/cdp/types"
|
|
"github.com/0glabs/0g-chain/x/committee"
|
|
"github.com/0glabs/0g-chain/x/committee/keeper"
|
|
"github.com/0glabs/0g-chain/x/committee/testutil"
|
|
"github.com/0glabs/0g-chain/x/committee/types"
|
|
)
|
|
|
|
type ModuleTestSuite struct {
|
|
suite.Suite
|
|
|
|
keeper keeper.Keeper
|
|
app app.TestApp
|
|
ctx sdk.Context
|
|
|
|
addresses []sdk.AccAddress
|
|
}
|
|
|
|
func (suite *ModuleTestSuite) SetupTest() {
|
|
suite.app = app.NewTestApp()
|
|
suite.keeper = suite.app.GetCommitteeKeeper()
|
|
suite.ctx = suite.app.NewContext(true, tmproto.Header{})
|
|
_, suite.addresses = app.GeneratePrivKeyAddressPairs(5)
|
|
}
|
|
|
|
func (suite *ModuleTestSuite) TestBeginBlock_ClosesExpired() {
|
|
suite.app.InitializeFromGenesisStates()
|
|
|
|
memberCom := types.MustNewMemberCommittee(
|
|
12,
|
|
"This committee is for testing.",
|
|
suite.addresses[:2],
|
|
[]types.Permission{&types.GodPermission{}},
|
|
testutil.D("0.8"),
|
|
time.Hour*24*7,
|
|
types.TALLY_OPTION_DEADLINE,
|
|
)
|
|
suite.keeper.SetCommittee(suite.ctx, memberCom)
|
|
|
|
pprop1 := govv1beta1.NewTextProposal("Title 1", "A description of this proposal.")
|
|
id1, err := suite.keeper.SubmitProposal(suite.ctx, memberCom.Members[0], memberCom.ID, pprop1)
|
|
suite.NoError(err)
|
|
|
|
oneHrLaterCtx := suite.ctx.WithBlockTime(suite.ctx.BlockTime().Add(time.Hour))
|
|
pprop2 := govv1beta1.NewTextProposal("Title 2", "A description of this proposal.")
|
|
id2, err := suite.keeper.SubmitProposal(oneHrLaterCtx, memberCom.Members[0], memberCom.ID, pprop2)
|
|
suite.NoError(err)
|
|
|
|
// Run BeginBlocker
|
|
proposalDurationLaterCtx := suite.ctx.WithBlockTime(suite.ctx.BlockTime().Add(memberCom.ProposalDuration))
|
|
suite.NotPanics(func() {
|
|
committee.BeginBlocker(proposalDurationLaterCtx, abci.RequestBeginBlock{}, suite.keeper)
|
|
})
|
|
|
|
// Check expired proposals are gone
|
|
_, found := suite.keeper.GetProposal(suite.ctx, id1)
|
|
suite.False(found, "expected expired proposal to be closed")
|
|
_, found = suite.keeper.GetProposal(suite.ctx, id2)
|
|
suite.True(found, "expected non expired proposal to be not closed")
|
|
}
|
|
|
|
// func (suite *ModuleTestSuite) TestBeginBlock_EnactsPassed() {
|
|
// suite.app.InitializeFromGenesisStates()
|
|
|
|
// // setup committee
|
|
// normalCom := types.MustNewMemberCommittee(12, "committee description", suite.addresses[:2],
|
|
// []types.Permission{&types.GodPermission{}}, testutil.D("0.8"), time.Hour*24*7, types.TALLY_OPTION_FIRST_PAST_THE_POST)
|
|
|
|
// suite.keeper.SetCommittee(suite.ctx, normalCom)
|
|
|
|
// // setup 2 proposals
|
|
// previousCDPDebtThreshold := suite.app.GetCDPKeeper().GetParams(suite.ctx).DebtAuctionThreshold
|
|
// newDebtThreshold := previousCDPDebtThreshold.Add(i(1000000))
|
|
// evenNewerDebtThreshold := newDebtThreshold.Add(i(1000000))
|
|
|
|
// pprop1 := params.NewParameterChangeProposal("Title 1", "A description of this proposal.",
|
|
// []params.ParamChange{{
|
|
// Subspace: cdptypes.ModuleName,
|
|
// Key: string(cdp.KeyDebtThreshold),
|
|
// Value: string(cdp.ModuleCdc.MustMarshalJSON(newDebtThreshold)),
|
|
// }},
|
|
// )
|
|
// id1, err := suite.keeper.SubmitProposal(suite.ctx, normalCom.Members[0], normalCom.ID, pprop1)
|
|
// suite.NoError(err)
|
|
|
|
// pprop2 := params.NewParameterChangeProposal("Title 2", "A description of this proposal.",
|
|
// []params.ParamChange{{
|
|
// Subspace: cdptypes.ModuleName,
|
|
// Key: string(cdp.KeyDebtThreshold),
|
|
// Value: string(cdp.ModuleCdc.MustMarshalJSON(evenNewerDebtThreshold)),
|
|
// }},
|
|
// )
|
|
// id2, err := suite.keeper.SubmitProposal(suite.ctx, normalCom.Members[0], normalCom.ID, pprop2)
|
|
// suite.NoError(err)
|
|
|
|
// // add enough votes to make the first proposal pass, but not the second
|
|
// suite.NoError(suite.keeper.AddVote(suite.ctx, id1, suite.addresses[0], types.Yes))
|
|
// suite.NoError(suite.keeper.AddVote(suite.ctx, id1, suite.addresses[1], types.Yes))
|
|
// suite.NoError(suite.keeper.AddVote(suite.ctx, id2, suite.addresses[0], types.Yes))
|
|
|
|
// // Run BeginBlocker
|
|
// suite.NotPanics(func() {
|
|
// committee.BeginBlocker(suite.ctx, abci.RequestBeginBlock{}, suite.keeper)
|
|
// })
|
|
|
|
// // Check the param has been updated
|
|
// suite.Equal(newDebtThreshold, suite.app.GetCDPKeeper().GetParams(suite.ctx).DebtAuctionThreshold)
|
|
// // Check the passed proposal has gone
|
|
// _, found := suite.keeper.GetProposal(suite.ctx, id1)
|
|
// suite.False(found, "expected passed proposal to be enacted and closed")
|
|
// _, found = suite.keeper.GetProposal(suite.ctx, id2)
|
|
// suite.True(found, "expected non passed proposal to be not closed")
|
|
// }
|
|
|
|
// func (suite *ModuleTestSuite) TestBeginBlock_DoesntEnactFailed() {
|
|
// suite.app.InitializeFromGenesisStates()
|
|
|
|
// // setup committee
|
|
// memberCom := types.MustNewMemberCommittee(12, "committee description", suite.addresses[:1],
|
|
// []types.Permission{types.SoftwareUpgradePermission{}}, testutil.D("1.0"), time.Hour*24*7, types.TALLY_OPTION_FIRST_PAST_THE_POST)
|
|
|
|
// firstBlockTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
// ctx := suite.ctx.WithBlockTime(firstBlockTime)
|
|
// suite.keeper.SetCommittee(ctx, memberCom)
|
|
|
|
// // setup an upgrade proposal
|
|
// pprop1 := upgradetypes.NewSoftwareUpgradeProposal("Title 1", "A description of this proposal.",
|
|
// upgradetypes.Plan{
|
|
// Name: "upgrade-version-v0.23.1",
|
|
// Time: firstBlockTime.Add(time.Second * 5),
|
|
// Info: "some information about the upgrade",
|
|
// },
|
|
// )
|
|
// id1, err := suite.keeper.SubmitProposal(ctx, memberCom.Members[0], memberCom.ID, pprop1)
|
|
// suite.NoError(err)
|
|
|
|
// // add enough votes to make the proposal pass
|
|
// suite.NoError(suite.keeper.AddVote(ctx, id1, suite.addresses[0], types.Yes))
|
|
|
|
// // Run BeginBlocker 10 seconds later (5 seconds after upgrade expires)
|
|
// tenSecLaterCtx := ctx.WithBlockTime(ctx.BlockTime().Add(time.Second * 10))
|
|
// suite.NotPanics(func() {
|
|
// suite.app.BeginBlocker(tenSecLaterCtx, abci.RequestBeginBlock{})
|
|
// })
|
|
|
|
// // Check the plan has not been stored
|
|
// _, found := suite.app.GetUpgradeKeeper().GetUpgradePlan(tenSecLaterCtx)
|
|
// suite.False(found)
|
|
// // Check the passed proposal has gone
|
|
// _, found = suite.keeper.GetProposal(tenSecLaterCtx, id1)
|
|
// suite.False(found, "expected failed proposal to be not enacted and closed")
|
|
|
|
// // Check the chain doesn't halt
|
|
// oneMinLaterCtx := ctx.WithBlockTime(ctx.BlockTime().Add(time.Minute).Add(time.Second))
|
|
// suite.NotPanics(func() {
|
|
// suite.app.BeginBlocker(oneMinLaterCtx, abci.RequestBeginBlock{})
|
|
// })
|
|
// }
|
|
|
|
// func (suite *ModuleTestSuite) TestBeginBlock_EnactsPassedUpgrade() {
|
|
// suite.app.InitializeFromGenesisStates()
|
|
|
|
// // setup committee
|
|
// memberCom := types.MustNewMemberCommittee(
|
|
// 12,
|
|
// "committee description",
|
|
// suite.addresses[:1],
|
|
// []types.Permission{types.SoftwareUpgradePermission{}},
|
|
// testutil.D("1.0"),
|
|
// time.Hour*24*7,
|
|
// types.TALLY_OPTION_FIRST_PAST_THE_POST,
|
|
// )
|
|
|
|
// firstBlockTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
// ctx := suite.ctx.WithBlockTime(firstBlockTime)
|
|
// suite.keeper.SetCommittee(ctx, memberCom)
|
|
|
|
// // setup an upgrade proposal
|
|
// pprop1 := upgradetypes.NewSoftwareUpgradeProposal("Title 1", "A description of this proposal.",
|
|
// upgradetypes.Plan{
|
|
// Name: "upgrade-version-v0.23.1",
|
|
// Time: firstBlockTime.Add(time.Minute * 1),
|
|
// Info: "some information about the upgrade",
|
|
// },
|
|
// )
|
|
// id1, err := suite.keeper.SubmitProposal(ctx, memberCom.Members[0], memberCom.ID, pprop1)
|
|
// suite.NoError(err)
|
|
|
|
// // add enough votes to make the proposal pass
|
|
// suite.NoError(suite.keeper.AddVote(ctx, id1, suite.addresses[0], types.Yes))
|
|
|
|
// // Run BeginBlocker
|
|
// fiveSecLaterCtx := ctx.WithBlockTime(ctx.BlockTime().Add(time.Second * 5))
|
|
// suite.NotPanics(func() {
|
|
// suite.app.BeginBlocker(fiveSecLaterCtx, abci.RequestBeginBlock{})
|
|
// })
|
|
|
|
// // Check the plan has been stored
|
|
// _, found := suite.app.GetUpgradeKeeper().GetUpgradePlan(fiveSecLaterCtx)
|
|
// suite.True(found)
|
|
// // Check the passed proposal has gone
|
|
// _, found = suite.keeper.GetProposal(fiveSecLaterCtx, id1)
|
|
// suite.False(found, "expected passed proposal to be enacted and closed")
|
|
|
|
// // Check the chain halts
|
|
// oneMinLaterCtx := ctx.WithBlockTime(ctx.BlockTime().Add(time.Minute))
|
|
// suite.Panics(func() {
|
|
// suite.app.BeginBlocker(oneMinLaterCtx, abci.RequestBeginBlock{})
|
|
// })
|
|
// }
|
|
|
|
func TestModuleTestSuite(t *testing.T) {
|
|
suite.Run(t, new(ModuleTestSuite))
|
|
}
|