Fix poolmanager query (#4403)

* add NewQuerier func * poolmanager regis query server * lint * check query not alter state
osmosis-labs · Feb 23, 2023 · 46530dc · 46530dc
1 parent f3c2e1f
commit 46530dc
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/x/poolmanager/client/cli/query_test.go b/x/poolmanager/client/cli/query_test.go
@@ -36,6 +36,18 @@ func (s *QueryTestSuite) TestQueriesNeverAlterState() {
 			&poolmanagerqueryproto.NumPoolsRequest{},
 			&poolmanagerqueryproto.NumPoolsResponse{},
 		},
+		{
+			"Query estimate swap in",
+			"/osmosis.poolmanager.v1beta1.Query/EstimateSwapExactAmountIn",
+			&poolmanagerqueryproto.EstimateSwapExactAmountInRequest{},
+			&poolmanagerqueryproto.EstimateSwapExactAmountInResponse{},
+		},
+		{
+			"Query estimate swap out",
+			"/osmosis.poolmanager.v1beta1.Query/EstimateSwapExactAmountOut",
+			&poolmanagerqueryproto.EstimateSwapExactAmountOutRequest{},
+			&poolmanagerqueryproto.EstimateSwapExactAmountOutResponse{},
+		},
 	}
 
 	for _, tc := range testCases {

diff --git a/x/poolmanager/client/query_proto_wrap.go b/x/poolmanager/client/query_proto_wrap.go
@@ -16,6 +16,10 @@ type Querier struct {
 	K poolmanager.Keeper
 }
 
+func NewQuerier(k poolmanager.Keeper) Querier {
+	return Querier{k}
+}
+
 func (q Querier) Params(ctx sdk.Context,
 	req queryproto.ParamsRequest,
 ) (*queryproto.ParamsResponse, error) {

diff --git a/x/poolmanager/module/module.go b/x/poolmanager/module/module.go
@@ -18,7 +18,9 @@ import (
 	"github.com/osmosis-labs/osmosis/v14/simulation/simtypes"
 	gammsimulation "github.com/osmosis-labs/osmosis/v14/x/gamm/simulation"
 	"github.com/osmosis-labs/osmosis/v14/x/poolmanager"
+	pmclient "github.com/osmosis-labs/osmosis/v14/x/poolmanager/client"
 	"github.com/osmosis-labs/osmosis/v14/x/poolmanager/client/cli"
+	"github.com/osmosis-labs/osmosis/v14/x/poolmanager/client/grpc"
 	"github.com/osmosis-labs/osmosis/v14/x/poolmanager/client/queryproto"
 	"github.com/osmosis-labs/osmosis/v14/x/poolmanager/types"
 )
@@ -82,6 +84,7 @@ type AppModule struct {
 
 func (am AppModule) RegisterServices(cfg module.Configurator) {
 	types.RegisterMsgServer(cfg.MsgServer(), poolmanager.NewMsgServerImpl(&am.k))
+	queryproto.RegisterQueryServer(cfg.QueryServer(), grpc.Querier{Q: pmclient.NewQuerier(am.k)})
 }
 
 func NewAppModule(poolmanagerKeeper poolmanager.Keeper, gammKeeper types.SwapI) AppModule {