Add SetFailureConverter to the activity and workflow test environments (#1484)

Author: boxofrad

internal/internal_workflow_testsuite.go

@@ -451,6 +451,10 @@ func (env *testWorkflowEnvironmentImpl) setDataConverter(dataConverter converter
 	env.dataConverter = dataConverter
 }
 
+func (env *testWorkflowEnvironmentImpl) setFailureConverter(failureConverter converter.FailureConverter) {
+	env.failureConverter = failureConverter
+}
+
 func (env *testWorkflowEnvironmentImpl) setContextPropagators(contextPropagators []ContextPropagator) {
 	env.contextPropagators = contextPropagators
 }

internal/workflow_testsuite.go

@@ -214,6 +214,12 @@ func (t *TestActivityEnvironment) SetDataConverter(dataConverter converter.DataC
 	return t
 }
 
+// SetFailureConverter sets the failure converter.
+func (t *TestActivityEnvironment) SetFailureConverter(failureConverter converter.FailureConverter) *TestActivityEnvironment {
+	t.impl.setFailureConverter(failureConverter)
+	return t
+}
+
 // SetIdentity sets identity.
 func (t *TestActivityEnvironment) SetIdentity(identity string) *TestActivityEnvironment {
 	t.impl.setIdentity(identity)
@@ -649,6 +655,12 @@ func (e *TestWorkflowEnvironment) SetDataConverter(dataConverter converter.DataC
 	return e
 }
 
+// SetFailureConverter sets the failure converter.
+func (t *TestWorkflowEnvironment) SetFailureConverter(failureConverter converter.FailureConverter) *TestWorkflowEnvironment {
+	t.impl.setFailureConverter(failureConverter)
+	return t
+}
+
 // SetContextPropagators sets context propagators.
 func (e *TestWorkflowEnvironment) SetContextPropagators(contextPropagators []ContextPropagator) *TestWorkflowEnvironment {
 	e.impl.setContextPropagators(contextPropagators)