fix: set localStorage for IncompatibleWorkspace story to display error view

Author: ammar-agent

src/browser/App.stories.tsx

@@ -386,49 +386,71 @@ export const ManyWorkspaces: Story = {
  */
 export const IncompatibleWorkspace: Story = {
   render: () => {
-    const workspaceId = "incompatible-ws";
+    const AppWithIncompatibleWorkspace = () => {
+      const initialized = useRef(false);
 
-    const projects = new Map<string, ProjectConfig>([
-      [
-        "/home/user/projects/my-app",
-        {
-          workspaces: [
-            { path: "/home/user/.mux/src/my-app/main", id: "my-app-main", name: "main" },
-            {
-              path: "/home/user/.mux/src/my-app/incompatible",
-              id: workspaceId,
-              name: "incompatible",
-            },
-          ],
-        },
-      ],
-    ]);
+      if (!initialized.current) {
+        const workspaceId = "incompatible-ws";
 
-    const workspaces: FrontendWorkspaceMetadata[] = [
-      {
-        id: "my-app-main",
-        name: "main",
-        projectPath: "/home/user/projects/my-app",
-        projectName: "my-app",
-        namedWorkspacePath: "/home/user/.mux/src/my-app/main",
-        runtimeConfig: DEFAULT_RUNTIME_CONFIG,
-      },
-      {
-        id: workspaceId,
-        name: "incompatible",
-        projectPath: "/home/user/projects/my-app",
-        projectName: "my-app",
-        namedWorkspacePath: "/home/user/.mux/src/my-app/incompatible",
-        runtimeConfig: DEFAULT_RUNTIME_CONFIG,
-        // This field is set when a workspace has an incompatible runtime config
-        incompatibleRuntime:
-          "This workspace was created with a newer version of mux.\nPlease upgrade mux to use this workspace.",
-      },
-    ];
+        const workspaces: FrontendWorkspaceMetadata[] = [
+          {
+            id: "my-app-main",
+            name: "main",
+            projectPath: "/home/user/projects/my-app",
+            projectName: "my-app",
+            namedWorkspacePath: "/home/user/.mux/src/my-app/main",
+            runtimeConfig: DEFAULT_RUNTIME_CONFIG,
+          },
+          {
+            id: workspaceId,
+            name: "incompatible",
+            projectPath: "/home/user/projects/my-app",
+            projectName: "my-app",
+            namedWorkspacePath: "/home/user/.mux/src/my-app/incompatible",
+            runtimeConfig: DEFAULT_RUNTIME_CONFIG,
+            // This field is set when a workspace has an incompatible runtime config
+            incompatibleRuntime:
+              "This workspace was created with a newer version of mux.\nPlease upgrade mux to use this workspace.",
+          },
+        ];
+
+        setupMockAPI({
+          projects: new Map([
+            [
+              "/home/user/projects/my-app",
+              {
+                workspaces: [
+                  { path: "/home/user/.mux/src/my-app/main", id: "my-app-main", name: "main" },
+                  {
+                    path: "/home/user/.mux/src/my-app/incompatible",
+                    id: workspaceId,
+                    name: "incompatible",
+                  },
+                ],
+              },
+            ],
+          ]),
+          workspaces,
+        });
+
+        // Set initial workspace selection to the incompatible workspace
+        localStorage.setItem(
+          "selectedWorkspace",
+          JSON.stringify({
+            workspaceId: workspaceId,
+            projectPath: "/home/user/projects/my-app",
+            projectName: "my-app",
+            namedWorkspacePath: "/home/user/.mux/src/my-app/incompatible",
+          })
+        );
+
+        initialized.current = true;
+      }
+
+      return <AppLoader />;
+    };
 
-    return (
-      <AppWithMocks projects={projects} workspaces={workspaces} selectedWorkspaceId={workspaceId} />
-    );
+    return <AppWithIncompatibleWorkspace />;
   },
 };