mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 03:55:19 +00:00
3fc2a63556
* update claim attribute type to MultiRewardIndexes * update param attribute type to MultiRewardPeriods * keeper: update params to match types * keeper: update delegator core keeper methods * keeper: update InitializeHardDelegatorReward * keeper: update SynchronizeHardDelegatorRewards * remove reward factor in favor of reward indexes * update querier * fix test: delegator init test * fix test: delegator sync test * implement delegator reward accumulation * fix test: delegator general tests * add legact types, update v0_11 -> v0_14 migration * remove duplicate import form v0_15 migration * implement v0_15incentive migration * test data and migration test * add multiple reward denoms to init/sync tests * update delegator test with multiple reward coins * clean up simulation sync * types: introduce DelegatorClaim, refactor HardClaim * add core DelegateClaim store methods * refactor delegator reward init, accumulation, sync * update hooks * update params and genesis logic * update abci * update types tests * update querier types/keeper for compile * update supply rewards tests * update borrow reward tests * update delegator reward tests * update handler/genesis test for compile * add new msg type * implement delegator claim payouts * submission + handling of new msg * implement new querier types/keeper logic * add new queries to cli/rest * update migration * register new msgs/types on codec * remove delegator syncing from hard sync method
90 lines
2.9 KiB
Go
90 lines
2.9 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
hardtypes "github.com/kava-labs/kava/x/hard/types"
|
|
"github.com/kava-labs/kava/x/incentive/types"
|
|
)
|
|
|
|
// InitializeHardBorrowRewardTests runs unit tests for the keeper.InitializeHardBorrowReward method
|
|
//
|
|
// inputs
|
|
// - claim in store if it exists (only claim.BorrowRewardIndexes)
|
|
// - global indexes in store
|
|
// - borrow function arg (only borrow.Amount)
|
|
//
|
|
// outputs
|
|
// - sets or creates a claim
|
|
type InitializeHardBorrowRewardTests struct {
|
|
unitTester
|
|
}
|
|
|
|
func TestInitializeHardBorrowReward(t *testing.T) {
|
|
suite.Run(t, new(InitializeHardBorrowRewardTests))
|
|
}
|
|
|
|
func (suite *InitializeHardBorrowRewardTests) TestClaimIndexesAreSetWhenClaimExists() {
|
|
claim := types.HardLiquidityProviderClaim{
|
|
BaseMultiClaim: types.BaseMultiClaim{
|
|
Owner: arbitraryAddress(),
|
|
},
|
|
// Indexes should always be empty when initialize is called.
|
|
// If initialize is called then the user must have repaid their borrow positions,
|
|
// which means UpdateHardBorrowIndexDenoms was called and should have remove indexes.
|
|
BorrowRewardIndexes: types.MultiRewardIndexes{},
|
|
}
|
|
suite.storeHardClaim(claim)
|
|
|
|
globalIndexes := nonEmptyMultiRewardIndexes
|
|
suite.storeGlobalBorrowIndexes(globalIndexes)
|
|
|
|
borrow := hardtypes.Borrow{
|
|
Borrower: claim.Owner,
|
|
Amount: arbitraryCoinsWithDenoms(extractCollateralTypes(globalIndexes)...),
|
|
}
|
|
|
|
suite.keeper.InitializeHardBorrowReward(suite.ctx, borrow)
|
|
|
|
syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, claim.Owner)
|
|
suite.Equal(globalIndexes, syncedClaim.BorrowRewardIndexes)
|
|
}
|
|
func (suite *InitializeHardBorrowRewardTests) TestClaimIndexesAreSetWhenClaimDoesNotExist() {
|
|
globalIndexes := nonEmptyMultiRewardIndexes
|
|
suite.storeGlobalBorrowIndexes(globalIndexes)
|
|
|
|
owner := arbitraryAddress()
|
|
borrow := hardtypes.Borrow{
|
|
Borrower: owner,
|
|
Amount: arbitraryCoinsWithDenoms(extractCollateralTypes(globalIndexes)...),
|
|
}
|
|
|
|
suite.keeper.InitializeHardBorrowReward(suite.ctx, borrow)
|
|
|
|
syncedClaim, found := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, owner)
|
|
suite.True(found)
|
|
suite.Equal(globalIndexes, syncedClaim.BorrowRewardIndexes)
|
|
}
|
|
func (suite *InitializeHardBorrowRewardTests) TestClaimIndexesAreSetEmptyForMissingIndexes() {
|
|
|
|
globalIndexes := nonEmptyMultiRewardIndexes
|
|
suite.storeGlobalBorrowIndexes(globalIndexes)
|
|
|
|
owner := arbitraryAddress()
|
|
// Borrow a denom that is not in the global indexes.
|
|
// This happens when a borrow denom has no rewards associated with it.
|
|
expectedIndexes := appendUniqueEmptyMultiRewardIndex(globalIndexes)
|
|
borrowedDenoms := extractCollateralTypes(expectedIndexes)
|
|
borrow := hardtypes.Borrow{
|
|
Borrower: owner,
|
|
Amount: arbitraryCoinsWithDenoms(borrowedDenoms...),
|
|
}
|
|
|
|
suite.keeper.InitializeHardBorrowReward(suite.ctx, borrow)
|
|
|
|
syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, owner)
|
|
suite.Equal(expectedIndexes, syncedClaim.BorrowRewardIndexes)
|
|
}
|