Add default handle(update:) implementation

trafi · Aug 22, 2019 · 9c3a69c · 9c3a69c
1 parent e603957
commit 9c3a69c
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/StoryFlow/Interfaces/UpdateHandling.swift b/StoryFlow/Interfaces/UpdateHandling.swift
@@ -4,3 +4,7 @@ public protocol UpdateHandling: _AnyUpdateHandling {
     associatedtype UpdateType
     func handle(update: UpdateType)
 }
+
+extension UpdateHandling where Self: UIViewController {
+    public func handle(update: UpdateType) {}
+}
diff --git a/StoryFlowTests/ExplicitFlowTests.swift b/StoryFlowTests/ExplicitFlowTests.swift
@@ -90,6 +90,27 @@ class ExplicitFlowTests: XCTestCase {
 
     // MARK: Unwind
 
+    func testProduce_itUnwindsToUpdateHandlingVc() {
+
+        // Arrange
+        class From: UIViewController, OutputProducing { typealias OutputType = Int }
+        class To: UIViewController, UpdateHandling { typealias UpdateType = Int }
+
+        let to = To().visible()
+        let from = From()
+        from.flow = .unwind()
+
+        to.show(from, sender: nil)
+        XCTAssert(currentVc.didAppear())
+
+        // Act
+        from.produce(2018)
+        XCTAssert(currentVc.didDismiss())
+
+        // Assert
+        XCTAssert(currentVc == to)
+    }
+
     func testProduce_itUnwindsToUpdateHandlingVcAndPassesOutput() {
 
         // Arrange