Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license

apps/desktop/electron/hud-snap-shortcut.test.ts

@@ -0,0 +1,55 @@
+import { describe, expect, it, vi } from 'vitest'
+
+import { createHudSnapShortcut, DEFAULT_HUD_SNAP_SHORTCUT } from './hud-snap-shortcut'
+import type { GlobalShortcutLike } from './quick-entry'
+
+function fakeGlobalShortcut(options: { register?: boolean; taken?: string[] } = {}) {
+  const held = new Set(options.taken ?? [])
+
+  const globalShortcut: GlobalShortcutLike = {
+    isRegistered: vi.fn((accelerator: string) => held.has(accelerator)),
+    register: vi.fn((accelerator: string, _callback: () => void) => {
+      if (options.register === false || held.has(accelerator)) {
+        return false
+      }
+
+      held.add(accelerator)
+
+      return true
+    }),
+    unregister: vi.fn((accelerator: string) => void held.delete(accelerator))
+  }
+
+  return { globalShortcut, held }
+}
+
+describe('createHudSnapShortcut', () => {
+  it('registers CommandOrControl+Shift+G while the HUD is up', () => {
+    const snap = vi.fn<() => void>()
+    const { globalShortcut } = fakeGlobalShortcut()
+
+    const controller = createHudSnapShortcut(globalShortcut, snap)
+
+    expect(controller.register()).toBe(true)
+    expect(globalShortcut.isRegistered(DEFAULT_HUD_SNAP_SHORTCUT)).toBe(true)
+  })
+
+  it('dispose releases the accelerator', () => {
+    const snap = vi.fn<() => void>()
+    const { globalShortcut } = fakeGlobalShortcut()
+    const controller = createHudSnapShortcut(globalShortcut, snap)
+
+    controller.register()
+    controller.dispose()
+
+    expect(globalShortcut.isRegistered(DEFAULT_HUD_SNAP_SHORTCUT)).toBe(false)
+  })
+
+  it('register fails when the chord is already taken', () => {
+    const snap = vi.fn<() => void>()
+    const { globalShortcut } = fakeGlobalShortcut({ taken: [DEFAULT_HUD_SNAP_SHORTCUT] })
+    const controller = createHudSnapShortcut(globalShortcut, snap)
+
+    expect(controller.register()).toBe(false)
+  })
+})