-
Notifications
You must be signed in to change notification settings - Fork 170
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* feat: add params quota zero check for infinite quota * add unit tests * changelog * renames
- Loading branch information
1 parent
98e7e93
commit 6cc69bb
Showing
10 changed files
with
249 additions
and
65 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,56 @@ | ||
| package keeper | ||
|
|
||
| import ( | ||
| "errors" | ||
|
|
||
| sdk "github.com/cosmos/cosmos-sdk/types" | ||
|
|
||
| ltypes "github.com/umee-network/umee/v4/x/leverage/types" | ||
| ) | ||
|
|
||
| type LeverageKeeper struct { | ||
| tokenSettings map[string]ltypes.Token | ||
| } | ||
|
|
||
| func (k LeverageKeeper) GetTokenSettings(ctx sdk.Context, baseDenom string) (ltypes.Token, error) { | ||
| ts, ok := k.tokenSettings[baseDenom] | ||
| if !ok { | ||
| return ts, errors.New("token settings not found") | ||
| } | ||
| return ts, nil | ||
| } | ||
| func (k LeverageKeeper) ExchangeUToken(ctx sdk.Context, uToken sdk.Coin) (sdk.Coin, error) { | ||
| panic("not implemented") | ||
| } | ||
| func (k LeverageKeeper) DeriveExchangeRate(ctx sdk.Context, denom string) sdk.Dec { | ||
| panic("not implemented") | ||
| } | ||
|
|
||
| func NewLeverageKeeperMock(denoms ...string) LeverageKeeper { | ||
| tokenSettings := map[string]ltypes.Token{} | ||
| for _, d := range denoms { | ||
| tokenSettings[d] = ltypes.Token{ | ||
| BaseDenom: d, | ||
| SymbolDenom: d, | ||
| } | ||
| } | ||
| return LeverageKeeper{tokenSettings: tokenSettings} | ||
| } | ||
|
|
||
| type Oracle struct { | ||
| prices map[string]sdk.Dec | ||
| } | ||
|
|
||
| func (o Oracle) Price(ctx sdk.Context, denom string) (sdk.Dec, error) { | ||
| p, ok := o.prices[denom] | ||
| if !ok { | ||
| return p, ltypes.ErrNotRegisteredToken.Wrap(denom) | ||
| } | ||
| return p, nil | ||
| } | ||
|
|
||
| func NewOracleMock(denom string, price sdk.Dec) Oracle { | ||
| prices := map[string]sdk.Dec{} | ||
| prices[denom] = price | ||
| return Oracle{prices: prices} | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,108 @@ | ||
| package keeper | ||
|
|
||
| import ( | ||
| "testing" | ||
|
|
||
| sdk "github.com/cosmos/cosmos-sdk/types" | ||
| "github.com/stretchr/testify/require" | ||
|
|
||
| ibcutil "github.com/umee-network/umee/v4/util/ibc" | ||
| ) | ||
|
|
||
| func TestGetQuotas(t *testing.T) { | ||
| k := initUmeeKeeper(t) | ||
| ctx := *k.ctx | ||
|
|
||
| quotas, err := k.GetAllOutflows(ctx) | ||
| require.NoError(t, err) | ||
| require.Equal(t, len(quotas), 0) | ||
|
|
||
| setQuotas := sdk.DecCoins{sdk.NewInt64DecCoin("test_uumee", 10000)} | ||
|
|
||
| k.SetTokenOutflows(ctx, setQuotas) | ||
| quotas, err = k.GetAllOutflows(ctx) | ||
| require.NoError(t, err) | ||
| require.Equal(t, setQuotas, quotas) | ||
|
|
||
| // get the quota of denom | ||
| quota, err := k.GetTokenOutflows(ctx, setQuotas[0].Denom) | ||
| require.NoError(t, err) | ||
| require.Equal(t, quota.Denom, setQuotas[0].Denom) | ||
| } | ||
|
|
||
| func TestGetLocalDenom(t *testing.T) { | ||
| out := ibcutil.GetLocalDenom("umee") | ||
| require.Equal(t, "umee", out) | ||
| } | ||
|
|
||
| func TestCheckAndUpdateQuota(t *testing.T) { | ||
| k := initUmeeKeeper(t) | ||
|
|
||
| // initUmeeKeeper sets umee price: 2usd | ||
|
|
||
| // 1. We set the quota param to 10 and existing outflow sum to 6 USD. | ||
| // Transfer of 2 USD in Umee should work, but additional transfer of 4 USD should fail. | ||
| // | ||
| k.setQuotaParams(10, 100) | ||
| k.SetTokenOutflow(*k.ctx, sdk.NewInt64DecCoin(umee, 6)) | ||
| k.SetTotalOutflowSum(*k.ctx, sdk.NewDec(50)) | ||
|
|
||
| err := k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(1)) | ||
| require.NoError(t, err) | ||
| k.checkOutflows(umee, 8, 52) | ||
|
|
||
| // transferring 2 umee => 4USD, will exceed the quota (8+4 > 10) | ||
| err = k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(2)) | ||
| require.ErrorContains(t, err, "quota") | ||
| k.checkOutflows(umee, 8, 52) | ||
|
|
||
| // transferring 1 umee => 2USD, will will be still OK (8+2 <= 10) | ||
| err = k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(1)) | ||
| require.NoError(t, err) | ||
| k.checkOutflows(umee, 10, 54) | ||
|
|
||
| // 2. Setting TokenQuota param to 0 should unlimit the token quota check | ||
| // | ||
| k.setQuotaParams(0, 100) | ||
|
|
||
| // transferring 20 umee => 40USD, will skip the token quota check, but will update outflows | ||
| err = k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(20)) | ||
| require.NoError(t, err) | ||
| k.checkOutflows(umee, 50, 94) | ||
|
|
||
| // transferring additional 5 umee => 10USD, will fail total quota check | ||
| err = k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(5)) | ||
| require.ErrorContains(t, err, "quota") | ||
| k.checkOutflows(umee, 50, 94) | ||
|
|
||
| // 3. Setting TotalQuota param to 0 should unlimit the total quota check | ||
| // | ||
| k.setQuotaParams(0, 0) | ||
| err = k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(5)) | ||
| require.NoError(t, err) | ||
| k.checkOutflows(umee, 60, 104) | ||
|
|
||
| // 4. Setting TokenQuota to 65 | ||
| // | ||
| k.setQuotaParams(65, 0) | ||
| err = k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(1)) | ||
| require.NoError(t, err) | ||
| k.checkOutflows(umee, 62, 106) | ||
|
|
||
| err = k.CheckAndUpdateQuota(*k.ctx, umee, sdk.NewInt(2)) // exceeds token quota | ||
| require.ErrorContains(t, err, "quota") | ||
| } | ||
|
|
||
| func TestGetExchangePrice(t *testing.T) { | ||
| k := initUmeeKeeper(t) | ||
| p, err := k.getExchangePrice(*k.ctx, umee, sdk.NewInt(12)) | ||
| require.NoError(t, err) | ||
| require.Equal(t, sdk.NewDec(24), p) | ||
|
|
||
| p, err = k.getExchangePrice(*k.ctx, atom, sdk.NewInt(3)) | ||
| require.NoError(t, err) | ||
| require.Equal(t, sdk.NewDec(30), p) | ||
|
|
||
| _, err = k.getExchangePrice(*k.ctx, "notexisting", sdk.NewInt(10)) | ||
| require.ErrorContains(t, err, "not found") | ||
| } |
Oops, something went wrong.