WorkflowTesting changes to support GUWT

Author: dhavalshreyas

WorkflowTesting/Sources/RenderExpectations.swift

@@ -15,6 +15,7 @@
  */
 
 import Workflow
+import XCTest
 
 /// A set of expectations for use with the `WorkflowRenderTester`. All of the expectations must be fulfilled
 /// for a `render` test to pass.
@@ -105,8 +106,8 @@ public struct ExpectedWorker {
 }
 
 public struct ExpectedSideEffect<WorkflowType: Workflow> {
-    let key: AnyHashable
-    let action: ((RenderContext<WorkflowType>) -> Void)?
+    public let key: AnyHashable
+    public let action: ((RenderContext<WorkflowType>) -> Void)?
 }
 
 extension ExpectedSideEffect {
@@ -123,15 +124,44 @@ extension ExpectedSideEffect {
 }
 
 public struct ExpectedWorkflow {
-    let workflowType: Any.Type
-    let key: String
     let rendering: Any
     let output: Any?
+    let doesMatch: (_ childWorkflow: Any, _ childKey: String) -> Bool
+    let notFoundAssertion: (_ file: StaticString, _ line: UInt) -> Void
+
+    public init<WorkflowType: Workflow>(
+        type: WorkflowType.Type,
+        key: String = "",
+        rendering: WorkflowType.Rendering,
+        output: WorkflowType.Output? = nil
+    ) {
+        self.rendering = rendering
+        self.output = output
+
+        self.doesMatch = { child, childKey in
+            guard child is WorkflowType,
+                key == childKey
+            else {
+                return false
+            }
+
+            return true
+        }
 
-    public init<WorkflowType: Workflow>(type: WorkflowType.Type, key: String = "", rendering: WorkflowType.Rendering, output: WorkflowType.Output? = nil) {
-        self.workflowType = type
-        self.key = key
+        self.notFoundAssertion = { file, line in
+            XCTFail("Expected child workflow of type: \(type) key: \(key)", file: file, line: line)
+        }
+    }
+
+    public init(
+        rendering: Any,
+        output: Any?,
+        doesMatch: @escaping (_ childWorkflow: Any, _ childKey: String) -> Bool,
+        notFoundAssertion: @escaping (_ file: StaticString, _ line: UInt) -> Void
+    ) {
         self.rendering = rendering
         self.output = output
+        self.doesMatch = doesMatch
+        self.notFoundAssertion = notFoundAssertion
     }
 }

WorkflowTesting/Sources/WorkflowRenderTester.swift

@@ -249,7 +249,7 @@
 
         func render<Child, Action>(workflow: Child, key: String, outputMap: @escaping (Child.Output) -> Action) -> Child.Rendering where Child: Workflow, Action: WorkflowAction, RenderTestContext<T>.WorkflowType == Action.WorkflowType {
             guard let workflowIndex = expectations.expectedWorkflows.firstIndex(where: { expectedWorkflow -> Bool in
-                type(of: workflow) == expectedWorkflow.workflowType && key == expectedWorkflow.key
+                expectedWorkflow.doesMatch(workflow, key)
             }) else {
                 XCTFail("Unexpected child workflow of type \(workflow.self)", file: file, line: line)
                 fatalError()
@@ -337,7 +337,7 @@
 
             if !expectations.expectedWorkflows.isEmpty {
                 for expectedWorkflow in expectations.expectedWorkflows {
-                    XCTFail("Expected child workflow of type: \(expectedWorkflow.workflowType) key: \(expectedWorkflow.key)", file: file, line: line)
+                    expectedWorkflow.notFoundAssertion(file, line)
                 }
             }