mirror of
				https://github.com/0glabs/0g-chain.git
				synced 2025-11-04 08:49:26 +00:00 
			
		
		
		
	* update savings module macc balances getter * add savings keeper to incentive module * add savings keeper to incentive module #2 * savings reward syncing * claim savings reward * update txs, queries * update txs, queries #2 * update claim test * add savings keeper to incentive module in app.go * re-commit files to disk * define and call hooks * keeper methods for init/sync savings reward * update other tests for easier extendibility * init savings reward test * add helper methods to global incentive unit tester * sync savings test progress * savings init fix + completed tests * sync savings updates + tests * nit: simplify false check * fix: calculate set difference of incoming deposit denoms Co-authored-by: karzak <kjydavis3@gmail.com>
		
			
				
	
	
		
			107 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			107 lines
		
	
	
		
			3.7 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package keeper_test
 | 
						|
 | 
						|
import (
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/suite"
 | 
						|
 | 
						|
	"github.com/kava-labs/kava/x/incentive/types"
 | 
						|
)
 | 
						|
 | 
						|
// UpdateHardSupplyIndexDenomsTests runs unit tests for the keeper.UpdateHardSupplyIndexDenoms method
 | 
						|
type UpdateHardSupplyIndexDenomsTests struct {
 | 
						|
	unitTester
 | 
						|
}
 | 
						|
 | 
						|
func TestUpdateHardSupplyIndexDenoms(t *testing.T) {
 | 
						|
	suite.Run(t, new(UpdateHardSupplyIndexDenomsTests))
 | 
						|
}
 | 
						|
 | 
						|
func (suite *UpdateHardSupplyIndexDenomsTests) TestClaimIndexesAreRemovedForDenomsNoLongerSupplied() {
 | 
						|
	claim := types.HardLiquidityProviderClaim{
 | 
						|
		BaseMultiClaim: types.BaseMultiClaim{
 | 
						|
			Owner: arbitraryAddress(),
 | 
						|
		},
 | 
						|
		SupplyRewardIndexes: nonEmptyMultiRewardIndexes,
 | 
						|
	}
 | 
						|
	suite.storeHardClaim(claim)
 | 
						|
	suite.storeGlobalSupplyIndexes(claim.SupplyRewardIndexes)
 | 
						|
 | 
						|
	// remove one denom from the indexes already in the deposit
 | 
						|
	expectedIndexes := claim.SupplyRewardIndexes[1:]
 | 
						|
	deposit := NewHardDepositBuilder(claim.Owner).
 | 
						|
		WithArbitrarySourceShares(extractCollateralTypes(expectedIndexes)...).
 | 
						|
		Build()
 | 
						|
 | 
						|
	suite.keeper.UpdateHardSupplyIndexDenoms(suite.ctx, deposit)
 | 
						|
 | 
						|
	syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, claim.Owner)
 | 
						|
	suite.Equal(expectedIndexes, syncedClaim.SupplyRewardIndexes)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *UpdateHardSupplyIndexDenomsTests) TestClaimIndexesAreAddedForNewlySuppliedDenoms() {
 | 
						|
	claim := types.HardLiquidityProviderClaim{
 | 
						|
		BaseMultiClaim: types.BaseMultiClaim{
 | 
						|
			Owner: arbitraryAddress(),
 | 
						|
		},
 | 
						|
		SupplyRewardIndexes: nonEmptyMultiRewardIndexes,
 | 
						|
	}
 | 
						|
	suite.storeHardClaim(claim)
 | 
						|
	globalIndexes := appendUniqueMultiRewardIndex(claim.SupplyRewardIndexes)
 | 
						|
	suite.storeGlobalSupplyIndexes(globalIndexes)
 | 
						|
 | 
						|
	deposit := NewHardDepositBuilder(claim.Owner).
 | 
						|
		WithArbitrarySourceShares(extractCollateralTypes(globalIndexes)...).
 | 
						|
		Build()
 | 
						|
 | 
						|
	suite.keeper.UpdateHardSupplyIndexDenoms(suite.ctx, deposit)
 | 
						|
 | 
						|
	syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, claim.Owner)
 | 
						|
	suite.Equal(globalIndexes, syncedClaim.SupplyRewardIndexes)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *UpdateHardSupplyIndexDenomsTests) TestClaimIndexesAreUnchangedWhenSuppliedDenomsUnchanged() {
 | 
						|
	claim := types.HardLiquidityProviderClaim{
 | 
						|
		BaseMultiClaim: types.BaseMultiClaim{
 | 
						|
			Owner: arbitraryAddress(),
 | 
						|
		},
 | 
						|
		SupplyRewardIndexes: nonEmptyMultiRewardIndexes,
 | 
						|
	}
 | 
						|
	suite.storeHardClaim(claim)
 | 
						|
	// Set global indexes with same denoms but different values.
 | 
						|
	// UpdateHardSupplyIndexDenoms should ignore the new values.
 | 
						|
	suite.storeGlobalSupplyIndexes(increaseAllRewardFactors(claim.SupplyRewardIndexes))
 | 
						|
 | 
						|
	deposit := NewHardDepositBuilder(claim.Owner).
 | 
						|
		WithArbitrarySourceShares(extractCollateralTypes(claim.SupplyRewardIndexes)...).
 | 
						|
		Build()
 | 
						|
 | 
						|
	suite.keeper.UpdateHardSupplyIndexDenoms(suite.ctx, deposit)
 | 
						|
 | 
						|
	syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, claim.Owner)
 | 
						|
	suite.Equal(claim.SupplyRewardIndexes, syncedClaim.SupplyRewardIndexes)
 | 
						|
}
 | 
						|
 | 
						|
func (suite *UpdateHardSupplyIndexDenomsTests) TestEmptyClaimIndexesAreAddedForNewlySuppliedButNotRewardedDenoms() {
 | 
						|
	claim := types.HardLiquidityProviderClaim{
 | 
						|
		BaseMultiClaim: types.BaseMultiClaim{
 | 
						|
			Owner: arbitraryAddress(),
 | 
						|
		},
 | 
						|
		SupplyRewardIndexes: nonEmptyMultiRewardIndexes,
 | 
						|
	}
 | 
						|
	suite.storeHardClaim(claim)
 | 
						|
	suite.storeGlobalSupplyIndexes(claim.SupplyRewardIndexes)
 | 
						|
 | 
						|
	// add a denom to the deposited amount that is not in the global or claim's indexes
 | 
						|
	expectedIndexes := appendUniqueEmptyMultiRewardIndex(claim.SupplyRewardIndexes)
 | 
						|
	depositedDenoms := extractCollateralTypes(expectedIndexes)
 | 
						|
	deposit := NewHardDepositBuilder(claim.Owner).
 | 
						|
		WithArbitrarySourceShares(depositedDenoms...).
 | 
						|
		Build()
 | 
						|
 | 
						|
	suite.keeper.UpdateHardSupplyIndexDenoms(suite.ctx, deposit)
 | 
						|
 | 
						|
	syncedClaim, _ := suite.keeper.GetHardLiquidityProviderClaim(suite.ctx, claim.Owner)
 | 
						|
	suite.Equal(expectedIndexes, syncedClaim.SupplyRewardIndexes)
 | 
						|
}
 |