Fix refresh behavior

Source/Turbo Navigator/TurboNavigationHierarchyController.swift

@@ -168,21 +168,28 @@ class TurboNavigationHierarchyController {
         if navigationController.presentedViewController != nil {
             if modalNavigationController.viewControllers.count == 1 {
                 navigationController.dismiss(animated: proposal.animated)
-                delegate.refresh(navigationStack: .main)
+                refreshIfTopViewControllerIsVisitable(from: .main)
             } else {
                 modalNavigationController.popViewController(animated: proposal.animated)
-                delegate.refresh(navigationStack: .modal)
+                refreshIfTopViewControllerIsVisitable(from: .modal)
             }
         } else {
             navigationController.popViewController(animated: proposal.animated)
-            delegate.refresh(navigationStack: .main)
+            refreshIfTopViewControllerIsVisitable(from: .main)
+        }
+    }
+
+    private func refreshIfTopViewControllerIsVisitable(from stack: NavigationStackType) {
+        if let navControllerTopmostVisitable = navController(for: stack).topViewController as? Visitable {
+            delegate.refreshVisitable(navigationStack: stack,
+                                      newTopmostVisitable: navControllerTopmostVisitable)
         }
     }
 
     private func clearAll(via proposal: VisitProposal) {
         navigationController.dismiss(animated: proposal.animated)
         navigationController.popToRootViewController(animated: proposal.animated)
-        delegate.refresh(navigationStack: .main)
+        refreshIfTopViewControllerIsVisitable(from: .main)
     }
 
     private func replaceRoot(with controller: UIViewController, via proposal: VisitProposal) {

Source/Turbo Navigator/TurboNavigationHierarchyControllerDelegate.swift

@@ -1,9 +1,23 @@
 import SafariServices
 import WebKit
 
-/// Implement to be notified when certain navigations are performed
-/// or to render a native controller instead of a Turbo web visit.
 protocol TurboNavigationHierarchyControllerDelegate: AnyObject {
-    func visit(_ : Visitable, on: TurboNavigationHierarchyController.NavigationStackType, with: VisitOptions)
-    func refresh(navigationStack: TurboNavigationHierarchyController.NavigationStackType)
+
+    /// Once the navigation hierarchy is modified, begin a visit on a navigation controller.
+    ///
+    /// - Parameters:
+    ///   - _: the Visitable destination
+    ///   - on: the navigation controller that was modified
+    ///   - with: the visit options
+    func visit(_ : Visitable,
+               on: TurboNavigationHierarchyController.NavigationStackType,
+               with: VisitOptions)
+
+    /// A refresh will pop (or dismiss) then ask the session to refresh the previous (or underlying) Visitable.
+    ///
+    /// - Parameters:
+    ///   - navigationStack: the stack where the refresh is happening
+    ///   - newTopmostVisitable: the visitable to be refreshed
+    func refreshVisitable(navigationStack: TurboNavigationHierarchyController.NavigationStackType,
+                          newTopmostVisitable: Visitable)
 }

Source/Turbo Navigator/TurboNavigator.swift

@@ -215,10 +215,13 @@ extension TurboNavigator: TurboNavigationHierarchyControllerDelegate {
         }
     }
 
-    func refresh(navigationStack: TurboNavigationHierarchyController.NavigationStackType) {
+    func refreshVisitable(navigationStack: TurboNavigationHierarchyController.NavigationStackType,
+                          newTopmostVisitable: any Visitable) {
         switch navigationStack {
-        case .main: session.reload()
-        case .modal: modalSession.reload()
+        case .main:
+            session.visit(newTopmostVisitable, action: .restore)
+        case .modal:
+            modalSession.visit(newTopmostVisitable, action: .restore)
         }
     }
 }

Tests/Turbo Navigator/TurboNavigatorTests.swift

@@ -82,6 +82,61 @@ final class TurboNavigationHierarchyControllerTests: XCTestCase {
         XCTAssert(navigationController.viewControllers.last is VisitableViewController)
         assertVisited(url: proposal.url, on: .main)
     }
+
+    func test_default_default_refresh_refreshesPreviousController() {
+        navigator.route(oneURL)
+        XCTAssertEqual(navigationController.viewControllers.count, 1)
+
+        navigator.route(twoURL)
+        XCTAssertEqual(navigator.rootViewController.viewControllers.count, 2)
+
+        /// Refreshing should pop the view controller and refresh the underlying controller.
+        let proposal = VisitProposal(presentation: .refresh)
+        navigator.route(proposal)
+
+        let visitable = navigator.session.activeVisitable as! VisitableViewController
+        XCTAssertEqual(visitable.visitableURL, oneURL)
+        XCTAssertEqual(navigator.rootViewController.viewControllers.count, 1)
+    }
+
+    func test_default_modal_refresh_refreshesPreviousController() {
+        navigationController.pushViewController(UIViewController(), animated: false)
+        XCTAssertEqual(navigationController.viewControllers.count, 1)
+
+        let oneURLProposal = VisitProposal(path: "/one", context: .modal)
+        navigator.route(oneURLProposal)
+
+        let twoURLProposal = VisitProposal(path: "/two", context: .modal)
+        navigator.route(twoURLProposal)
+        XCTAssertEqual(modalNavigationController.viewControllers.count, 2)
+
+        /// Refreshing should pop the view controller and refresh the underlying controller.
+        let proposal = VisitProposal(presentation: .refresh)
+        navigator.route(proposal)
+
+        let visitable = navigator.modalSession.activeVisitable as! VisitableViewController
+        XCTAssertEqual(visitable.visitableURL, oneURL)
+        XCTAssertEqual(modalNavigationController.viewControllers.count, 1)
+    }
+
+    func test_default_modal_refresh_dismissesAndRefreshesMainStackTopViewController() {
+        navigator.route(oneURL)
+        XCTAssertEqual(navigationController.viewControllers.count, 1)
+
+        let twoURLProposal = VisitProposal(path: "/two", context: .modal)
+        navigator.route(twoURLProposal)
+        XCTAssertEqual(modalNavigationController.viewControllers.count, 1)
+
+        /// Refreshing should dismiss the view controller and refresh the underlying controller.
+        let proposal = VisitProposal(context: .modal, presentation: .refresh)
+        navigator.route(proposal)
+
+        let visitable = navigator.session.activeVisitable as! VisitableViewController
+        XCTAssertEqual(visitable.visitableURL, oneURL)
+
+        XCTAssertNil(navigationController.presentedViewController)
+        XCTAssertEqual(navigator.rootViewController.viewControllers.count, 1)
+    }
 
     func test_default_modal_default_presentsModal() {
         navigationController.pushViewController(UIViewController(), animated: false)
@@ -309,7 +364,7 @@ final class TurboNavigationHierarchyControllerTests: XCTestCase {
 
 private class EmptyNavigationDelegate: TurboNavigationHierarchyControllerDelegate {
     func visit(_: Visitable, on: TurboNavigationHierarchyController.NavigationStackType, with: VisitOptions) {}
-    func refresh(navigationStack: TurboNavigationHierarchyController.NavigationStackType) {}
+    func refreshVisitable(navigationStack: TurboNavigationHierarchyController.NavigationStackType, newTopmostVisitable: any Visitable) { }
 }
 
 // MARK: - VisitProposal extension