0g-chain/x/incentive/testutil/staking_builder.go
Derrick Lee 73bc32a183
Add incentive earn tests with real keepers (#1354)
* Update incentive test to use beginblocker instead manual accumulation

* Update integration test suite

* Add base integration test, wip staking reward calculation

* Get actual staking reward amounts from BeginBlocker events, calculate expected indexes

* Simplify event parsing

* Add initial earn accum test with real keepers

* Add the rest of the accum integration tests with real keepers

* Check if delegation rewards are zero before transferring

* Update staking integration test to use updated methods
2022-10-19 16:13:37 -07:00

39 lines
989 B
Go

package testutil
import (
"github.com/cosmos/cosmos-sdk/codec"
"github.com/kava-labs/kava/app"
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
)
// StakingGenesisBuilder is a tool for creating a staking genesis state.
// Helper methods add values onto a default genesis state.
// All methods are immutable and return updated copies of the builder.
type StakingGenesisBuilder struct {
stakingtypes.GenesisState
}
var _ GenesisBuilder = (*StakingGenesisBuilder)(nil)
func NewStakingGenesisBuilder() StakingGenesisBuilder {
gen := stakingtypes.DefaultGenesisState()
gen.Params.BondDenom = "ukava"
return StakingGenesisBuilder{
GenesisState: *gen,
}
}
func (builder StakingGenesisBuilder) Build() stakingtypes.GenesisState {
return builder.GenesisState
}
func (builder StakingGenesisBuilder) BuildMarshalled(cdc codec.JSONCodec) app.GenesisState {
built := builder.Build()
return app.GenesisState{
stakingtypes.ModuleName: cdc.MustMarshalJSON(&built),
}
}