/
superfluid.go
26 lines (20 loc) · 961 Bytes
/
superfluid.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
package apptesting
import (
sdk "github.com/cosmos/cosmos-sdk/types"
stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
gammtypes "github.com/osmosis-labs/osmosis/v20/x/gamm/types"
"github.com/osmosis-labs/osmosis/v20/x/superfluid/types"
)
func (s *KeeperTestHelper) SuperfluidDelegateToDefaultVal(sender sdk.AccAddress, poolId uint64, lockId uint64) error {
valAddr := s.SetupValidator(stakingtypes.Bonded)
return s.SuperfluidDelegateToVal(sender, poolId, lockId, valAddr.String())
}
func (s *KeeperTestHelper) SuperfluidDelegateToVal(sender sdk.AccAddress, poolId uint64, lockId uint64, valAddr string) error {
poolDenom := gammtypes.GetPoolShareDenom(poolId)
err := s.App.SuperfluidKeeper.AddNewSuperfluidAsset(s.Ctx, types.SuperfluidAsset{
Denom: poolDenom,
AssetType: types.SuperfluidAssetTypeLPShare,
})
s.Require().NoError(err)
return s.App.SuperfluidKeeper.SuperfluidDelegate(s.Ctx, sender.String(), lockId, valAddr)
}