Drill into TAB_NAVIGATOR in session storage fallback for workspace restore

The session-storage fallback in useRestoreWorkspacesTabOnNavigate searched
for WORKSPACE_NAVIGATOR at the root level of the state from getStateFromPath,
but getStateFromPath returns state rooted at TAB_NAVIGATOR. This caused the
findLast to never find WORKSPACE_NAVIGATOR, falling through to the workspaces
list instead of restoring the last workspace sub-page after a refresh.

Updated the test mock to match the real getStateFromPath output shape.

Co-authored-by: Sobit Neupane <sobitneupane@users.noreply.github.com>

Author: MelvinBot

src/hooks/useRestoreWorkspacesTabOnNavigate.ts

@@ -55,7 +55,9 @@ function useRestoreWorkspacesTabOnNavigate() {
         }
 
         // Fall back to session storage when no workspace route exists anywhere in the navigation tree.
-        const sessionRoute = getWorkspacesTabStateFromSessionStorage()
+        // getStateFromPath returns state rooted at TAB_NAVIGATOR, so we must drill into it first.
+        const sessionTabNavigatorRoute = getWorkspacesTabStateFromSessionStorage()?.routes?.findLast((route) => route.name === NAVIGATORS.TAB_NAVIGATOR);
+        const sessionRoute = getTabState(sessionTabNavigatorRoute)
             ?.routes?.findLast((route) => route.name === NAVIGATORS.WORKSPACE_NAVIGATOR)
             ?.state?.routes?.findLast((route) => isWorkspaceNavigatorRouteName(route.name));
         if (sessionRoute) {

tests/unit/hooks/useRestoreWorkspacesTabOnNavigate.test.ts

@@ -280,16 +280,23 @@ describe('useRestoreWorkspacesTabOnNavigate', () => {
         lastVisitedTabPathUtils.getWorkspacesTabStateFromSessionStorage.mockReturnValue({
             routes: [
                 {
-                    name: NAVIGATORS.WORKSPACE_NAVIGATOR,
+                    name: NAVIGATORS.TAB_NAVIGATOR,
                     state: {
                         routes: [
                             {
-                                name: NAVIGATORS.WORKSPACE_SPLIT_NAVIGATOR,
+                                name: NAVIGATORS.WORKSPACE_NAVIGATOR,
                                 state: {
-                                    index: 1,
                                     routes: [
-                                        {name: SCREENS.WORKSPACE.INITIAL, params: {policyID: fakePolicyID}},
-                                        {name: SCREENS.WORKSPACE.WORKFLOWS, params: {policyID: fakePolicyID}},
+                                        {
+                                            name: NAVIGATORS.WORKSPACE_SPLIT_NAVIGATOR,
+                                            state: {
+                                                index: 1,
+                                                routes: [
+                                                    {name: SCREENS.WORKSPACE.INITIAL, params: {policyID: fakePolicyID}},
+                                                    {name: SCREENS.WORKSPACE.WORKFLOWS, params: {policyID: fakePolicyID}},
+                                                ],
+                                            },
+                                        },
                                     ],
                                 },
                             },