Skip to content

Commit 72f82ce

Browse files
committed
fix render tester
1 parent 5099500 commit 72f82ce

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

WorkflowTesting/Sources/Internal/RenderTester+TestContext.swift

+4-1
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@ import XCTest
2121

2222
extension 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)

WorkflowTesting/Sources/WorkflowRenderTester.swift

+1
Original file line numberDiff line numberDiff line change
@@ -261,6 +261,7 @@ public struct RenderTester<WorkflowType: Workflow> {
261261
assertions: (WorkflowType.Rendering) throws -> Void
262262
) rethrows -> RenderTesterResult<WorkflowType> {
263263
let contextImplementation = TestContext(
264+
workflow: workflow,
264265
state: state,
265266
expectedWorkflows: expectedWorkflows,
266267
expectedSideEffects: expectedSideEffects,

0 commit comments

Comments
 (0)