0g-chain/x/committee/keeper/integration_test.go
2024-08-03 15:03:23 +08:00

65 lines
1.9 KiB
Go

package keeper_test
import (
"time"
"github.com/cosmos/cosmos-sdk/codec"
sdk "github.com/cosmos/cosmos-sdk/types"
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
govv1beta1 "github.com/cosmos/cosmos-sdk/x/gov/types/v1beta1"
"github.com/0glabs/0g-chain/app"
"github.com/0glabs/0g-chain/x/committee/keeper"
"github.com/0glabs/0g-chain/x/committee/testutil"
"github.com/0glabs/0g-chain/x/committee/types"
)
// getProposalVoteMap collects up votes into a map indexed by proposalID
func getProposalVoteMap(k keeper.Keeper, ctx sdk.Context) map[uint64]([]types.Vote) {
proposalVoteMap := map[uint64]([]types.Vote){}
k.IterateProposals(ctx, func(p types.Proposal) bool {
proposalVoteMap[p.ID] = k.GetVotesByProposal(ctx, p.ID)
return false
})
return proposalVoteMap
}
func (suite *keeperTestSuite) getAccount(addr sdk.AccAddress) authtypes.AccountI {
ak := suite.App.GetAccountKeeper()
return ak.GetAccount(suite.Ctx, addr)
}
func mustNewTestMemberCommittee(addresses []sdk.AccAddress) *types.MemberCommittee {
com, err := types.NewMemberCommittee(
12,
"This committee is for testing.",
addresses,
[]types.Permission{&types.GodPermission{}},
testutil.D("0.667"),
time.Hour*24*7,
types.TALLY_OPTION_FIRST_PAST_THE_POST,
)
if err != nil {
panic(err)
}
return com
}
// mustNewTestProposal returns a new test proposal.
func mustNewTestProposal() types.Proposal {
proposal, err := types.NewProposal(
govv1beta1.NewTextProposal("A Title", "A description of this proposal."),
1, 1, time.Date(2010, time.January, 1, 0, 0, 0, 0, time.UTC),
)
if err != nil {
panic(err)
}
return proposal
}
// NewCommitteeGenesisState marshals a committee genesis state into json for use in initializing test apps.
func NewCommitteeGenesisState(cdc codec.Codec, gs *types.GenesisState) app.GenesisState {
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(gs)}
}