0g-chain/x/incentive/keeper/rewards_earn_proportional_test.go
drklee3 3375484f79
Update deprecated Cosmos methods (#1530)
* Use cosmossdk.io/errors for deprecated error methods

* Update error registration with cosmossdk.io/errors

* Use cosmossdk.io/math for deprecated sdk.Int alias

* Fix modified proto file

* Update sdk.Int usage in swap hooks

* Update e2e test deprecated method usage
2023-04-05 16:21:59 -07:00

88 lines
1.8 KiB
Go

package keeper_test
import (
"testing"
"time"
sdkmath "cosmossdk.io/math"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/kava-labs/kava/x/incentive/keeper"
"github.com/kava-labs/kava/x/incentive/types"
"github.com/stretchr/testify/require"
)
func TestGetProportionalRewardPeriod(t *testing.T) {
tests := []struct {
name string
giveRewardPeriod types.MultiRewardPeriod
giveTotalBkavaSupply sdkmath.Int
giveSingleBkavaSupply sdkmath.Int
wantRewardsPerSecond sdk.DecCoins
}{
{
"full amount",
types.NewMultiRewardPeriod(
true,
"",
time.Time{},
time.Time{},
cs(c("ukava", 100), c("hard", 200)),
),
i(100),
i(100),
toDcs(c("ukava", 100), c("hard", 200)),
},
{
"3/4 amount",
types.NewMultiRewardPeriod(
true,
"",
time.Time{},
time.Time{},
cs(c("ukava", 100), c("hard", 200)),
),
i(10_000000),
i(7_500000),
toDcs(c("ukava", 75), c("hard", 150)),
},
{
"half amount",
types.NewMultiRewardPeriod(
true,
"",
time.Time{},
time.Time{},
cs(c("ukava", 100), c("hard", 200)),
),
i(100),
i(50),
toDcs(c("ukava", 50), c("hard", 100)),
},
{
"under 1 unit",
types.NewMultiRewardPeriod(
true,
"",
time.Time{},
time.Time{},
cs(c("ukava", 100), c("hard", 200)),
),
i(1000), // total bkava
i(1), // bkava supply of this specific vault
dcs(dc("ukava", "0.1"), dc("hard", "0.2")), // rewards per second rounded to 0 if under 1ukava/1hard
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
rewardsPerSecond := keeper.GetProportionalRewardsPerSecond(
tt.giveRewardPeriod,
tt.giveTotalBkavaSupply,
tt.giveSingleBkavaSupply,
)
require.Equal(t, tt.wantRewardsPerSecond, rewardsPerSecond)
})
}
}