mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-24 23:35:19 +00:00
88d4868316
* Simplify strategies to lend and savings * Add hard and savings keepers * Add ctx to strategy interface, fill in lend strategy * Rename lend strategy to hard * Fix hard deposit query, fix withdraw bank send * Fix misleading borrow instead of withdraw for hard * Remove liquidateall strategy method * Withdraw tests * Add hard gs to testutil suite * Update withdraw tests with working hard strategy, clean strategy interface methods * Check allowed denom for strategy * Update GetVaultTotalValue doc note * Update error wrap message for unsupported denom * Remove unnecessary viewvault keeper * Withdraw amount from account value, not supplied value * Test value > supplied withdraw * Use dec when dividing for withdrawAmountPercent * Use the correct store prefix for vault shares * Update swap references to earn * Simplify vault shares, use a single share for all coins per address
242 lines
7.9 KiB
Go
242 lines
7.9 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/kava-labs/kava/x/earn/testutil"
|
|
"github.com/kava-labs/kava/x/earn/types"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type vaultTestSuite struct {
|
|
testutil.Suite
|
|
}
|
|
|
|
func (suite *vaultTestSuite) SetupTest() {
|
|
suite.Suite.SetupTest()
|
|
suite.Keeper.SetParams(suite.Ctx, types.DefaultParams())
|
|
}
|
|
|
|
func TestVaultTestSuite(t *testing.T) {
|
|
suite.Run(t, new(vaultTestSuite))
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultTotalSupplied() {
|
|
vaultDenom := "usdx"
|
|
startBalance := sdk.NewInt64Coin(vaultDenom, 1000)
|
|
depositAmount := sdk.NewInt64Coin(vaultDenom, 100)
|
|
|
|
suite.CreateVault(vaultDenom, types.STRATEGY_TYPE_HARD)
|
|
|
|
acc := suite.CreateAccount(sdk.NewCoins(startBalance), 0)
|
|
|
|
err := suite.Keeper.Deposit(suite.Ctx, acc.GetAddress(), depositAmount)
|
|
suite.Require().NoError(err)
|
|
|
|
vaultTotalSupplied, err := suite.Keeper.GetVaultTotalSupplied(suite.Ctx, vaultDenom)
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Equal(depositAmount, vaultTotalSupplied)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultTotalSupplied_NotFound() {
|
|
vaultDenom := "usdx"
|
|
|
|
_, err := suite.Keeper.GetVaultTotalSupplied(suite.Ctx, vaultDenom)
|
|
suite.Require().Error(err)
|
|
suite.Require().ErrorIs(err, types.ErrVaultRecordNotFound)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultTotalValue() {
|
|
// TODO: After strategy implemented GetEstimatedTotalAssets
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultTotalValue_NotFound() {
|
|
vaultDenom := "usdx"
|
|
|
|
_, err := suite.Keeper.GetVaultTotalValue(suite.Ctx, vaultDenom)
|
|
suite.Require().Error(err)
|
|
suite.Require().ErrorIs(err, types.ErrVaultRecordNotFound)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultTotalValue_InvalidStrategy() {
|
|
vaultDenom := "usdx"
|
|
suite.CreateVault(vaultDenom, 99999) // not valid strategy type
|
|
|
|
_, err := suite.Keeper.GetVaultTotalValue(suite.Ctx, vaultDenom)
|
|
suite.Require().Error(err)
|
|
suite.Require().ErrorIs(err, types.ErrInvalidVaultStrategy)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultAccountSupplied() {
|
|
vaultDenom := "usdx"
|
|
startBalance := sdk.NewInt64Coin(vaultDenom, 1000)
|
|
deposit1Amount := sdk.NewInt64Coin(vaultDenom, 100)
|
|
deposit2Amount := sdk.NewInt64Coin(vaultDenom, 100)
|
|
|
|
suite.CreateVault(vaultDenom, types.STRATEGY_TYPE_HARD)
|
|
|
|
acc1 := suite.CreateAccount(sdk.NewCoins(startBalance), 0)
|
|
acc2 := suite.CreateAccount(sdk.NewCoins(startBalance), 1)
|
|
|
|
// Before deposit, account supplied is 0
|
|
|
|
_, err := suite.Keeper.GetVaultAccountSupplied(suite.Ctx, acc1.GetAddress())
|
|
suite.Require().Error(err)
|
|
suite.Require().ErrorIs(err, types.ErrVaultShareRecordNotFound)
|
|
|
|
_, err = suite.Keeper.GetVaultAccountSupplied(suite.Ctx, acc2.GetAddress())
|
|
suite.Require().Error(err)
|
|
suite.Require().ErrorIs(err, types.ErrVaultShareRecordNotFound)
|
|
|
|
// Deposits from both accounts
|
|
|
|
err = suite.Keeper.Deposit(suite.Ctx, acc1.GetAddress(), deposit1Amount)
|
|
suite.Require().NoError(err)
|
|
|
|
err = suite.Keeper.Deposit(suite.Ctx, acc2.GetAddress(), deposit2Amount)
|
|
suite.Require().NoError(err)
|
|
|
|
// Check balances
|
|
|
|
vaultAcc1Supplied, err := suite.Keeper.GetVaultAccountSupplied(suite.Ctx, acc1.GetAddress())
|
|
suite.Require().NoError(err)
|
|
|
|
vaultAcc2Supplied, err := suite.Keeper.GetVaultAccountSupplied(suite.Ctx, acc2.GetAddress())
|
|
suite.Require().NoError(err)
|
|
|
|
// Account supply only includes the deposit from respective accounts
|
|
suite.Equal(sdk.NewCoins(deposit1Amount), vaultAcc1Supplied)
|
|
suite.Equal(sdk.NewCoins(deposit1Amount), vaultAcc2Supplied)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultAccountValue() {
|
|
vaultDenom := "usdx"
|
|
startBalance := sdk.NewInt64Coin(vaultDenom, 1000)
|
|
depositAmount := sdk.NewInt64Coin(vaultDenom, 100)
|
|
|
|
acc := suite.CreateAccount(sdk.NewCoins(startBalance), 0)
|
|
|
|
suite.CreateVault(vaultDenom, types.STRATEGY_TYPE_HARD)
|
|
|
|
err := suite.Keeper.Deposit(suite.Ctx, acc.GetAddress(), depositAmount)
|
|
suite.Require().NoError(err)
|
|
|
|
accValue, err := suite.Keeper.GetVaultAccountValue(suite.Ctx, vaultDenom, acc.GetAddress())
|
|
suite.Require().NoError(err)
|
|
suite.Equal(depositAmount, accValue, "value should be same as deposit amount")
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultAccountValue_VaultNotFound() {
|
|
vaultDenom := "usdx"
|
|
acc := suite.CreateAccount(sdk.NewCoins(), 0)
|
|
|
|
_, err := suite.Keeper.GetVaultAccountValue(suite.Ctx, vaultDenom, acc.GetAddress())
|
|
suite.Require().Error(err)
|
|
suite.Require().ErrorIs(err, types.ErrVaultRecordNotFound)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultAccountValue_ShareNotFound() {
|
|
vaultDenom := "usdx"
|
|
startBalance := sdk.NewInt64Coin(vaultDenom, 1000)
|
|
depositAmount := sdk.NewInt64Coin(vaultDenom, 100)
|
|
|
|
acc1 := suite.CreateAccount(sdk.NewCoins(startBalance), 0)
|
|
acc2 := suite.CreateAccount(sdk.NewCoins(startBalance), 1)
|
|
|
|
suite.CreateVault(vaultDenom, types.STRATEGY_TYPE_HARD)
|
|
|
|
// Deposit from acc1 so that vault record exists
|
|
err := suite.Keeper.Deposit(suite.Ctx, acc1.GetAddress(), depositAmount)
|
|
suite.Require().NoError(err)
|
|
|
|
// Query from acc2 with no share record
|
|
_, err = suite.Keeper.GetVaultAccountValue(suite.Ctx, vaultDenom, acc2.GetAddress())
|
|
suite.Require().Error(err)
|
|
suite.Require().ErrorIs(err, types.ErrVaultShareRecordNotFound)
|
|
}
|
|
|
|
// ----------------------------------------------------------------------------
|
|
// State methods
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultRecord() {
|
|
record := types.NewVaultRecord("usdx")
|
|
|
|
_, found := suite.Keeper.GetVaultRecord(suite.Ctx, record.Denom)
|
|
suite.Require().False(found)
|
|
|
|
suite.Keeper.SetVaultRecord(suite.Ctx, record)
|
|
|
|
stateRecord, found := suite.Keeper.GetVaultRecord(suite.Ctx, record.Denom)
|
|
suite.Require().True(found)
|
|
suite.Require().Equal(record, stateRecord)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestUpdateVaultRecord() {
|
|
record := types.NewVaultRecord("usdx")
|
|
|
|
record.TotalSupply = sdk.NewInt64Coin("usdx", 100)
|
|
|
|
// Update vault
|
|
suite.Keeper.UpdateVaultRecord(suite.Ctx, record)
|
|
|
|
stateRecord, found := suite.Keeper.GetVaultRecord(suite.Ctx, record.Denom)
|
|
suite.Require().True(found, "vault record with supply should exist")
|
|
suite.Require().Equal(record, stateRecord)
|
|
|
|
// Remove supply
|
|
record.TotalSupply = sdk.NewInt64Coin("usdx", 0)
|
|
suite.Keeper.UpdateVaultRecord(suite.Ctx, record)
|
|
|
|
_, found = suite.Keeper.GetVaultRecord(suite.Ctx, record.Denom)
|
|
suite.Require().False(found, "vault record with 0 supply should be deleted")
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestGetVaultShareRecord() {
|
|
vaultDenom := "usdx"
|
|
startBalance := sdk.NewInt64Coin(vaultDenom, 1000)
|
|
depositAmount := sdk.NewInt64Coin(vaultDenom, 100)
|
|
acc := suite.CreateAccount(sdk.NewCoins(startBalance), 0)
|
|
|
|
record := types.NewVaultShareRecord(acc.GetAddress())
|
|
|
|
// Check share doesn't exist before deposit
|
|
|
|
_, found := suite.Keeper.GetVaultShareRecord(suite.Ctx, acc.GetAddress())
|
|
suite.Require().False(found, "vault share record should not exist before deposit")
|
|
|
|
// Update share record
|
|
record.AmountSupplied = sdk.NewCoins(depositAmount)
|
|
suite.Keeper.SetVaultShareRecord(suite.Ctx, record)
|
|
|
|
// Check share exists and matches set value
|
|
stateRecord, found := suite.Keeper.GetVaultShareRecord(suite.Ctx, acc.GetAddress())
|
|
suite.Require().True(found)
|
|
suite.Require().Equal(record, stateRecord)
|
|
}
|
|
|
|
func (suite *vaultTestSuite) TestUpdateVaultShareRecord() {
|
|
vaultDenom := "usdx"
|
|
startBalance := sdk.NewInt64Coin(vaultDenom, 1000)
|
|
depositAmount := sdk.NewInt64Coin(vaultDenom, 100)
|
|
acc := suite.CreateAccount(sdk.NewCoins(startBalance), 0)
|
|
|
|
record := types.NewVaultShareRecord(acc.GetAddress(), depositAmount)
|
|
|
|
// Update vault
|
|
suite.Keeper.UpdateVaultShareRecord(suite.Ctx, record)
|
|
|
|
stateRecord, found := suite.Keeper.GetVaultShareRecord(suite.Ctx, acc.GetAddress())
|
|
suite.Require().True(found, "vault share record with supply should exist")
|
|
suite.Require().Equal(record, stateRecord)
|
|
|
|
// Remove supply
|
|
record.AmountSupplied = sdk.NewCoins()
|
|
suite.Keeper.UpdateVaultShareRecord(suite.Ctx, record)
|
|
|
|
_, found = suite.Keeper.GetVaultShareRecord(suite.Ctx, acc.GetAddress())
|
|
suite.Require().False(found, "vault share record with 0 supply should be deleted")
|
|
}
|