fix: issue when generating QueryClient with titled QueryMsg variants (#…

…94) * fix: titled query varient client issue * test: fix test
srdtrk · Jun 22, 2024 · cac4d56 · cac4d56
1 parent f4a783d
commit cac4d56
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/integration_test/integration_test.go b/integration_test/integration_test.go
@@ -104,7 +104,7 @@ func (s *MySuite) TestQueryClient() {
 	s.GenerateQueryClientTest("testdata/cw721-base.json")
 	s.GenerateQueryClientTest("testdata/cw2981-royalties.json")
 	s.GenerateQueryClientTest("testdata/ics721.json")
-	s.GenerateMessageTypesTest("testdata/dao-dao-core.json")
+	s.GenerateQueryClientTest("testdata/dao-dao-core.json")
 	s.GenerateQueryClientTest("testdata/axone-objectarium.json")
 }
 

diff --git a/pkg/codegen/querier.go b/pkg/codegen/querier.go
@@ -155,8 +155,8 @@ func generateQueryFunc(queryTitle, queryName string, respSchema *schemas.JSONSch
 		jen.Id("req").Op("*").Id(reqType),
 		jen.Id("opts").Op("...").Qual("google.golang.org/grpc", "CallOption"),
 	).Params(jen.Op("*").Id(respType), jen.Error()).Block(
-		jen.List(jen.Id("rawQueryData"), jen.Err()).Op(":=").Qual("encoding/json", "Marshal").Call(jen.Op("&").Id(queryTitle).Values(jen.Dict{
-			jen.Id(pascalName): jen.Id("req"),
+		jen.List(jen.Id("rawQueryData"), jen.Err()).Op(":=").Qual("encoding/json", "Marshal").Call(jen.Map(jen.String()).Any().Values(jen.Dict{
+			jen.Lit(queryName): jen.Id("req"),
 		})),
 		jen.If(jen.Err().Op("!=").Nil()).Block(
 			jen.Return(jen.Nil(), jen.Err()),