From ad2d6f0678b4a208a327f524315b4443f5008b00 Mon Sep 17 00:00:00 2001 From: yedf2 <120050102@qq.com> Date: Fri, 25 Nov 2022 19:25:58 +0800 Subject: [PATCH] add Context update --- test/workflow_grpc_test.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/workflow_grpc_test.go b/test/workflow_grpc_test.go index 726f83b..1385cd2 100644 --- a/test/workflow_grpc_test.go +++ b/test/workflow_grpc_test.go @@ -16,6 +16,7 @@ import ( "github.com/dtm-labs/dtm/client/workflow" "github.com/dtm-labs/dtm/test/busi" "github.com/stretchr/testify/assert" + "google.golang.org/grpc/metadata" ) func TestWorkflowGrpcSimple(t *testing.T) { @@ -84,6 +85,7 @@ func TestWorkflowMixed(t *testing.T) { if err != nil { return err } + wf.Context = metadata.NewOutgoingContext(wf.Context, metadata.Pairs("k1", "v1")) req2 := &busi.ReqHTTP{Amount: int(req.Amount / 2)} _, err = wf.NewBranch().OnCommit(func(bb *dtmcli.BranchBarrier) error {