mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 03:55:19 +00:00
72a6df17fd
* add staking keeper to incentive module * update hard with delegator methods * add delegator methods to incentive * implement delegator hook scaffolds * implement hard delegator reward accumulation * update claim names to delegator * stakingKeeper expected keeper methods * accumulate delegator rewards * initialize delegator reward * synchronize delegator reward * add TODO comments to rewards * implement staking hooks interface * initial revisions * remove outdated TODO * update methods for test compatibility * update method names for test compatibility * implement initial accumulate delegator reward test * attempt validator set up in staking module * initial synchronize delegator reward test * delegator accumulation test passing * synchronize delegator rewards test (not passing) * synchronize delegator rewards passing * revisions
122 lines
4.1 KiB
Go
122 lines
4.1 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
authexported "github.com/cosmos/cosmos-sdk/x/auth/exported"
|
|
"github.com/cosmos/cosmos-sdk/x/auth/vesting"
|
|
stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
|
|
supplyexported "github.com/cosmos/cosmos-sdk/x/supply/exported"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
hardkeeper "github.com/kava-labs/kava/x/hard/keeper"
|
|
"github.com/kava-labs/kava/x/incentive/keeper"
|
|
"github.com/kava-labs/kava/x/incentive/types"
|
|
)
|
|
|
|
// Test suite used for all keeper tests
|
|
type KeeperTestSuite struct {
|
|
suite.Suite
|
|
|
|
keeper keeper.Keeper
|
|
hardKeeper hardkeeper.Keeper
|
|
stakingKeeper stakingkeeper.Keeper
|
|
app app.TestApp
|
|
ctx sdk.Context
|
|
addrs []sdk.AccAddress
|
|
validatorAddrs []sdk.ValAddress
|
|
}
|
|
|
|
// The default state used by each test
|
|
func (suite *KeeperTestSuite) SetupTest() {
|
|
tApp := app.NewTestApp()
|
|
ctx := tApp.NewContext(true, abci.Header{Height: 1, Time: tmtime.Now()})
|
|
|
|
tApp.InitializeFromGenesisStates()
|
|
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(5)
|
|
keeper := tApp.GetIncentiveKeeper()
|
|
suite.app = tApp
|
|
suite.ctx = ctx
|
|
suite.keeper = keeper
|
|
suite.addrs = addrs
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) getAccount(addr sdk.AccAddress) authexported.Account {
|
|
ak := suite.app.GetAccountKeeper()
|
|
return ak.GetAccount(suite.ctx, addr)
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) getModuleAccount(name string) supplyexported.ModuleAccountI {
|
|
sk := suite.app.GetSupplyKeeper()
|
|
return sk.GetModuleAccount(suite.ctx, name)
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) TestGetSetDeleteUSDXMintingClaim() {
|
|
c := types.NewUSDXMintingClaim(suite.addrs[0], c("ukava", 1000000), types.RewardIndexes{types.NewRewardIndex("bnb-a", sdk.ZeroDec())})
|
|
_, found := suite.keeper.GetUSDXMintingClaim(suite.ctx, suite.addrs[0])
|
|
suite.Require().False(found)
|
|
suite.Require().NotPanics(func() {
|
|
suite.keeper.SetUSDXMintingClaim(suite.ctx, c)
|
|
})
|
|
testC, found := suite.keeper.GetUSDXMintingClaim(suite.ctx, suite.addrs[0])
|
|
suite.Require().True(found)
|
|
suite.Require().Equal(c, testC)
|
|
suite.Require().NotPanics(func() {
|
|
suite.keeper.DeleteUSDXMintingClaim(suite.ctx, suite.addrs[0])
|
|
})
|
|
_, found = suite.keeper.GetUSDXMintingClaim(suite.ctx, suite.addrs[0])
|
|
suite.Require().False(found)
|
|
}
|
|
|
|
func (suite *KeeperTestSuite) TestIterateUSDXMintingClaims() {
|
|
for i := 0; i < len(suite.addrs); i++ {
|
|
c := types.NewUSDXMintingClaim(suite.addrs[i], c("ukava", 100000), types.RewardIndexes{types.NewRewardIndex("bnb-a", sdk.ZeroDec())})
|
|
suite.Require().NotPanics(func() {
|
|
suite.keeper.SetUSDXMintingClaim(suite.ctx, c)
|
|
})
|
|
}
|
|
claims := types.USDXMintingClaims{}
|
|
suite.keeper.IterateUSDXMintingClaims(suite.ctx, func(c types.USDXMintingClaim) bool {
|
|
claims = append(claims, c)
|
|
return false
|
|
})
|
|
suite.Require().Equal(len(suite.addrs), len(claims))
|
|
|
|
claims = suite.keeper.GetAllUSDXMintingClaims(suite.ctx)
|
|
suite.Require().Equal(len(suite.addrs), len(claims))
|
|
}
|
|
|
|
func createPeriodicVestingAccount(origVesting sdk.Coins, periods vesting.Periods, startTime, endTime int64) (*vesting.PeriodicVestingAccount, error) {
|
|
_, addr := app.GeneratePrivKeyAddressPairs(1)
|
|
bacc := auth.NewBaseAccountWithAddress(addr[0])
|
|
bacc.Coins = origVesting
|
|
bva, err := vesting.NewBaseVestingAccount(&bacc, origVesting, endTime)
|
|
if err != nil {
|
|
return &vesting.PeriodicVestingAccount{}, err
|
|
}
|
|
pva := vesting.NewPeriodicVestingAccountRaw(bva, startTime, periods)
|
|
err = pva.Validate()
|
|
if err != nil {
|
|
return &vesting.PeriodicVestingAccount{}, err
|
|
}
|
|
return pva, nil
|
|
}
|
|
|
|
// Avoid cluttering test cases with long function names
|
|
func i(in int64) sdk.Int { return sdk.NewInt(in) }
|
|
func d(str string) sdk.Dec { return sdk.MustNewDecFromStr(str) }
|
|
func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) }
|
|
func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) }
|
|
|
|
func TestKeeperTestSuite(t *testing.T) {
|
|
suite.Run(t, new(KeeperTestSuite))
|
|
}
|