| ... | ... | @@ -10,6 +10,7 @@ import type { RemoteOpenState } from "../../remoteOpen"; |
| 10 | 10 | const state = vi.hoisted(() => ({ |
| 11 | 11 | allowed: new Set<string>(), |
| 12 | 12 | remote: { mode: "local-exec" } as RemoteOpenState, |
| 13 | clientEditors: ["vscode"] as readonly EditorId[], |
| 13 | 14 | run: vi.fn(), |
| 14 | 15 | openUrl: vi.fn(), |
| 15 | 16 | setPreferred: vi.fn(), |
| ... | ... | @@ -37,15 +38,25 @@ vi.mock("../../state/session", async () => { |
| 37 | 38 | vi.mock("../../state/shell", () => ({ shellEnvironment: { openInEditor: "openInEditor" } })); |
| 38 | 39 | vi.mock("../../state/use-atom-command", () => ({ useAtomCommand: () => state.run })); |
| 39 | 40 | vi.mock("../../state/environments", () => ({ useEnvironment: () => ({ label: "Test host" }) })); |
| 40 | | vi.mock("../../editorPreferences", () => ({ |
| 41 | | usePreferredEditor: (available: readonly EditorId[]) => [ |
| 42 | | available[0] ?? null, |
| 43 | | state.setPreferred, |
| 44 | | ], |
| 45 | | })); |
| 41 | vi.mock("../../editorPreferences", async (importOriginal) => { |
| 42 | const actual = await importOriginal<typeof import("../../editorPreferences")>(); |
| 43 | return { |
| 44 | ...actual, |
| 45 | usePreferredEditor: (available: readonly EditorId[]) => { |
| 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 | 57 | vi.mock("../../remoteOpen", () => ({ |
| 47 | 58 | useRemoteOpenState: () => state.remote, |
| 48 | | useRemoteCapableEditors: () => editors, |
| 59 | useRemoteCapableEditors: () => state.clientEditors, |
| 49 | 60 | useRemoteOpenHint: () => [false, state.markHintSeen], |
| 50 | 61 | openRemoteEditorUrl: state.openUrl, |
| 51 | 62 | })); |
| ... | ... | @@ -104,6 +115,7 @@ let renderer: ReactTestRenderer | undefined; |
| 104 | 115 | beforeEach(() => { |
| 105 | 116 | state.allowed.clear(); |
| 106 | 117 | state.remote = { mode: "local-exec" }; |
| 118 | state.clientEditors = editors; |
| 107 | 119 | state.run.mockReset().mockResolvedValue(AsyncResult.success(undefined)); |
| 108 | 120 | state.openUrl.mockReset().mockResolvedValue(true); |
| 109 | 121 | state.setPreferred.mockReset(); |
| ... | ... | @@ -112,11 +124,21 @@ beforeEach(() => { |
| 112 | 124 | state.listeners.clear(); |
| 113 | 125 | vi.stubGlobal("IS_REACT_ACT_ENVIRONMENT", true); |
| 114 | 126 | vi.stubGlobal("navigator", { platform: "Linux" }); |
| 127 | const events = new EventTarget(); |
| 128 | const storage = new Map<string, string>(); |
| 115 | 129 | vi.stubGlobal("window", { |
| 116 | | addEventListener: (_type: string, callback: (event: KeyboardEvent) => void) => { |
| 117 | | state.keydown = callback; |
| 130 | localStorage: { |
| 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 | 142 | if (state.keydown === callback) state.keydown = null; |
| 121 | 143 | }, |
| 122 | 144 | }); |
| ... | ... | @@ -128,16 +150,17 @@ afterEach(async () => { |
| 128 | 150 | vi.unstubAllGlobals(); |
| 129 | 151 | }); |
| 130 | 152 | |
| 131 | | async function renderPicker() { |
| 153 | async function renderPicker(availableEditors: readonly EditorId[] = editors, panel = false) { |
| 132 | 154 | await act(async () => { |
| 133 | 155 | for (const listener of state.listeners) listener(); |
| 134 | 156 | const node = ( |
| 135 | 157 | <OpenInPicker |
| 136 | 158 | environmentId={selected} |
| 137 | 159 | keybindings={DEFAULT_RESOLVED_KEYBINDINGS} |
| 138 | | availableEditors={editors} |
| 160 | availableEditors={availableEditors} |
| 139 | 161 | openInCwd="/work/project" |
| 140 | | compact |
| 162 | displayMode={panel ? "panel" : "toolbar"} |
| 163 | compact={!panel} |
| 141 | 164 | /> |
| 142 | 165 | ); |
| 143 | 166 | if (renderer) renderer.update(node); |
| ... | ... | @@ -153,6 +176,37 @@ function keyboardEvent() { |
| 153 | 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 | 210 | describe("host editor access", () => { |
| 157 | 211 | it("uses the selected grant for both the button and a retained shortcut", async () => { |
| 158 | 212 | state.allowed.add(selected); |