mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 10:05:18 +00:00
65 lines
1.9 KiB
Go
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)}
|
|
}
|