@@ -21,6 +21,7 @@ import XCTest
21
21
22
22
extension RenderTester {
23
23
final class TestContext : RenderContextType {
24
+ let workflow : WorkflowType
24
25
var state : WorkflowType . State
25
26
var expectedWorkflows : [ AnyExpectedWorkflow ]
26
27
var expectedSideEffects : [ AnyHashable : ExpectedSideEffect ]
@@ -32,12 +33,14 @@ extension RenderTester {
32
33
private var usedWorkflowKeys : Set < WorkflowKey > = [ ]
33
34
34
35
init (
36
+ workflow: WorkflowType ,
35
37
state: WorkflowType . State ,
36
38
expectedWorkflows: [ AnyExpectedWorkflow ] ,
37
39
expectedSideEffects: [ AnyHashable : ExpectedSideEffect ] ,
38
40
file: StaticString ,
39
41
line: UInt
40
42
) {
43
+ self . workflow = workflow
41
44
self . state = state
42
45
self . expectedWorkflows = expectedWorkflows
43
46
self . expectedSideEffects = expectedSideEffects
@@ -117,7 +120,7 @@ extension RenderTester {
117
120
private func apply< ActionType> ( action: ActionType ) where ActionType: WorkflowAction , ActionType. WorkflowType == WorkflowType {
118
121
XCTAssertNil ( appliedAction, " Received multiple actions in a single render test " , file: file, line: line)
119
122
appliedAction = AppliedAction ( action)
120
- let output = action. apply ( toState: & state)
123
+ let output = action. apply ( toState: & state, workflow : workflow )
121
124
122
125
if let output {
123
126
XCTAssertNil ( producedOutput, " Received multiple outputs in a single render test " , file: file, line: line)
0 commit comments