| ... | @@ -10,6 +10,7 @@ import type { RemoteOpenState } from "../../remoteOpen"; | ... | @@ -10,6 +10,7 @@ import type { RemoteOpenState } from "../../remoteOpen"; |
| 10 | const state = vi.hoisted(() => ({ | 10 | const state = vi.hoisted(() => ({ |
| 11 | allowed: new Set<string>(), | 11 | allowed: new Set<string>(), |
| 12 | remote: { mode: "local-exec" } as RemoteOpenState, | 12 | remote: { mode: "local-exec" } as RemoteOpenState, |
| | 13 | clientEditors: ["vscode"] as readonly EditorId[], |
| 13 | run: vi.fn(), | 14 | run: vi.fn(), |
| 14 | openUrl: vi.fn(), | 15 | openUrl: vi.fn(), |
| 15 | setPreferred: vi.fn(), | 16 | setPreferred: vi.fn(), |
| ... | @@ -37,15 +38,25 @@ vi.mock("../../state/session", async () => { | ... | @@ -37,15 +38,25 @@ vi.mock("../../state/session", async () => { |
| 37 | vi.mock("../../state/shell", () => ({ shellEnvironment: { openInEditor: "openInEditor" } })); | 38 | vi.mock("../../state/shell", () => ({ shellEnvironment: { openInEditor: "openInEditor" } })); |
| 38 | vi.mock("../../state/use-atom-command", () => ({ useAtomCommand: () => state.run })); | 39 | vi.mock("../../state/use-atom-command", () => ({ useAtomCommand: () => state.run })); |
| 39 | vi.mock("../../state/environments", () => ({ useEnvironment: () => ({ label: "Test host" }) })); | 40 | vi.mock("../../state/environments", () => ({ useEnvironment: () => ({ label: "Test host" }) })); |
| 40 | vi.mock("../../editorPreferences", () => ({ | 41 | vi.mock("../../editorPreferences", async (importOriginal) => { |
| 41 | usePreferredEditor: (available: readonly EditorId[]) => [ | 42 | const actual = await importOriginal<typeof import("../../editorPreferences")>(); |
| 42 | available[0] ?? null, | 43 | return { |
| 43 | state.setPreferred, | 44 | ...actual, |
| 44 | ], | 45 | usePreferredEditor: (available: readonly EditorId[]) => { |
| 45 | })); | 46 | const [preferred, setPreferred] = actual.usePreferredEditor(available); |
| | 47 | return [ |
| | 48 | preferred, |
| | 49 | (editor: EditorId) => { |
| | 50 | state.setPreferred(editor); |
| | 51 | setPreferred(editor); |
| | 52 | }, |
| | 53 | ] as const; |
| | 54 | }, |
| | 55 | }; |
| | 56 | }); |
| 46 | vi.mock("../../remoteOpen", () => ({ | 57 | vi.mock("../../remoteOpen", () => ({ |
| 47 | useRemoteOpenState: () => state.remote, | 58 | useRemoteOpenState: () => state.remote, |
| 48 | useRemoteCapableEditors: () => editors, | 59 | useRemoteCapableEditors: () => state.clientEditors, |
| 49 | useRemoteOpenHint: () => [false, state.markHintSeen], | 60 | useRemoteOpenHint: () => [false, state.markHintSeen], |
| 50 | openRemoteEditorUrl: state.openUrl, | 61 | openRemoteEditorUrl: state.openUrl, |
| 51 | })); | 62 | })); |
| ... | @@ -104,6 +115,7 @@ let renderer: ReactTestRenderer | undefined; | ... | @@ -104,6 +115,7 @@ let renderer: ReactTestRenderer | undefined; |
| 104 | beforeEach(() => { | 115 | beforeEach(() => { |
| 105 | state.allowed.clear(); | 116 | state.allowed.clear(); |
| 106 | state.remote = { mode: "local-exec" }; | 117 | state.remote = { mode: "local-exec" }; |
| | 118 | state.clientEditors = editors; |
| 107 | state.run.mockReset().mockResolvedValue(AsyncResult.success(undefined)); | 119 | state.run.mockReset().mockResolvedValue(AsyncResult.success(undefined)); |
| 108 | state.openUrl.mockReset().mockResolvedValue(true); | 120 | state.openUrl.mockReset().mockResolvedValue(true); |
| 109 | state.setPreferred.mockReset(); | 121 | state.setPreferred.mockReset(); |
| ... | @@ -112,11 +124,21 @@ beforeEach(() => { | ... | @@ -112,11 +124,21 @@ beforeEach(() => { |
| 112 | state.listeners.clear(); | 124 | state.listeners.clear(); |
| 113 | vi.stubGlobal("IS_REACT_ACT_ENVIRONMENT", true); | 125 | vi.stubGlobal("IS_REACT_ACT_ENVIRONMENT", true); |
| 114 | vi.stubGlobal("navigator", { platform: "Linux" }); | 126 | vi.stubGlobal("navigator", { platform: "Linux" }); |
| | 127 | const events = new EventTarget(); |
| | 128 | const storage = new Map<string, string>(); |
| 115 | vi.stubGlobal("window", { | 129 | vi.stubGlobal("window", { |
| 116 | addEventListener: (_type: string, callback: (event: KeyboardEvent) => void) => { | 130 | localStorage: { |
| 117 | state.keydown = callback; | 131 | getItem: (key: string) => storage.get(key) ?? null, |
| | 132 | setItem: (key: string, value: string) => storage.set(key, value), |
| | 133 | removeItem: (key: string) => storage.delete(key), |
| | 134 | }, |
| | 135 | dispatchEvent: (event: Event) => events.dispatchEvent(event), |
| | 136 | addEventListener: (type: string, callback: EventListener) => { |
| | 137 | events.addEventListener(type, callback); |
| | 138 | if (type === "keydown") state.keydown = callback; |
| 118 | }, | 139 | }, |
| 119 | removeEventListener: (_type: string, callback: (event: KeyboardEvent) => void) => { | 140 | removeEventListener: (type: string, callback: EventListener) => { |
| | 141 | events.removeEventListener(type, callback); |
| 120 | if (state.keydown === callback) state.keydown = null; | 142 | if (state.keydown === callback) state.keydown = null; |
| 121 | }, | 143 | }, |
| 122 | }); | 144 | }); |
| ... | @@ -128,16 +150,17 @@ afterEach(async () => { | ... | @@ -128,16 +150,17 @@ afterEach(async () => { |
| 128 | vi.unstubAllGlobals(); | 150 | vi.unstubAllGlobals(); |
| 129 | }); | 151 | }); |
| 130 | | 152 | |
| 131 | async function renderPicker() { | 153 | async function renderPicker(availableEditors: readonly EditorId[] = editors, panel = false) { |
| 132 | await act(async () => { | 154 | await act(async () => { |
| 133 | for (const listener of state.listeners) listener(); | 155 | for (const listener of state.listeners) listener(); |
| 134 | const node = ( | 156 | const node = ( |
| 135 | <OpenInPicker | 157 | <OpenInPicker |
| 136 | environmentId={selected} | 158 | environmentId={selected} |
| 137 | keybindings={DEFAULT_RESOLVED_KEYBINDINGS} | 159 | keybindings={DEFAULT_RESOLVED_KEYBINDINGS} |
| 138 | availableEditors={editors} | 160 | availableEditors={availableEditors} |
| 139 | openInCwd="/work/project" | 161 | openInCwd="/work/project" |
| 140 | compact | 162 | displayMode={panel ? "panel" : "toolbar"} |
| | 163 | compact={!panel} |
| 141 | /> | 164 | /> |
| 142 | ); | 165 | ); |
| 143 | if (renderer) renderer.update(node); | 166 | if (renderer) renderer.update(node); |
| ... | @@ -153,6 +176,37 @@ function keyboardEvent() { | ... | @@ -153,6 +176,37 @@ function keyboardEvent() { |
| 153 | return { preventDefault: vi.fn() } as unknown as KeyboardEvent; | 176 | return { preventDefault: vi.fn() } as unknown as KeyboardEvent; |
| 154 | } | 177 | } |
| 155 | | 178 | |
| | 179 | it.each(["local-exec", "remote-links"] as const)( |
| | 180 | "selects the preferred editor without launching a %s workspace", |
| | 181 | async (mode) => { |
| | 182 | state.allowed.add(selected); |
| | 183 | state.clientEditors = ["cursor", "vscode"]; |
| | 184 | state.remote = |
| | 185 | mode === "local-exec" ? { mode } : { mode, host: { kind: "ssh-alias", host: "test-host" } }; |
| | 186 | await renderPicker(["cursor", "vscode"], true); |
| | 187 | expect(renderer!.root.findByProps({ "aria-label": "Open in Cursor" })).toBeDefined(); |
| | 188 | await act(async () => { |
| | 189 | renderer!.root.findByProps({ children: "VS Code" }).parent!.props.onClick(); |
| | 190 | }); |
| | 191 | expect(window.localStorage.getItem("t3code:last-editor")).toBe('"vscode"'); |
| | 192 | const openButton = renderer!.root.findByProps({ "aria-label": "Open in VS Code" }); |
| | 193 | expect(state.run).not.toHaveBeenCalled(); |
| | 194 | expect(state.openUrl).not.toHaveBeenCalled(); |
| | 195 | |
| | 196 | await act(async () => openButton.props.onClick()); |
| | 197 | if (mode === "local-exec") { |
| | 198 | expect(state.run).toHaveBeenCalledExactlyOnceWith({ |
| | 199 | environmentId: selected, |
| | 200 | input: { cwd: "/work/project", editor: "vscode" }, |
| | 201 | }); |
| | 202 | } else { |
| | 203 | expect(state.openUrl).toHaveBeenCalledExactlyOnceWith( |
| | 204 | "vscode://vscode-remote/ssh-remote+test-host/work/project", |
| | 205 | ); |
| | 206 | } |
| | 207 | }, |
| | 208 | ); |
| | 209 | |
| 156 | describe("host editor access", () => { | 210 | describe("host editor access", () => { |
| 157 | it("uses the selected grant for both the button and a retained shortcut", async () => { | 211 | it("uses the selected grant for both the button and a retained shortcut", async () => { |
| 158 | state.allowed.add(selected); | 212 | state.allowed.add(selected); |