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) }