Allow access to activity mock

Author: cschleiden

tester/tester.go

@@ -85,6 +85,11 @@ type WorkflowTester[TResult any] interface {
 	// Registry returns the registry used by the tester.
 	Registry() *registry.Registry
 
+	// ActivityMock returns the mock instance that's used for activities. In most cases, OnActivity or OnActivityByName
+	// should be used, but sometimes it's necessary to access the mock directly. For instance, to ensure
+	// a certain activity is not called at all
+	ActivityMock() *mock.Mock
+
 	// OnActivity registers a mock activity.
 	OnActivity(activity workflow.Activity, args ...interface{}) *mock.Call
 
@@ -268,6 +273,11 @@ func (wt *workflowTester[TResult]) ListenSubWorkflow(listener func(*core.Workflo
 	wt.subWorkflowListener = listener
 }
 
+// ActivityMock returns the mock for activities.
+func (wt *workflowTester[TResult]) ActivityMock() *mock.Mock {
+	return wt.ma
+}
+
 // OnActivityByName registers a mock activity with the given name.
 func (wt *workflowTester[TResult]) OnActivityByName(name string, activity workflow.Activity, args ...any) *mock.Call {
 	// Register activity so that we can correctly identify its arguments later