Return struct instead of interface

Allow accessing internal state in tests

Author: cschleiden

tester/tester.go

@@ -38,36 +38,6 @@ func (t *testHistoryProvider) GetWorkflowInstanceHistory(ctx context.Context, in
 	return t.history, nil
 }
 
-type WorkflowTester[TResult any] interface {
-	// Now returns the current time of the simulated clock in the tester
-	Now() time.Time
-
-	Execute(args ...interface{})
-
-	Registry() *workflow.Registry
-
-	OnActivity(activity interface{}, args ...interface{}) *mock.Call
-
-	OnSubWorkflow(workflow interface{}, args ...interface{}) *mock.Call
-
-	SignalWorkflow(signalName string, value interface{})
-
-	SignalWorkflowInstance(wfi *core.WorkflowInstance, signalName string, value interface{}) error
-
-	WorkflowFinished() bool
-
-	WorkflowResult() (TResult, string)
-
-	// AssertExpectations asserts any assertions set up for mock activities and sub-workflow
-	AssertExpectations(t *testing.T)
-
-	// ScheduleCallback schedules the given callback after the given delay in workflow time (not wall clock).
-	ScheduleCallback(delay time.Duration, callback func())
-
-	// ListenSubWorkflow registers a handler to be called when a sub-workflow is started.
-	ListenSubWorkflow(listener func(instance *core.WorkflowInstance, name string))
-}
-
 type testTimer struct {
 	// Instance is the workflow instance this timer is for
 	Instance *core.WorkflowInstance
@@ -161,7 +131,7 @@ func WithTestTimeout(timeout time.Duration) WorkflowTesterOption {
 	}
 }
 
-func NewWorkflowTester[TResult any](wf interface{}, opts ...WorkflowTesterOption) WorkflowTester[TResult] {
+func NewWorkflowTester[TResult any](wf interface{}, opts ...WorkflowTesterOption) *workflowTester[TResult] {
 	if err := margs.ReturnTypeMatch[TResult](wf); err != nil {
 		panic(fmt.Sprintf("workflow return type does not match: %s", err))
 	}