0g-chain/x/incentive/keeper/grpc_query_test.go
drklee3 35c848dce6
Add incentive grpc query service (#1503)
* Add incentive grpc query service

* Register query server

* Add grpc query test

* Add grpc query test

* Fix proto lints

* Fix missing savings field, update tests

* Update proto docs

* Use consts for query reward types

* Add owner to queryRewards err

* Add empty reward type to be valid
2023-03-21 15:47:21 -07:00

319 lines
12 KiB
Go

package keeper_test
import (
"testing"
"time"
"github.com/cosmos/cosmos-sdk/baseapp"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/kava-labs/kava/app"
hardtypes "github.com/kava-labs/kava/x/hard/types"
"github.com/kava-labs/kava/x/incentive/keeper"
"github.com/kava-labs/kava/x/incentive/types"
"github.com/stretchr/testify/suite"
tmprototypes "github.com/tendermint/tendermint/proto/tendermint/types"
)
const (
oneYear time.Duration = 365 * 24 * time.Hour
)
type grpcQueryTestSuite struct {
suite.Suite
tApp app.TestApp
ctx sdk.Context
keeper keeper.Keeper
queryClient types.QueryClient
addrs []sdk.AccAddress
genesisTime time.Time
genesisState types.GenesisState
}
func (suite *grpcQueryTestSuite) SetupTest() {
suite.tApp = app.NewTestApp()
cdc := suite.tApp.AppCodec()
_, addrs := app.GeneratePrivKeyAddressPairs(5)
suite.genesisTime = time.Date(1998, 1, 1, 0, 0, 0, 0, time.UTC)
suite.addrs = addrs
suite.ctx = suite.tApp.NewContext(true, tmprototypes.Header{}).
WithBlockTime(time.Now().UTC())
suite.keeper = suite.tApp.GetIncentiveKeeper()
queryHelper := baseapp.NewQueryServerTestHelper(suite.ctx, suite.tApp.InterfaceRegistry())
types.RegisterQueryServer(queryHelper, keeper.NewQueryServerImpl(suite.keeper))
suite.queryClient = types.NewQueryClient(queryHelper)
loanToValue, _ := sdk.NewDecFromStr("0.6")
borrowLimit := sdk.NewDec(1000000000000000)
hardGS := hardtypes.NewGenesisState(
hardtypes.NewParams(
hardtypes.MoneyMarkets{
hardtypes.NewMoneyMarket("ukava", hardtypes.NewBorrowLimit(false, borrowLimit, loanToValue), "kava:usd", sdk.NewInt(1000000), hardtypes.NewInterestRateModel(sdk.MustNewDecFromStr("0.05"), sdk.MustNewDecFromStr("2"), sdk.MustNewDecFromStr("0.8"), sdk.MustNewDecFromStr("10")), sdk.MustNewDecFromStr("0.05"), sdk.ZeroDec()),
hardtypes.NewMoneyMarket("bnb", hardtypes.NewBorrowLimit(false, borrowLimit, loanToValue), "bnb:usd", sdk.NewInt(1000000), hardtypes.NewInterestRateModel(sdk.MustNewDecFromStr("0.05"), sdk.MustNewDecFromStr("2"), sdk.MustNewDecFromStr("0.8"), sdk.MustNewDecFromStr("10")), sdk.MustNewDecFromStr("0.05"), sdk.ZeroDec()),
},
sdk.NewDec(10),
),
hardtypes.DefaultAccumulationTimes,
hardtypes.DefaultDeposits,
hardtypes.DefaultBorrows,
hardtypes.DefaultTotalSupplied,
hardtypes.DefaultTotalBorrowed,
hardtypes.DefaultTotalReserves,
)
suite.genesisState = types.NewGenesisState(
types.NewParams(
types.RewardPeriods{types.NewRewardPeriod(true, "bnb-a", suite.genesisTime.Add(-1*oneYear), suite.genesisTime.Add(oneYear), c("ukava", 122354))},
types.MultiRewardPeriods{types.NewMultiRewardPeriod(true, "bnb", suite.genesisTime.Add(-1*oneYear), suite.genesisTime.Add(oneYear), cs(c("hard", 122354)))},
types.MultiRewardPeriods{types.NewMultiRewardPeriod(true, "bnb", suite.genesisTime.Add(-1*oneYear), suite.genesisTime.Add(oneYear), cs(c("hard", 122354)))},
types.MultiRewardPeriods{types.NewMultiRewardPeriod(true, "ukava", suite.genesisTime.Add(-1*oneYear), suite.genesisTime.Add(oneYear), cs(c("hard", 122354)))},
types.MultiRewardPeriods{types.NewMultiRewardPeriod(true, "btcb/usdx", suite.genesisTime.Add(-1*oneYear), suite.genesisTime.Add(oneYear), cs(c("swp", 122354)))},
types.MultiRewardPeriods{types.NewMultiRewardPeriod(true, "ukava", suite.genesisTime.Add(-1*oneYear), suite.genesisTime.Add(oneYear), cs(c("hard", 122354)))},
types.MultiRewardPeriods{types.NewMultiRewardPeriod(true, "ukava", suite.genesisTime.Add(-1*oneYear), suite.genesisTime.Add(oneYear), cs(c("hard", 122354)))},
types.MultipliersPerDenoms{
{
Denom: "ukava",
Multipliers: types.Multipliers{
types.NewMultiplier("large", 12, d("1.0")),
},
},
{
Denom: "hard",
Multipliers: types.Multipliers{
types.NewMultiplier("small", 1, d("0.25")),
types.NewMultiplier("large", 12, d("1.0")),
},
},
{
Denom: "swp",
Multipliers: types.Multipliers{
types.NewMultiplier("small", 1, d("0.25")),
types.NewMultiplier("medium", 6, d("0.8")),
},
},
},
suite.genesisTime.Add(5*oneYear),
),
types.NewGenesisRewardState(
types.AccumulationTimes{
types.NewAccumulationTime("bnb-a", suite.genesisTime),
},
types.MultiRewardIndexes{
types.NewMultiRewardIndex("bnb-a", types.RewardIndexes{{CollateralType: "ukava", RewardFactor: d("0.3")}}),
},
),
types.NewGenesisRewardState(
types.AccumulationTimes{
types.NewAccumulationTime("bnb", suite.genesisTime.Add(-1*time.Hour)),
},
types.MultiRewardIndexes{
types.NewMultiRewardIndex("bnb", types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.1")}}),
},
),
types.NewGenesisRewardState(
types.AccumulationTimes{
types.NewAccumulationTime("bnb", suite.genesisTime.Add(-2*time.Hour)),
},
types.MultiRewardIndexes{
types.NewMultiRewardIndex("bnb", types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.05")}}),
},
),
types.NewGenesisRewardState(
types.AccumulationTimes{
types.NewAccumulationTime("ukava", suite.genesisTime.Add(-3*time.Hour)),
},
types.MultiRewardIndexes{
types.NewMultiRewardIndex("ukava", types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.2")}}),
},
),
types.NewGenesisRewardState(
types.AccumulationTimes{
types.NewAccumulationTime("bctb/usdx", suite.genesisTime.Add(-4*time.Hour)),
},
types.MultiRewardIndexes{
types.NewMultiRewardIndex("btcb/usdx", types.RewardIndexes{{CollateralType: "swap", RewardFactor: d("0.001")}}),
},
),
types.NewGenesisRewardState(
types.AccumulationTimes{
types.NewAccumulationTime("ukava", suite.genesisTime.Add(-3*time.Hour)),
},
types.MultiRewardIndexes{
types.NewMultiRewardIndex("ukava", types.RewardIndexes{{CollateralType: "ukava", RewardFactor: d("0.2")}}),
},
),
types.NewGenesisRewardState(
types.AccumulationTimes{
types.NewAccumulationTime("usdx", suite.genesisTime.Add(-3*time.Hour)),
},
types.MultiRewardIndexes{
types.NewMultiRewardIndex("usdx", types.RewardIndexes{{CollateralType: "usdx", RewardFactor: d("0.2")}}),
},
),
types.USDXMintingClaims{
types.NewUSDXMintingClaim(
suite.addrs[0],
c("ukava", 1e9),
types.RewardIndexes{{CollateralType: "bnb-a", RewardFactor: d("0.3")}},
),
types.NewUSDXMintingClaim(
suite.addrs[1],
c("ukava", 1),
types.RewardIndexes{{CollateralType: "bnb-a", RewardFactor: d("0.001")}},
),
},
types.HardLiquidityProviderClaims{
types.NewHardLiquidityProviderClaim(
suite.addrs[0],
cs(c("ukava", 1e9), c("hard", 1e9)),
types.MultiRewardIndexes{{CollateralType: "bnb", RewardIndexes: types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.01")}}}},
types.MultiRewardIndexes{{CollateralType: "bnb", RewardIndexes: types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.0")}}}},
),
types.NewHardLiquidityProviderClaim(
suite.addrs[1],
cs(c("hard", 1)),
types.MultiRewardIndexes{{CollateralType: "bnb", RewardIndexes: types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.1")}}}},
types.MultiRewardIndexes{{CollateralType: "bnb", RewardIndexes: types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.0")}}}},
),
},
types.DelegatorClaims{
types.NewDelegatorClaim(
suite.addrs[2],
cs(c("hard", 5)),
types.MultiRewardIndexes{{CollateralType: "ukava", RewardIndexes: types.RewardIndexes{{CollateralType: "hard", RewardFactor: d("0.2")}}}},
),
},
types.SwapClaims{
types.NewSwapClaim(
suite.addrs[3],
nil,
types.MultiRewardIndexes{{CollateralType: "btcb/usdx", RewardIndexes: types.RewardIndexes{{CollateralType: "swap", RewardFactor: d("0.0")}}}},
),
},
types.SavingsClaims{
types.NewSavingsClaim(
suite.addrs[3],
nil,
types.MultiRewardIndexes{{CollateralType: "ukava", RewardIndexes: types.RewardIndexes{{CollateralType: "ukava", RewardFactor: d("0.0")}}}},
),
},
types.EarnClaims{
types.NewEarnClaim(
suite.addrs[3],
nil,
types.MultiRewardIndexes{{CollateralType: "usdx", RewardIndexes: types.RewardIndexes{{CollateralType: "usdx", RewardFactor: d("0.0")}}}},
),
},
)
err := suite.genesisState.Validate()
suite.Require().NoError(err)
suite.tApp = suite.tApp.InitializeFromGenesisStatesWithTime(
suite.genesisTime,
app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&suite.genesisState)},
app.GenesisState{hardtypes.ModuleName: cdc.MustMarshalJSON(&hardGS)},
NewCDPGenStateMulti(cdc),
NewPricefeedGenStateMultiFromTime(cdc, suite.genesisTime),
)
}
func (suite *grpcQueryTestSuite) TestGrpcQueryParams() {
res, err := suite.queryClient.Params(sdk.WrapSDKContext(suite.ctx), &types.QueryParamsRequest{})
suite.Require().NoError(err)
expected := suite.keeper.GetParams(suite.ctx)
suite.Equal(expected, res.Params, "params should equal default params")
}
func (suite *grpcQueryTestSuite) TestGrpcQueryRewards() {
res, err := suite.queryClient.Rewards(sdk.WrapSDKContext(suite.ctx), &types.QueryRewardsRequest{
Unsynchronized: true,
})
suite.Require().NoError(err)
suite.Equal(suite.genesisState.USDXMintingClaims, res.USDXMintingClaims)
suite.Equal(suite.genesisState.HardLiquidityProviderClaims, res.HardLiquidityProviderClaims)
suite.Equal(suite.genesisState.DelegatorClaims, res.DelegatorClaims)
suite.Equal(suite.genesisState.SwapClaims, res.SwapClaims)
suite.Equal(suite.genesisState.SavingsClaims, res.SavingsClaims)
suite.Equal(suite.genesisState.EarnClaims, res.EarnClaims)
}
func (suite *grpcQueryTestSuite) TestGrpcQueryRewards_Owner() {
res, err := suite.queryClient.Rewards(sdk.WrapSDKContext(suite.ctx), &types.QueryRewardsRequest{
Owner: suite.addrs[0].String(),
})
suite.Require().NoError(err)
suite.Len(res.USDXMintingClaims, 1)
suite.Len(res.HardLiquidityProviderClaims, 1)
suite.Equal(suite.genesisState.USDXMintingClaims[0], res.USDXMintingClaims[0])
suite.Equal(suite.genesisState.HardLiquidityProviderClaims[0], res.HardLiquidityProviderClaims[0])
// No other claims - owner has none
suite.Empty(res.DelegatorClaims)
suite.Empty(res.SwapClaims)
suite.Empty(res.SavingsClaims)
suite.Empty(res.EarnClaims)
}
func (suite *grpcQueryTestSuite) TestGrpcQueryRewards_RewardType() {
res, err := suite.queryClient.Rewards(sdk.WrapSDKContext(suite.ctx), &types.QueryRewardsRequest{
RewardType: keeper.RewardTypeHard,
Unsynchronized: true,
})
suite.Require().NoError(err)
suite.Equal(suite.genesisState.HardLiquidityProviderClaims, res.HardLiquidityProviderClaims)
// No other reward types when specifying rewardType
suite.Empty(res.USDXMintingClaims)
suite.Empty(res.DelegatorClaims)
suite.Empty(res.SwapClaims)
suite.Empty(res.SavingsClaims)
suite.Empty(res.EarnClaims)
}
func (suite *grpcQueryTestSuite) TestGrpcQueryRewards_RewardType_and_Owner() {
res, err := suite.queryClient.Rewards(sdk.WrapSDKContext(suite.ctx), &types.QueryRewardsRequest{
Owner: suite.addrs[0].String(),
RewardType: keeper.RewardTypeHard,
})
suite.Require().NoError(err)
suite.Len(res.HardLiquidityProviderClaims, 1)
suite.Equal(suite.genesisState.HardLiquidityProviderClaims[0], res.HardLiquidityProviderClaims[0])
suite.Empty(res.USDXMintingClaims)
suite.Empty(res.DelegatorClaims)
suite.Empty(res.SwapClaims)
suite.Empty(res.SavingsClaims)
suite.Empty(res.EarnClaims)
}
func (suite *grpcQueryTestSuite) TestGrpcQueryRewardFactors() {
res, err := suite.queryClient.RewardFactors(sdk.WrapSDKContext(suite.ctx), &types.QueryRewardFactorsRequest{})
suite.Require().NoError(err)
suite.NotEmpty(res.UsdxMintingRewardFactors)
suite.NotEmpty(res.HardSupplyRewardFactors)
suite.NotEmpty(res.HardBorrowRewardFactors)
suite.NotEmpty(res.DelegatorRewardFactors)
suite.NotEmpty(res.SwapRewardFactors)
suite.NotEmpty(res.SavingsRewardFactors)
suite.NotEmpty(res.EarnRewardFactors)
}
func TestGrpcQueryTestSuite(t *testing.T) {
suite.Run(t, new(grpcQueryTestSuite))
}