@@ -24,6 +24,7 @@ import (
24
24
"github.com/cschleiden/go-workflows/log"
25
25
"github.com/google/uuid"
26
26
"github.com/stretchr/testify/mock"
27
+ "go.opentelemetry.io/otel/trace"
27
28
)
28
29
29
30
type testHistoryProvider struct {
@@ -118,6 +119,8 @@ type workflowTester[TResult any] struct {
118
119
runningActivities int32
119
120
120
121
logger log.Logger
122
+
123
+ tracer trace.Tracer
121
124
}
122
125
123
126
type WorkflowTesterOption func (* options )
@@ -154,6 +157,8 @@ func NewWorkflowTester[TResult any](wf interface{}, opts ...WorkflowTesterOption
154
157
options .Logger = logger .NewDefaultLogger ()
155
158
}
156
159
160
+ tracer := trace .NewNoopTracerProvider ().Tracer ("workflow-tester" )
161
+
157
162
wt := & workflowTester [TResult ]{
158
163
options : options ,
159
164
@@ -176,6 +181,7 @@ func NewWorkflowTester[TResult any](wf interface{}, opts ...WorkflowTesterOption
176
181
callbacks : make (chan func () * history.WorkflowEvent , 1024 ),
177
182
178
183
logger : options .Logger ,
184
+ tracer : tracer ,
179
185
}
180
186
181
187
// Always register the workflow under test
@@ -246,7 +252,7 @@ func (wt *workflowTester[TResult]) Execute(args ...interface{}) {
246
252
tw .pendingEvents = tw .pendingEvents [:0 ]
247
253
248
254
// Execute task
249
- e , err := workflow .NewExecutor (wt .logger , wt .registry , & testHistoryProvider {tw .history }, tw .instance , wt .clock )
255
+ e , err := workflow .NewExecutor (wt .logger , wt .tracer , wt . registry , & testHistoryProvider {tw .history }, tw .instance , wt .clock )
250
256
if err != nil {
251
257
panic ("could not create workflow executor" + err .Error ())
252
258
}
@@ -474,7 +480,7 @@ func (wt *workflowTester[TResult]) scheduleActivity(wfi *core.WorkflowInstance,
474
480
}
475
481
476
482
} else {
477
- executor := activity .NewExecutor (wt .logger , wt .registry )
483
+ executor := activity .NewExecutor (wt .logger , wt .tracer , wt . registry )
478
484
activityResult , activityErr = executor .ExecuteActivity (context .Background (), & task.Activity {
479
485
ID : uuid .NewString (),
480
486
WorkflowInstance : wfi ,
0 commit comments