@@ -21,6 +21,7 @@ import XCTest
2121
2222extension RenderTester {
2323 final class TestContext : RenderContextType {
24+ let workflow : WorkflowType
2425 var state : WorkflowType . State
2526 var expectedWorkflows : [ AnyExpectedWorkflow ]
2627 var expectedSideEffects : [ AnyHashable : ExpectedSideEffect ]
@@ -32,12 +33,14 @@ extension RenderTester {
3233 private var usedWorkflowKeys : Set < WorkflowKey > = [ ]
3334
3435 init (
36+ workflow: WorkflowType ,
3537 state: WorkflowType . State ,
3638 expectedWorkflows: [ AnyExpectedWorkflow ] ,
3739 expectedSideEffects: [ AnyHashable : ExpectedSideEffect ] ,
3840 file: StaticString ,
3941 line: UInt
4042 ) {
43+ self . workflow = workflow
4144 self . state = state
4245 self . expectedWorkflows = expectedWorkflows
4346 self . expectedSideEffects = expectedSideEffects
@@ -117,7 +120,7 @@ extension RenderTester {
117120 private func apply< ActionType> ( action: ActionType ) where ActionType: WorkflowAction , ActionType. WorkflowType == WorkflowType {
118121 XCTAssertNil ( appliedAction, " Received multiple actions in a single render test " , file: file, line: line)
119122 appliedAction = AppliedAction ( action)
120- let output = action. apply ( toState: & state)
123+ let output = action. apply ( toState: & state, workflow : workflow )
121124
122125 if let output {
123126 XCTAssertNil ( producedOutput, " Received multiple outputs in a single render test " , file: file, line: line)
0 commit comments