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

apps/desktop/electron/power-save.test.ts

@@ -0,0 +1,58 @@
+import { describe, expect, it, vi } from 'vitest'
+
+import { createKeepAwake, type PowerSaveBlockerLike } from './power-save'
+
+function fakeBlocker() {
+  let next = 1
+  const started = new Set<number>()
+
+  const blocker: PowerSaveBlockerLike = {
+    isStarted: id => started.has(id),
+    start: vi.fn(() => {
+      const id = next++
+      started.add(id)
+
+      return id
+    }),
+    stop: vi.fn(id => void started.delete(id))
+  }
+
+  return { blocker, started }
+}
+
+describe('createKeepAwake', () => {
+  it('starts once, is idempotent, and stops', () => {
+    const { blocker } = fakeBlocker()
+    const keepAwake = createKeepAwake(blocker)
+
+    expect(keepAwake.isActive()).toBe(false)
+    expect(keepAwake.set(true)).toBe(true)
+    keepAwake.set(true) // idempotent — no second blocker
+    expect(blocker.start).toHaveBeenCalledTimes(1)
+    expect(blocker.start).toHaveBeenCalledWith('prevent-app-suspension')
+
+    expect(keepAwake.set(false)).toBe(false)
+    keepAwake.set(false)
+    expect(blocker.stop).toHaveBeenCalledTimes(1)
+  })
+
+  it('re-arms after the OS dropped the blocker', () => {
+    const { blocker, started } = fakeBlocker()
+    const keepAwake = createKeepAwake(blocker)
+
+    keepAwake.set(true)
+    started.clear() // system released it out from under us
+    expect(keepAwake.isActive()).toBe(false)
+
+    keepAwake.set(true)
+    expect(blocker.start).toHaveBeenCalledTimes(2)
+    expect(keepAwake.isActive()).toBe(true)
+  })
+
+  it('honors a custom blocker type', () => {
+    const { blocker } = fakeBlocker()
+    createKeepAwake(blocker, 'prevent-display-sleep').set(true)
+
+    expect(blocker.start).toHaveBeenCalledWith('prevent-display-sleep')
+  })
+})