From 2ec2a5033e0e5de2115852b38c9e154053d6635a Mon Sep 17 00:00:00 2001 From: rick <1450685+LinuxSuRen@users.noreply.github.com> Date: Wed, 18 Oct 2023 10:08:10 +0800 Subject: [PATCH] fix: render the testSuite before generate code --- pkg/server/remote_server.go | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/pkg/server/remote_server.go b/pkg/server/remote_server.go index 4766645b..b7ba972c 100644 --- a/pkg/server/remote_server.go +++ b/pkg/server/remote_server.go @@ -632,8 +632,22 @@ func (s *server) GenerateCode(ctx context.Context, in *CodeGenerateRequest) (rep reply.Message = fmt.Sprintf("generator '%s' not found", in.Generator) } else { var result testing.TestCase + var suite testing.TestSuite + loader := s.getLoader(ctx) + + if suite, err = loader.GetTestSuite(in.TestSuite, true); err != nil { + return + } + + dataContext := map[string]interface{}{} + if err = suite.Render(dataContext); err != nil { + return + } + if result, err = loader.GetTestCase(in.TestSuite, in.TestCase); err == nil { + result.Request.RenderAPI(suite.API) + output, genErr := instance.Generate(&result) reply.Success = genErr == nil reply.Message = util.OrErrorMessage(genErr, output)