mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-29 17:55:19 +00:00
782 lines
24 KiB
Go
782 lines
24 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
earntypes "github.com/0glabs/0g-chain/x/earn/types"
|
|
"github.com/0glabs/0g-chain/x/incentive/types"
|
|
)
|
|
|
|
type AccumulateEarnRewardsTests struct {
|
|
unitTester
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) storedTimeEquals(vaultDenom string, expected time.Time) {
|
|
storedTime, found := suite.keeper.GetEarnRewardAccrualTime(suite.ctx, vaultDenom)
|
|
suite.Equal(found, expected != time.Time{}, "expected time is %v but time found = %v", expected, found)
|
|
if found {
|
|
suite.Equal(expected, storedTime)
|
|
} else {
|
|
suite.Empty(storedTime)
|
|
}
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) storedIndexesEqual(vaultDenom string, expected types.RewardIndexes) {
|
|
storedIndexes, found := suite.keeper.GetEarnRewardIndexes(suite.ctx, vaultDenom)
|
|
suite.Equal(found, expected != nil, "expected indexes is %v but indexes found = %v", expected, found)
|
|
if found {
|
|
suite.Equal(expected, storedIndexes)
|
|
} else {
|
|
suite.Empty(storedIndexes)
|
|
}
|
|
}
|
|
|
|
func TestAccumulateEarnRewards(t *testing.T) {
|
|
suite.Run(t, new(AccumulateEarnRewardsTests))
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestStateUpdatedWhenBlockTimeHasIncreased() {
|
|
vaultDenom := "usdx"
|
|
|
|
earnKeeper := newFakeEarnKeeper().addVault(vaultDenom, earntypes.NewVaultShare(vaultDenom, d("1000000")))
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, nil, earnKeeper)
|
|
|
|
suite.storeGlobalEarnIndexes(types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
})
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom, previousAccrualTime)
|
|
|
|
newAccrualTime := previousAccrualTime.Add(1 * time.Hour)
|
|
suite.ctx = suite.ctx.WithBlockTime(newAccrualTime)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
vaultDenom,
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)), // same denoms as in global indexes
|
|
)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// check time and factors
|
|
|
|
suite.storedTimeEquals(vaultDenom, newAccrualTime)
|
|
suite.storedIndexesEqual(vaultDenom, types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("7.22"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("3.64"),
|
|
},
|
|
})
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestStateUpdatedWhenBlockTimeHasIncreased_bkava() {
|
|
vaultDenom1 := "bkava-meow"
|
|
vaultDenom2 := "bkava-woof"
|
|
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.ctx.WithBlockTime(previousAccrualTime)
|
|
|
|
earnKeeper := newFakeEarnKeeper().
|
|
addVault(vaultDenom1, earntypes.NewVaultShare(vaultDenom1, d("800000"))).
|
|
addVault(vaultDenom2, earntypes.NewVaultShare(vaultDenom2, d("200000")))
|
|
|
|
liquidKeeper := newFakeLiquidKeeper().
|
|
addDerivative(suite.ctx, vaultDenom1, i(800000)).
|
|
addDerivative(suite.ctx, vaultDenom2, i(200000))
|
|
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, liquidKeeper, earnKeeper)
|
|
|
|
globalIndexes := types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom1,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
CollateralType: vaultDenom2,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
suite.storeGlobalEarnIndexes(globalIndexes)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom1, previousAccrualTime)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom2, previousAccrualTime)
|
|
|
|
newAccrualTime := previousAccrualTime.Add(1 * time.Hour)
|
|
suite.ctx = suite.ctx.WithBlockTime(newAccrualTime)
|
|
|
|
rewardPeriod := types.NewMultiRewardPeriod(
|
|
true,
|
|
"bkava", // reward period is set for "bkava" to apply to all vaults
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)), // same denoms as in global indexes
|
|
)
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, rewardPeriod)
|
|
|
|
// check time and factors
|
|
|
|
suite.storedTimeEquals(vaultDenom1, newAccrualTime)
|
|
suite.storedTimeEquals(vaultDenom2, newAccrualTime)
|
|
|
|
// Each vault gets the same ukava per second, assuming shares prices are the same.
|
|
// The share amount determines how much is actually distributed to the vault.
|
|
expectedIndexes := types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("7.22"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("3.64"). // base incentive
|
|
Add(d("360")), // staking rewards, 10% of total bkava per second
|
|
},
|
|
}
|
|
|
|
suite.storedIndexesEqual(vaultDenom1, expectedIndexes)
|
|
suite.storedIndexesEqual(vaultDenom2, expectedIndexes)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestStateUpdatedWhenBlockTimeHasIncreased_bkava_partialDeposit() {
|
|
vaultDenom1 := "bkava-meow"
|
|
vaultDenom2 := "bkava-woof"
|
|
|
|
vaultDenom1Supply := i(800000)
|
|
vaultDenom2Supply := i(200000)
|
|
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.ctx.WithBlockTime(previousAccrualTime)
|
|
|
|
liquidKeeper := newFakeLiquidKeeper().
|
|
addDerivative(suite.ctx, vaultDenom1, vaultDenom1Supply).
|
|
addDerivative(suite.ctx, vaultDenom2, vaultDenom2Supply)
|
|
|
|
vault1Shares := d("700000")
|
|
vault2Shares := d("100000")
|
|
|
|
// More bkava minted than deposited into earn
|
|
// Rewards are higher per-share as a result
|
|
earnKeeper := newFakeEarnKeeper().
|
|
addVault(vaultDenom1, earntypes.NewVaultShare(vaultDenom1, vault1Shares)).
|
|
addVault(vaultDenom2, earntypes.NewVaultShare(vaultDenom2, vault2Shares))
|
|
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, liquidKeeper, earnKeeper)
|
|
|
|
globalIndexes := types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom1,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
CollateralType: vaultDenom2,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
suite.storeGlobalEarnIndexes(globalIndexes)
|
|
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom1, previousAccrualTime)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom2, previousAccrualTime)
|
|
|
|
newAccrualTime := previousAccrualTime.Add(1 * time.Hour)
|
|
suite.ctx = suite.ctx.WithBlockTime(newAccrualTime)
|
|
|
|
rewardPeriod := types.NewMultiRewardPeriod(
|
|
true,
|
|
"bkava", // reward period is set for "bkava" to apply to all vaults
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)), // same denoms as in global indexes
|
|
)
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, rewardPeriod)
|
|
|
|
// check time and factors
|
|
|
|
suite.storedTimeEquals(vaultDenom1, newAccrualTime)
|
|
suite.storedTimeEquals(vaultDenom2, newAccrualTime)
|
|
|
|
// Slightly increased rewards due to less bkava deposited
|
|
suite.storedIndexesEqual(vaultDenom1, types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("8.248571428571428571"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("4.154285714285714286"). // base incentive
|
|
Add(sdk.NewDecFromInt(vaultDenom1Supply). // staking rewards
|
|
QuoInt64(10).
|
|
MulInt64(3600).
|
|
Quo(vault1Shares),
|
|
),
|
|
},
|
|
})
|
|
|
|
// Much higher rewards per share because only a small amount of bkava is
|
|
// deposited. The **total** amount of incentives distributed to this vault
|
|
// is still the same proportional amount.
|
|
|
|
// Fixed amount total rewards distributed to the vault
|
|
// Fewer shares deposited -> higher rewards per share
|
|
|
|
// 7.2ukava shares per second for 1 hour (started with 0.04)
|
|
// total rewards claimable = 7.2 * 100000 shares = 720000 ukava
|
|
|
|
// 720000ukava distributed which is 20% of total bkava ukava rewards
|
|
// total rewards for *all* bkava vaults for 1 hour
|
|
// = 1000ukava per second * 3600 == 3600000ukava
|
|
// vaultDenom2 has 20% of the total bkava amount so it should get 20% of 3600000ukava == 720000ukava
|
|
|
|
vault2expectedIndexes := types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("14.42"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("7.24").
|
|
Add(sdk.NewDecFromInt(vaultDenom2Supply).
|
|
QuoInt64(10).
|
|
MulInt64(3600).
|
|
Quo(vault2Shares),
|
|
),
|
|
},
|
|
}
|
|
suite.storedIndexesEqual(vaultDenom2, vault2expectedIndexes)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestStateUnchangedWhenBlockTimeHasNotIncreased() {
|
|
vaultDenom := "usdx"
|
|
|
|
earnKeeper := newFakeEarnKeeper().addVault(vaultDenom, earntypes.NewVaultShare(vaultDenom, d("1000000")))
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, nil, earnKeeper)
|
|
|
|
previousIndexes := types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
suite.storeGlobalEarnIndexes(previousIndexes)
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom, previousAccrualTime)
|
|
|
|
suite.ctx = suite.ctx.WithBlockTime(previousAccrualTime)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
vaultDenom,
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)), // same denoms as in global indexes
|
|
)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// check time and factors
|
|
|
|
suite.storedTimeEquals(vaultDenom, previousAccrualTime)
|
|
expected, f := previousIndexes.Get(vaultDenom)
|
|
suite.True(f)
|
|
suite.storedIndexesEqual(vaultDenom, expected)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestStateUnchangedWhenBlockTimeHasNotIncreased_bkava() {
|
|
vaultDenom1 := "bkava-meow"
|
|
vaultDenom2 := "bkava-woof"
|
|
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.ctx.WithBlockTime(previousAccrualTime)
|
|
|
|
earnKeeper := newFakeEarnKeeper().
|
|
addVault(vaultDenom1, earntypes.NewVaultShare(vaultDenom1, d("1000000"))).
|
|
addVault(vaultDenom2, earntypes.NewVaultShare(vaultDenom2, d("1000000")))
|
|
|
|
liquidKeeper := newFakeLiquidKeeper().
|
|
addDerivative(suite.ctx, vaultDenom1, i(1000000)).
|
|
addDerivative(suite.ctx, vaultDenom2, i(1000000))
|
|
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, liquidKeeper, earnKeeper)
|
|
|
|
previousIndexes := types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom1,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
CollateralType: vaultDenom2,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
suite.storeGlobalEarnIndexes(previousIndexes)
|
|
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom1, previousAccrualTime)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom2, previousAccrualTime)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
"bkava",
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)), // same denoms as in global indexes
|
|
)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// check time and factors
|
|
|
|
suite.storedTimeEquals(vaultDenom1, previousAccrualTime)
|
|
suite.storedTimeEquals(vaultDenom2, previousAccrualTime)
|
|
|
|
expected, f := previousIndexes.Get(vaultDenom1)
|
|
suite.True(f)
|
|
suite.storedIndexesEqual(vaultDenom1, expected)
|
|
|
|
expected, f = previousIndexes.Get(vaultDenom2)
|
|
suite.True(f)
|
|
suite.storedIndexesEqual(vaultDenom2, expected)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestNoAccumulationWhenSourceSharesAreZero() {
|
|
vaultDenom := "usdx"
|
|
|
|
earnKeeper := newFakeEarnKeeper() // no vault, so no source shares
|
|
liquidKeeper := newFakeLiquidKeeper()
|
|
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, liquidKeeper, earnKeeper)
|
|
|
|
previousIndexes := types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
suite.storeGlobalEarnIndexes(previousIndexes)
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom, previousAccrualTime)
|
|
|
|
firstAccrualTime := previousAccrualTime.Add(7 * time.Second)
|
|
suite.ctx = suite.ctx.WithBlockTime(firstAccrualTime)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
vaultDenom,
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)), // same denoms as in global indexes
|
|
)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// check time and factors
|
|
|
|
suite.storedTimeEquals(vaultDenom, firstAccrualTime)
|
|
expected, f := previousIndexes.Get(vaultDenom)
|
|
suite.True(f)
|
|
suite.storedIndexesEqual(vaultDenom, expected)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestNoAccumulationWhenSourceSharesAreZero_bkava() {
|
|
vaultDenom1 := "bkava-meow"
|
|
vaultDenom2 := "bkava-woof"
|
|
|
|
earnKeeper := newFakeEarnKeeper() // no vault, so no source shares
|
|
liquidKeeper := newFakeLiquidKeeper()
|
|
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, liquidKeeper, earnKeeper)
|
|
|
|
previousIndexes := types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom1,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
{
|
|
CollateralType: vaultDenom2,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
suite.storeGlobalEarnIndexes(previousIndexes)
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom1, previousAccrualTime)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom2, previousAccrualTime)
|
|
|
|
firstAccrualTime := previousAccrualTime.Add(7 * time.Second)
|
|
suite.ctx = suite.ctx.WithBlockTime(firstAccrualTime)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
"bkava",
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)), // same denoms as in global indexes
|
|
)
|
|
|
|
// TODO: There are no bkava vaults to iterate over, so the accrual times are
|
|
// not updated
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// check time and factors
|
|
|
|
suite.storedTimeEquals(vaultDenom1, firstAccrualTime)
|
|
suite.storedTimeEquals(vaultDenom2, firstAccrualTime)
|
|
|
|
expected, f := previousIndexes.Get(vaultDenom1)
|
|
suite.True(f)
|
|
suite.storedIndexesEqual(vaultDenom1, expected)
|
|
|
|
expected, f = previousIndexes.Get(vaultDenom2)
|
|
suite.True(f)
|
|
suite.storedIndexesEqual(vaultDenom2, expected)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestStateAddedWhenStateDoesNotExist() {
|
|
vaultDenom := "usdx"
|
|
|
|
earnKeeper := newFakeEarnKeeper().addVault(vaultDenom, earntypes.NewVaultShare(vaultDenom, d("1000000")))
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, nil, earnKeeper)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
vaultDenom,
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)),
|
|
)
|
|
|
|
firstAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.ctx.WithBlockTime(firstAccrualTime)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// After the first accumulation only the current block time should be stored.
|
|
// The indexes will be empty as no time has passed since the previous block because it didn't exist.
|
|
suite.storedTimeEquals(vaultDenom, firstAccrualTime)
|
|
suite.storedIndexesEqual(vaultDenom, nil)
|
|
|
|
secondAccrualTime := firstAccrualTime.Add(10 * time.Second)
|
|
suite.ctx = suite.ctx.WithBlockTime(secondAccrualTime)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// After the second accumulation both current block time and indexes should be stored.
|
|
suite.storedTimeEquals(vaultDenom, secondAccrualTime)
|
|
suite.storedIndexesEqual(vaultDenom, types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.01"),
|
|
},
|
|
})
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestStateAddedWhenStateDoesNotExist_bkava() {
|
|
vaultDenom1 := "bkava-meow"
|
|
vaultDenom2 := "bkava-woof"
|
|
|
|
firstAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.ctx.WithBlockTime(firstAccrualTime)
|
|
|
|
earnKeeper := newFakeEarnKeeper().
|
|
addVault(vaultDenom1, earntypes.NewVaultShare(vaultDenom1, d("1000000"))).
|
|
addVault(vaultDenom2, earntypes.NewVaultShare(vaultDenom2, d("1000000")))
|
|
|
|
liquidKeeper := newFakeLiquidKeeper().
|
|
addDerivative(suite.ctx, vaultDenom1, i(1000000)).
|
|
addDerivative(suite.ctx, vaultDenom2, i(1000000))
|
|
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, liquidKeeper, earnKeeper)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
"bkava",
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)),
|
|
)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// After the first accumulation only the current block time should be stored.
|
|
// The indexes will be empty as no time has passed since the previous block because it didn't exist.
|
|
suite.storedTimeEquals(vaultDenom1, firstAccrualTime)
|
|
suite.storedTimeEquals(vaultDenom2, firstAccrualTime)
|
|
|
|
suite.storedIndexesEqual(vaultDenom1, nil)
|
|
suite.storedIndexesEqual(vaultDenom2, nil)
|
|
|
|
secondAccrualTime := firstAccrualTime.Add(10 * time.Second)
|
|
suite.ctx = suite.ctx.WithBlockTime(secondAccrualTime)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// After the second accumulation both current block time and indexes should be stored.
|
|
suite.storedTimeEquals(vaultDenom1, secondAccrualTime)
|
|
suite.storedTimeEquals(vaultDenom2, secondAccrualTime)
|
|
|
|
expectedIndexes := types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.01"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
// 10% of total bkava for rewards per second for 10 seconds
|
|
// 1ukava per share per second + regular 0.005ukava incentive rewards
|
|
RewardFactor: d("1.005"),
|
|
},
|
|
}
|
|
|
|
suite.storedIndexesEqual(vaultDenom1, expectedIndexes)
|
|
suite.storedIndexesEqual(vaultDenom2, expectedIndexes)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestNoPanicWhenStateDoesNotExist() {
|
|
vaultDenom := "usdx"
|
|
|
|
earnKeeper := newFakeEarnKeeper()
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, nil, earnKeeper)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
vaultDenom,
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(),
|
|
)
|
|
|
|
accrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.ctx.WithBlockTime(accrualTime)
|
|
|
|
// Accumulate with no earn shares and no rewards per second will result in no increment to the indexes.
|
|
// No increment and no previous indexes stored, results in an updated of nil. Setting this in the state panics.
|
|
// Check there is no panic.
|
|
suite.NotPanics(func() {
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
})
|
|
|
|
suite.storedTimeEquals(vaultDenom, accrualTime)
|
|
suite.storedIndexesEqual(vaultDenom, nil)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestNoPanicWhenStateDoesNotExist_bkava() {
|
|
vaultDenom1 := "bkava-meow"
|
|
vaultDenom2 := "bkava-woof"
|
|
|
|
earnKeeper := newFakeEarnKeeper()
|
|
liquidKeeper := newFakeLiquidKeeper()
|
|
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, liquidKeeper, earnKeeper)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
"bkava",
|
|
time.Unix(0, 0), // ensure the test is within start and end times
|
|
distantFuture,
|
|
cs(),
|
|
)
|
|
|
|
accrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.ctx = suite.ctx.WithBlockTime(accrualTime)
|
|
|
|
// Accumulate with no earn shares and no rewards per second will result in no increment to the indexes.
|
|
// No increment and no previous indexes stored, results in an updated of nil. Setting this in the state panics.
|
|
// Check there is no panic.
|
|
suite.NotPanics(func() {
|
|
// This does not update any state, as there are no bkava vaults
|
|
// to iterate over, denoms are unknown
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
})
|
|
|
|
// Times are not stored for vaults with no state
|
|
suite.storedTimeEquals(vaultDenom1, time.Time{})
|
|
suite.storedTimeEquals(vaultDenom2, time.Time{})
|
|
suite.storedIndexesEqual(vaultDenom1, nil)
|
|
suite.storedIndexesEqual(vaultDenom2, nil)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestNoAccumulationWhenBeforeStartTime() {
|
|
vaultDenom := "usdx"
|
|
|
|
earnKeeper := newFakeEarnKeeper().addVault(vaultDenom, earntypes.NewVaultShare(vaultDenom, d("1000000")))
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, nil, earnKeeper)
|
|
|
|
previousIndexes := types.MultiRewardIndexes{
|
|
{
|
|
CollateralType: vaultDenom,
|
|
RewardIndexes: types.RewardIndexes{
|
|
{
|
|
CollateralType: "earn",
|
|
RewardFactor: d("0.02"),
|
|
},
|
|
{
|
|
CollateralType: "ukava",
|
|
RewardFactor: d("0.04"),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
suite.storeGlobalEarnIndexes(previousIndexes)
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom, previousAccrualTime)
|
|
|
|
firstAccrualTime := previousAccrualTime.Add(10 * time.Second)
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
vaultDenom,
|
|
firstAccrualTime.Add(time.Nanosecond), // start time after accrual time
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)),
|
|
)
|
|
|
|
suite.ctx = suite.ctx.WithBlockTime(firstAccrualTime)
|
|
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
|
|
// The accrual time should be updated, but the indexes unchanged
|
|
suite.storedTimeEquals(vaultDenom, firstAccrualTime)
|
|
expectedIndexes, f := previousIndexes.Get(vaultDenom)
|
|
suite.True(f)
|
|
suite.storedIndexesEqual(vaultDenom, expectedIndexes)
|
|
}
|
|
|
|
func (suite *AccumulateEarnRewardsTests) TestPanicWhenCurrentTimeLessThanPrevious() {
|
|
vaultDenom := "usdx"
|
|
|
|
earnKeeper := newFakeEarnKeeper().addVault(vaultDenom, earntypes.NewVaultShare(vaultDenom, d("1000000")))
|
|
suite.keeper = suite.NewKeeper(&fakeParamSubspace{}, nil, nil, nil, nil, nil, nil, nil, nil, earnKeeper)
|
|
|
|
previousAccrualTime := time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
|
|
suite.keeper.SetEarnRewardAccrualTime(suite.ctx, vaultDenom, previousAccrualTime)
|
|
|
|
firstAccrualTime := time.Time{}
|
|
|
|
period := types.NewMultiRewardPeriod(
|
|
true,
|
|
vaultDenom,
|
|
time.Time{}, // start time after accrual time
|
|
distantFuture,
|
|
cs(c("earn", 2000), c("ukava", 1000)),
|
|
)
|
|
|
|
suite.ctx = suite.ctx.WithBlockTime(firstAccrualTime)
|
|
|
|
suite.Panics(func() {
|
|
suite.keeper.AccumulateEarnRewards(suite.ctx, period)
|
|
})
|
|
}
|