mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 18:15:19 +00:00
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"
|
||
|
|
)
|
||
|
|
|
||
|
|
// InitializeHardSupplyRewardTests runs unit tests for the keeper.InitializeHardSupplyReward method
|
||
|
|
//
|
||
|
|
// inputs
|
||
|
|
// - claim in store if it exists (only claim.SupplyRewardIndexes)
|
||
|
|
// - global indexes in store
|
||
|
|
// - deposit function arg (only deposit.Amount)
|
||
|
|
//
|
||
|
|
// outputs
|
||
|
|
// - sets or creates a claim
|
||
|
|
type InitializeHardSupplyRewardTests struct {
|
||
|
|
unitTester
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInitializeHardSupplyReward(t *testing.T) {
|
||
|
|
suite.Run(t, new(InitializeHardSupplyRewardTests))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (suite *InitializeHardSupplyRewardTests) 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 deposit positions,
|
||
|
|
// which means UpdateHardSupplyIndexDenoms was called and should have remove indexes.
|
||
|
|
SupplyRewardIndexes: types.MultiRewardIndexes{},
|
||
|
|
}
|
||
|
|
suite.storeClaim(claim)
|
||
|
|
|
||
|
|
globalIndexes := nonEmptyMultiRewardIndexes
|
||
|
|
suite.storeGlobalSupplyIndexes(globalIndexes)
|
||
|
|
|
||
|
|
deposit := hardtypes.Deposit{
|
||
|
|
Depositor: claim.Owner,
|
||
|
|
Amount: arbitraryCoinsWithDenoms(extractCollateralTypes(globalIndexes)...),
|
||
|
|
}
|
||
|
|
|
||
|
|
suite.keeper.InitializeHardSupplyReward(suite.ctx, deposit)
|
||
|
|
|
||
|
|
syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, claim.Owner)
|
||
|
|
suite.Equal(globalIndexes, syncedClaim.SupplyRewardIndexes)
|
||
|
|
}
|
||
|
|
func (suite *InitializeHardSupplyRewardTests) TestClaimIndexesAreSetWhenClaimDoesNotExist() {
|
||
|
|
globalIndexes := nonEmptyMultiRewardIndexes
|
||
|
|
suite.storeGlobalSupplyIndexes(globalIndexes)
|
||
|
|
|
||
|
|
owner := arbitraryAddress()
|
||
|
|
deposit := hardtypes.Deposit{
|
||
|
|
Depositor: owner,
|
||
|
|
Amount: arbitraryCoinsWithDenoms(extractCollateralTypes(globalIndexes)...),
|
||
|
|
}
|
||
|
|
|
||
|
|
suite.keeper.InitializeHardSupplyReward(suite.ctx, deposit)
|
||
|
|
|
||
|
|
syncedClaim, found := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, owner)
|
||
|
|
suite.True(found)
|
||
|
|
suite.Equal(globalIndexes, syncedClaim.SupplyRewardIndexes)
|
||
|
|
}
|
||
|
|
func (suite *InitializeHardSupplyRewardTests) TestClaimIndexesAreSetEmptyForMissingIndexes() {
|
||
|
|
|
||
|
|
globalIndexes := nonEmptyMultiRewardIndexes
|
||
|
|
suite.storeGlobalSupplyIndexes(globalIndexes)
|
||
|
|
|
||
|
|
owner := arbitraryAddress()
|
||
|
|
// Supply a denom that is not in the global indexes.
|
||
|
|
// This happens when a deposit denom has no rewards associated with it.
|
||
|
|
expectedIndexes := appendUniqueEmptyMultiRewardIndex(globalIndexes)
|
||
|
|
depositedDenoms := extractCollateralTypes(expectedIndexes)
|
||
|
|
deposit := hardtypes.Deposit{
|
||
|
|
Depositor: owner,
|
||
|
|
Amount: arbitraryCoinsWithDenoms(depositedDenoms...),
|
||
|
|
}
|
||
|
|
|
||
|
|
suite.keeper.InitializeHardSupplyReward(suite.ctx, deposit)
|
||
|
|
|
||
|
|
syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, owner)
|
||
|
|
suite.Equal(expectedIndexes, syncedClaim.SupplyRewardIndexes)
|
||
|
|
}
|