From 33ffa2ccca58db5e546fe11ee99e2a8ceb9ca552 Mon Sep 17 00:00:00 2001
From: Cursor Agent <cursoragent@cursor.com>
Date: Sat, 16 May 2026 17:07:37 +0000
Subject: [PATCH 1/2] fix: preserve selected setup host

Co-authored-by: cooper <czxtm@users.noreply.github.com>
---
 .../widget/steps/setup-step.test.tsx          | 66 +++++++++++++++++++
 .../components/widget/steps/setup-step.tsx    |  3 +-
 2 files changed, 68 insertions(+), 1 deletion(-)
 create mode 100644 apps/native/src/components/widget/steps/setup-step.test.tsx

diff --git a/apps/native/src/components/widget/steps/setup-step.test.tsx b/apps/native/src/components/widget/steps/setup-step.test.tsx
new file mode 100644
index 00000000..605c9bd8
--- /dev/null
+++ b/apps/native/src/components/widget/steps/setup-step.test.tsx
@@ -0,0 +1,66 @@
+import "@testing-library/jest-dom";
+import { fireEvent, render, screen } from "@testing-library/react";
+import { beforeEach, describe, expect, it, vi } from "vitest";
+
+import { useWidgetStore } from "@/stores/widget-store";
+import { SetupStep } from "./setup-step";
+
+const mockSaveHost = vi.hoisted(() => vi.fn<(host: string) => Promise<void>>());
+
+vi.mock("@/hooks/use-darwin-config", () => ({
+  useDarwinConfig: () => ({
+    saveHost: mockSaveHost,
+  }),
+}));
+
+vi.mock("@/components/widget/controls/directory-picker", () => ({
+  DirectoryPicker: ({ label }: { label: string }) => (
+    <div data-testid="directory-picker">{label}</div>
+  ),
+}));
+
+vi.mock("@/components/widget/controls/bootstrap-config", () => ({
+  BootstrapConfig: ({ label }: { label: string }) => (
+    <div data-testid="bootstrap-config">{label}</div>
+  ),
+}));
+
+function resetStore() {
+  const store = useWidgetStore.getState();
+  store.setConfigDir("");
+  store.setHosts([]);
+  store.setHost("");
+  store.setError(null);
+  store.setBootstrapping(false);
+}
+
+describe("<SetupStep>", () => {
+  beforeEach(() => {
+    mockSaveHost.mockReset();
+    resetStore();
+  });
+
+  it("persists the displayed host when Next is clicked without changing the dropdown", async () => {
+    const store = useWidgetStore.getState();
+    store.setConfigDir("/Users/me/.darwin");
+    store.setHosts(["mbp"]);
+    store.setHost("mbp");
+
+    render(<SetupStep />);
+
+    fireEvent.click(await screen.findByRole("button", { name: "Next" }));
+
+    expect(mockSaveHost).toHaveBeenCalledWith("mbp");
+    expect(mockSaveHost).not.toHaveBeenCalledWith("");
+  });
+
+  it("does not allow Next to persist an empty host", async () => {
+    const store = useWidgetStore.getState();
+    store.setConfigDir("/Users/me/.darwin");
+    store.setHosts(["mbp"]);
+
+    render(<SetupStep />);
+
+    expect(await screen.findByRole("button", { name: "Next" })).toBeDisabled();
+  });
+});
diff --git a/apps/native/src/components/widget/steps/setup-step.tsx b/apps/native/src/components/widget/steps/setup-step.tsx
index a3ddf4f9..21ac369e 100644
--- a/apps/native/src/components/widget/steps/setup-step.tsx
+++ b/apps/native/src/components/widget/steps/setup-step.tsx
@@ -21,6 +21,7 @@ export function SetupStep() {
   const host = useWidgetStore((state) => state.host);
   const [configDirConfirmed, setConfigDirConfirmed] = useState(false);
   const [selectedHost, setSelectedHost] = useState<string>("");
+  const effectiveHost = selectedHost || host;
 
   const { saveHost } = useDarwinConfig();
 
@@ -82,7 +83,7 @@ export function SetupStep() {
           ) : (
             <BootstrapConfig label="2. Configuration" />
           )}
-          <Button onClick={() => saveHost(selectedHost)}>
+          <Button disabled={!effectiveHost} onClick={() => saveHost(effectiveHost)}>
             Next
           </Button>
         </div>

From f215ccbf0d70462e426c52e70d0a1ea171d6a2c1 Mon Sep 17 00:00:00 2001
From: Cursor Agent <cursoragent@cursor.com>
Date: Sat, 16 May 2026 17:08:25 +0000
Subject: [PATCH 2/2] test: cover setup host confirmation

Co-authored-by: cooper <czxtm@users.noreply.github.com>
---
 apps/native/src/components/widget/steps/setup-step.test.tsx | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/apps/native/src/components/widget/steps/setup-step.test.tsx b/apps/native/src/components/widget/steps/setup-step.test.tsx
index 605c9bd8..6ab126ed 100644
--- a/apps/native/src/components/widget/steps/setup-step.test.tsx
+++ b/apps/native/src/components/widget/steps/setup-step.test.tsx
@@ -7,6 +7,12 @@ import { SetupStep } from "./setup-step";
 
 const mockSaveHost = vi.hoisted(() => vi.fn<(host: string) => Promise<void>>());
 
+vi.mock("@/lib/env", () => ({
+  settings: {
+    NIX_INSTALLED_OVERRIDE: false,
+  },
+}));
+
 vi.mock("@/hooks/use-darwin-config", () => ({
   useDarwinConfig: () => ({
     saveHost: mockSaveHost,