LaconicNetwork/cosmos-sdk
7
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

test(x/staking): Add test for shares to tokens conversion in delegation (#20840)

Browse Source
This commit is contained in:
Lucas Francisco López 2024-07-03 14:42:11 +02:00 committed by GitHub
parent f82bb76439
commit 979f1e839b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
x/staking/keeper/delegation_test.go
View File

@ -24,6 +24,38 @@ func createValAddrs(count int) ([]sdk.AccAddress, []sdk.ValAddress) {
return addrs, valAddrs
}
func (s *KeeperTestSuite) TestSharesToTokensConversion() {
ctx, keeper := s.ctx, s.stakingKeeper
require := s.Require()
addrDels, valAddrs := createValAddrs(1)
s.accountKeeper.EXPECT().AddressCodec().Return(address.NewBech32Codec("cosmos")).AnyTimes()
initialTokens := math.NewInt(1000000)
validator := testutil.NewValidator(s.T(), valAddrs[0], PKs[0])
validator, issuedShares := validator.AddTokensFromDel(initialTokens)
require.NoError(keeper.SetValidator(ctx, validator))
// Delegate tokens
delegation := stakingtypes.NewDelegation(s.addressToString(addrDels[0]), s.valAddressToString(valAddrs[0]), issuedShares)
require.NoError(keeper.SetDelegation(ctx, delegation))
// Re-get the validator after delegation
validator, err := keeper.GetValidator(ctx, valAddrs[0])
require.NoError(err)
// Convert shares to tokens
shares := math.LegacyNewDecFromInt(initialTokens)
tokens := validator.TokensFromSharesTruncated(shares)
require.Equal(initialTokens, tokens.RoundInt())
// Convert tokens back to shares
newShares, err := validator.SharesFromTokens(initialTokens)
require.NoError(err)
require.True(shares.Equal(newShares))
}
// tests GetDelegation, GetDelegatorDelegations, SetDelegation, RemoveDelegation, GetDelegatorDelegations
func (s *KeeperTestSuite) TestDelegation() {
ctx, keeper := s.ctx, s.stakingKeeper