Adds tests for RoutableCollection utilities

johnpatrickmorgan · Sep 30, 2022 · 8d1bc57 · 8d1bc57
1 parent c4f332f
commit 8d1bc57
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 1 deletion.
diff --git a/Tests/FlowStacksTests/FlowStacksTests.swift → ...FlowStacksTests/CalculateStepsTests.swift b/Tests/FlowStacksTests/FlowStacksTests.swift → ...FlowStacksTests/CalculateStepsTests.swift
@@ -1,7 +1,7 @@
 @testable import FlowStacks
 import XCTest
 
-final class FlowStacksTests: XCTestCase {
+final class CalculateStepsTests: XCTestCase {
   typealias RouterState = [Route<Int>]
 
   func testPushOneAtATime() {

diff --git a/Tests/FlowStacksTests/RoutableCollectionTests.swift b/Tests/FlowStacksTests/RoutableCollectionTests.swift
@@ -0,0 +1,77 @@
+@testable import FlowStacks
+import XCTest
+
+final class RoutableCollectionTests: XCTestCase {
+  typealias RouterState = [Route<Int>]
+
+  func testPopToCurrentNavigationRootPresented() {
+    var routes: RouterState = [
+      .root(1, embedInNavigationView: true),
+      .sheet(-1, embedInNavigationView: true),
+      .push(-2),
+      .push(-3),
+      .push(-4)
+    ]
+
+    routes.popToCurrentNavigationRoot()
+
+    let expectedResult: RouterState = [
+      .root(1, embedInNavigationView: true),
+      .sheet(-1, embedInNavigationView: true)
+    ]
+    XCTAssertEqual(routes, expectedResult)
+  }
+
+  func testPopToCurrentNavigationRootNotPresented() {
+    var routes: RouterState = [
+      .root(1, embedInNavigationView: true),
+      .push(-2),
+      .push(-3),
+      .push(-4)
+    ]
+
+    routes.popToCurrentNavigationRoot()
+
+    let expectedResult: RouterState = [
+      .root(1, embedInNavigationView: true),
+    ]
+    XCTAssertEqual(routes, expectedResult)
+  }
+
+  func testDismissAll() {
+    var routes: RouterState = [
+      .root(1, embedInNavigationView: true),
+      .push(-2),
+      .sheet(3, embedInNavigationView: true),
+      .sheet(4, embedInNavigationView: true),
+      .sheet(5, embedInNavigationView: true),
+      .sheet(6, embedInNavigationView: true),
+      .push(-4)
+    ]
+
+    routes.dismissAll()
+
+    let expectedResult: RouterState = [
+      .root(1, embedInNavigationView: true),
+      .push(-2),
+    ]
+    XCTAssertEqual(routes, expectedResult)
+  }
+
+  func testDismissAllNoOpWithOnlyPushes() {
+    var routes: RouterState = [
+      .root(1, embedInNavigationView: true),
+      .push(2),
+      .push(3)
+    ]
+
+    routes.dismissAll()
+
+    let expectedResult: RouterState = [
+      .root(1, embedInNavigationView: true),
+      .push(2),
+      .push(3)
+    ]
+    XCTAssertEqual(routes, expectedResult)
+  }
+}