Skip to content
GitLab
Switch branch/tag
Find file BlameHistoryPermalink
stake_test.go 28.33 KiB
1 
package keeper_test
2
3 
import (
4 
	"time"
5
6 
	abci "github.com/tendermint/tendermint/abci/types"
7
8 
	lockuptypes "github.com/osmosis-labs/osmosis/v10/x/lockup/types"
9 
	"github.com/osmosis-labs/osmosis/v10/x/superfluid/keeper"
10 
	"github.com/osmosis-labs/osmosis/v10/x/superfluid/types"
11
12 
	sdk "github.com/cosmos/cosmos-sdk/types"
13 
	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
14 
)
15
16 
type superfluidDelegation struct {
17 
	delIndex int64
18 
	valIndex int64
19 
	lpIndex  int64
20 
	lpAmount int64
21 
}
22
23 
func (suite *KeeperTestSuite) TestSuperfluidDelegate() {
24 
	testCases := []struct {
25 
		name               string
26 
		validatorStats     []stakingtypes.BondStatus
27 
		superDelegations   []superfluidDelegation
28 
		expInterDelegation []sdk.Dec
29 
	}{
30 
		{
31 
			"with single validator and single superfluid delegation",
32 
			[]stakingtypes.BondStatus{stakingtypes.Bonded},
33 
			[]superfluidDelegation{{0, 0, 0, 1000000}},
34 
			[]sdk.Dec{sdk.NewDec(10000000)}, // 50% x 20 x 1000000
35 
		},
36 
		{
37 
			"with single validator and additional superfluid delegations",
38 
			[]stakingtypes.BondStatus{stakingtypes.Bonded},
39 
			[]superfluidDelegation{{0, 0, 0, 1000000}, {0, 0, 0, 1000000}},
40 
			[]sdk.Dec{sdk.NewDec(20000000)}, // 50% x 20 x 1000000 x 2
41 
		},
42 
		{
43 
			"with multiple validators and multiple superfluid delegations",
44 
			[]stakingtypes.BondStatus{stakingtypes.Bonded, stakingtypes.Bonded},
45 
			[]superfluidDelegation{{0, 0, 0, 1000000}, {1, 1, 0, 1000000}},
46 
			[]sdk.Dec{sdk.NewDec(10000000), sdk.NewDec(10000000)}, // 50% x 20 x 1000000
47 
		},
48 
	}
49
50 
	for _, tc := range testCases {
51 
		tc := tc
52 
		suite.Run(tc.name, func() {
53 
			suite.SetupTest()
54 
			bondDenom := suite.App.StakingKeeper.BondDenom(suite.Ctx)
55
56 
			// setup validators
57 
			valAddrs := suite.SetupValidators(tc.validatorStats)
58
59 
			denoms, _ := suite.SetupGammPoolsAndSuperfluidAssets([]sdk.Dec{sdk.NewDec(20), sdk.NewDec(20)})
60
61 
			// get pre-superfluid delgations osmo supply and supplyWithOffset
62 
			presupply := suite.App.BankKeeper.GetSupply(suite.Ctx, bondDenom)
63 
			presupplyWithOffset := suite.App.BankKeeper.GetSupplyWithOffset(suite.Ctx, bondDenom)
64
65 
			// setup superfluid delegations
66 
			_, intermediaryAccs, locks := suite.setupSuperfluidDelegations(valAddrs, tc.superDelegations, denoms)
67
68 
			// ensure post-superfluid delegations osmo supplywithoffset is the same while supply is not
69 
			postsupply := suite.App.BankKeeper.GetSupply(suite.Ctx, bondDenom)
70 
			postsupplyWithOffset := suite.App.BankKeeper.GetSupplyWithOffset(suite.Ctx, bondDenom)
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
suite.Require().False(postsupply.IsEqual(presupply), "presupply: %s postsupply: %s", presupply, postsupply) suite.Require().Equal(postsupplyWithOffset.String(), presupplyWithOffset.String()) unbondingDuration := suite.App.StakingKeeper.GetParams(suite.Ctx).UnbondingTime for index, del := range tc.superDelegations { lock := locks[index] valAddr := valAddrs[del.valIndex] // check synthetic lockup creation synthLock, err := suite.App.LockupKeeper.GetSyntheticLockup(suite.Ctx, lock.ID, keeper.StakingSyntheticDenom(lock.Coins[0].Denom, valAddr.String())) suite.Require().NoError(err) suite.Require().Equal(synthLock.UnderlyingLockId, lock.ID) suite.Require().Equal(synthLock.SynthDenom, keeper.StakingSyntheticDenom(lock.Coins[0].Denom, valAddr.String())) suite.Require().Equal(synthLock.EndTime, time.Time{}) expAcc := types.NewSuperfluidIntermediaryAccount(lock.Coins[0].Denom, valAddr.String(), 0) // Check lockID connection with intermediary account intAcc := suite.App.SuperfluidKeeper.GetLockIdIntermediaryAccountConnection(suite.Ctx, lock.ID) suite.Require().Equal(intAcc.String(), expAcc.GetAccAddress().String()) } for index, expAcc := range intermediaryAccs { // check intermediary account creation gotAcc := suite.App.SuperfluidKeeper.GetIntermediaryAccount(suite.Ctx, expAcc.GetAccAddress()) suite.Require().Equal(gotAcc.Denom, expAcc.Denom) suite.Require().Equal(gotAcc.ValAddr, expAcc.ValAddr) suite.Require().GreaterOrEqual(gotAcc.GaugeId, uint64(1)) valAddr, err := sdk.ValAddressFromBech32(expAcc.ValAddr) suite.Require().NoError(err) // check gauge creation gauge, err := suite.App.IncentivesKeeper.GetGaugeByID(suite.Ctx, gotAcc.GaugeId) suite.Require().NoError(err) suite.Require().Equal(gauge.Id, gotAcc.GaugeId) suite.Require().Equal(gauge.IsPerpetual, true) suite.Require().Equal(gauge.DistributeTo, lockuptypes.QueryCondition{ LockQueryType: lockuptypes.ByDuration, Denom: keeper.StakingSyntheticDenom(expAcc.Denom, valAddr.String()), Duration: unbondingDuration, }) suite.Require().Equal(gauge.Coins, sdk.Coins(nil)) suite.Require().Equal(gauge.StartTime, suite.Ctx.BlockTime()) suite.Require().Equal(gauge.NumEpochsPaidOver, uint64(1)) suite.Require().Equal(gauge.FilledEpochs, uint64(0)) suite.Require().Equal(gauge.DistributedCoins, sdk.Coins(nil)) // check delegation from intermediary account to validator delegation, found := suite.App.StakingKeeper.GetDelegation(suite.Ctx, expAcc.GetAccAddress(), valAddr) suite.Require().True(found) suite.Require().Equal(tc.expInterDelegation[index], delegation.Shares) } // check invariant is fine reason, broken := keeper.AllInvariants(*suite.App.SuperfluidKeeper)(suite.Ctx) suite.Require().False(broken, reason) // try delegating twice with same lockup for _, lock := range locks { err := suite.App.SuperfluidKeeper.SuperfluidDelegate(suite.Ctx, lock.Owner, lock.ID, valAddrs[0].String()) suite.Require().Error(err) } }) } } func (suite *KeeperTestSuite) TestSuperfluidUndelegate() { testCases := []struct {