Files
claude-code-haha/native/cu-helper/Tests/CuHelperTests/DaemonSessionGateTests.swift
T
程序员阿江(Relakkes) b8a90626ce feat(computer-use): native macOS engine, on top of main and nothing else
Rebuilt against main so the branch carries the Computer Use work and no other
divergence. Three unrelated efforts had been sitting uncommitted in this
worktree and were swept into an earlier commit; they are preserved on
cu-worktree-full-backup and belong on their own branches — adapter control
credentials, Electron asar sealing, and the sidecar code-loading audit. Every
file outside Computer Use now matches main exactly.

The engine
  A Swift helper drives apps through the accessibility tree, with coordinate
  actuation for the Chromium and Electron apps whose tree is a bare window
  frame. Ten primitives matching the shape Codex uses, so an app's guidance and
  the model's habits transfer.

  Coordinate actions resolve their target window once and refuse when none can
  be named. The unbound event they used to fall back to is discarded by custom
  renderers, so a minimized target produced a whole session of "Action
  completed" with nothing behind it.

  Input acceptance is established for typing and key presses as well as clicks:
  each MCP call is seconds apart, so the keyboard cannot inherit the focus a
  click established. The synthetic focus notification is gated on the target
  not already being active — sent unconditionally it names window 0 at an app
  that already owns a key window, and nine window-bound clicks were discarded
  with the traffic lights fully lit.

State the model can trust
  An off-screen target says so, and says which tools still reach it: element
  actions need no on-screen geometry, so an app with a real tree can still be
  driven from the Dock. A fully covered window is recovered once, then left
  alone — burying it again is the user wanting their screen back. A repeated
  capture is reported with the cause that actually applies rather than both,
  because coverage is something we compute.

Signing
  The helper is signed under a stable identity before electron-builder sees it,
  and excluded from re-signing: macOS ties Accessibility and Screen Recording
  grants to the signing identity, so rotating it drops both on every update.

Discoverability
  The desktop slash menu falls back to a directory scan while a session's CLI
  has not started, which is when the menu is first opened. Built-ins and
  bundled skills live in the binary, so /computer-use was absent until after
  the first message.
2026-08-05 22:06:23 +08:00

88 lines
3.2 KiB
Swift

import XCTest
@testable import cc_haha_computer_use
final class DaemonSessionGateTests: XCTestCase {
func testDisconnectedInflightSessionDefersCleanupAndRejectsNewConnection() {
var gate = DaemonSessionGate()
let first = acceptedGeneration(gate.acceptConnection())
XCTAssertTrue(gate.beginRequest(generation: first))
XCTAssertEqual(
gate.disconnect(generation: first),
.deferCleanup
)
XCTAssertEqual(gate.acceptConnection(), .rejectBusy)
XCTAssertFalse(gate.beginRequest(generation: first))
XCTAssertEqual(
gate.finishRequest(generation: first),
.terminateNow
)
let second = acceptedGeneration(gate.acceptConnection())
XCTAssertGreaterThan(second, first)
}
func testIdleSupersedeInvalidatesEveryQueuedOldRequest() {
var gate = DaemonSessionGate()
let first = acceptedGeneration(gate.acceptConnection())
let decision = gate.acceptConnection()
guard case let .accept(second, cleanupSuperseded) = decision else {
return XCTFail("expected accepted superseding connection")
}
XCTAssertTrue(cleanupSuperseded)
XCTAssertFalse(gate.beginRequest(generation: first))
XCTAssertTrue(gate.beginRequest(generation: second))
XCTAssertEqual(gate.finishRequest(generation: second), .none)
}
func testIdleDisconnectTerminatesImmediatelyAndStaleCallbacksAreIgnored() {
var gate = DaemonSessionGate()
let first = acceptedGeneration(gate.acceptConnection())
XCTAssertEqual(gate.disconnect(generation: first), .terminateNow)
XCTAssertEqual(gate.disconnect(generation: first), .none)
XCTAssertEqual(gate.finishRequest(generation: first), .none)
XCTAssertFalse(gate.beginRequest(generation: first))
}
func testOnlyOneRequestCanBeInflightPerGeneration() {
var gate = DaemonSessionGate()
let generation = acceptedGeneration(gate.acceptConnection())
XCTAssertTrue(gate.beginRequest(generation: generation))
XCTAssertFalse(gate.beginRequest(generation: generation))
XCTAssertEqual(gate.finishRequest(generation: generation), .none)
XCTAssertTrue(gate.beginRequest(generation: generation))
}
func testSupersededConnectionCloseCannotTerminateReplacement() {
var gate = DaemonSessionGate()
let first = acceptedGeneration(gate.acceptConnection())
let second = acceptedGeneration(gate.acceptConnection())
XCTAssertEqual(gate.disconnect(generation: first), .none)
XCTAssertTrue(gate.beginRequest(generation: second))
XCTAssertEqual(gate.finishRequest(generation: second), .none)
}
func testConnectionTokenInvalidationIsPermanent() {
let token = DaemonSessionToken()
XCTAssertTrue(token.isValid)
token.invalidate()
XCTAssertFalse(token.isValid)
token.invalidate()
XCTAssertFalse(token.isValid)
}
private func acceptedGeneration(
_ decision: DaemonSessionGate.AcceptDecision
) -> UInt64 {
guard case let .accept(generation, _) = decision else {
XCTFail("expected accepted connection")
return 0
}
return generation
}
}