2022-07-20 23:14:43 +00:00
|
|
|
package keeper_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
|
|
|
|
|
|
"github.com/kava-labs/kava/x/earn/keeper"
|
|
|
|
"github.com/kava-labs/kava/x/earn/testutil"
|
|
|
|
"github.com/kava-labs/kava/x/earn/types"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
)
|
|
|
|
|
|
|
|
type grpcQueryTestSuite struct {
|
|
|
|
testutil.Suite
|
|
|
|
|
|
|
|
queryClient types.QueryClient
|
|
|
|
}
|
|
|
|
|
|
|
|
func (suite *grpcQueryTestSuite) SetupTest() {
|
|
|
|
suite.Suite.SetupTest()
|
|
|
|
suite.Keeper.SetParams(suite.Ctx, types.DefaultParams())
|
|
|
|
|
|
|
|
queryHelper := baseapp.NewQueryServerTestHelper(suite.Ctx, suite.App.InterfaceRegistry())
|
|
|
|
types.RegisterQueryServer(queryHelper, keeper.NewQueryServerImpl(suite.Keeper))
|
|
|
|
|
|
|
|
suite.queryClient = types.NewQueryClient(queryHelper)
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestGrpcQueryTestSuite(t *testing.T) {
|
|
|
|
suite.Run(t, new(grpcQueryTestSuite))
|
|
|
|
}
|
|
|
|
|
|
|
|
func (suite *grpcQueryTestSuite) TestQueryParams() {
|
2022-07-28 16:39:57 +00:00
|
|
|
vaultDenom := "usdx"
|
2022-07-20 23:14:43 +00:00
|
|
|
|
|
|
|
res, err := suite.queryClient.Params(context.Background(), types.NewQueryParamsRequest())
|
|
|
|
suite.Require().NoError(err)
|
|
|
|
// ElementsMatch instead of Equal because AllowedVaults{} != AllowedVaults(nil)
|
|
|
|
suite.Require().ElementsMatch(types.DefaultParams().AllowedVaults, res.Params.AllowedVaults)
|
|
|
|
|
|
|
|
// Add vault to params
|
2022-07-28 16:39:57 +00:00
|
|
|
suite.CreateVault(vaultDenom, types.STRATEGY_TYPE_HARD)
|
2022-07-20 23:14:43 +00:00
|
|
|
|
|
|
|
// Query again for added vault
|
|
|
|
res, err = suite.queryClient.Params(context.Background(), types.NewQueryParamsRequest())
|
|
|
|
suite.Require().NoError(err)
|
|
|
|
suite.Require().Equal(
|
|
|
|
types.AllowedVaults{
|
2022-07-28 16:39:57 +00:00
|
|
|
types.NewAllowedVault(vaultDenom, types.STRATEGY_TYPE_HARD),
|
2022-07-20 23:14:43 +00:00
|
|
|
},
|
|
|
|
res.Params.AllowedVaults,
|
|
|
|
)
|
|
|
|
}
|