feat(keyboard): add Skills tab and extract-todos Shortcut export
Ship a Skills catalog with drag-to-reorder chips and a companion Shortcut that receives extracted titles and writes them to Reminders.
This commit is contained in:
Rocky
@@ -0,0 +1,199 @@
|
||||
// AIAgentSkillLayoutTests.swift
|
||||
// OSGKeyboardTests
|
||||
|
||||
import XCTest
|
||||
@testable import OSGKeyboardShared
|
||||
|
||||
@MainActor
|
||||
final class AIAgentSkillLayoutTests: XCTestCase {
|
||||
private func makeDefaults() -> UserDefaults {
|
||||
let suite = "group.com.osgkeyboard.shared.tests.skills.\(UUID().uuidString)"
|
||||
let defaults = UserDefaults(suiteName: suite)!
|
||||
defaults.removePersistentDomain(forName: suite)
|
||||
return defaults
|
||||
}
|
||||
|
||||
func testFreshInstallEnablesDefaultTransformSkills() {
|
||||
let defaults = makeDefaults()
|
||||
let layout = AppGroupStore(defaults: defaults).agentSkillLayout
|
||||
XCTAssertEqual(layout.enabledIDs, AIAgentSkillLayout.defaultEnabledIDs)
|
||||
XCTAssertTrue(layout.confirmedShortcutIDs.isEmpty)
|
||||
}
|
||||
|
||||
func testEmptyEnabledListIsPreserved() {
|
||||
let defaults = makeDefaults()
|
||||
let store = AppGroupStore(defaults: defaults)
|
||||
store.setAgentSkillLayout(
|
||||
AIAgentSkillLayout(enabledIDs: [], confirmedShortcutIDs: [])
|
||||
)
|
||||
XCTAssertEqual(store.agentSkillLayout.enabledIDs, [])
|
||||
}
|
||||
|
||||
func testCannotEnableExportSkillBeforeShortcutConfirmation() {
|
||||
let store = AIAgentSkillLayoutStore(defaults: makeDefaults())
|
||||
XCTAssertEqual(
|
||||
store.enable(AIClipboardSkillCatalog.extractTodosID),
|
||||
.needsShortcut
|
||||
)
|
||||
XCTAssertFalse(store.layout.isEnabled(AIClipboardSkillCatalog.extractTodosID))
|
||||
}
|
||||
|
||||
func testConfirmShortcutAutoEnablesWhenSlotAvailable() {
|
||||
let store = AIAgentSkillLayoutStore(defaults: makeDefaults())
|
||||
XCTAssertEqual(
|
||||
store.confirmShortcutAndEnable(AIClipboardSkillCatalog.extractTodosID),
|
||||
.enabled
|
||||
)
|
||||
XCTAssertTrue(store.layout.hasConfirmedShortcut(AIClipboardSkillCatalog.extractTodosID))
|
||||
XCTAssertEqual(
|
||||
store.layout.enabledIDs.last,
|
||||
AIClipboardSkillCatalog.extractTodosID
|
||||
)
|
||||
}
|
||||
|
||||
func testSanitizedDropsUnconfirmedExportAndUnknownIDs() {
|
||||
let layout = AIAgentSkillLayout(
|
||||
enabledIDs: ["reply", "extractTodos", "unknown"],
|
||||
confirmedShortcutIDs: []
|
||||
).sanitized()
|
||||
XCTAssertEqual(layout.enabledIDs, ["reply"])
|
||||
}
|
||||
|
||||
func testSanitizedKeepsConfirmedExport() {
|
||||
let layout = AIAgentSkillLayout(
|
||||
enabledIDs: ["reply", "extractTodos"],
|
||||
confirmedShortcutIDs: ["extractTodos"]
|
||||
).sanitized()
|
||||
XCTAssertEqual(layout.enabledIDs, ["reply", "extractTodos"])
|
||||
}
|
||||
|
||||
func testIsFullUsesEnabledCount() {
|
||||
let full = AIAgentSkillLayout(
|
||||
enabledIDs: (0..<AIAgentSkillLayout.maximumEnabled).map(String.init),
|
||||
confirmedShortcutIDs: []
|
||||
)
|
||||
XCTAssertTrue(full.isFull)
|
||||
XCTAssertEqual(AIAgentSkillLayout.maximumEnabled, 8)
|
||||
}
|
||||
|
||||
func testDisableKeepsShortcutConfirmation() {
|
||||
let store = AIAgentSkillLayoutStore(defaults: makeDefaults())
|
||||
_ = store.confirmShortcutAndEnable(AIClipboardSkillCatalog.extractTodosID)
|
||||
store.disable(AIClipboardSkillCatalog.extractTodosID)
|
||||
XCTAssertFalse(store.layout.isEnabled(AIClipboardSkillCatalog.extractTodosID))
|
||||
XCTAssertTrue(store.layout.hasConfirmedShortcut(AIClipboardSkillCatalog.extractTodosID))
|
||||
XCTAssertEqual(store.enable(AIClipboardSkillCatalog.extractTodosID), .enabled)
|
||||
}
|
||||
|
||||
func testReorderMovesEnabledSkill() {
|
||||
let store = AIAgentSkillLayoutStore(defaults: makeDefaults())
|
||||
store.moveEnabled(
|
||||
id: AIClipboardSkillCatalog.translateID,
|
||||
onto: AIClipboardSkillCatalog.replyID
|
||||
)
|
||||
XCTAssertEqual(
|
||||
store.layout.enabledIDs,
|
||||
[
|
||||
AIClipboardSkillCatalog.translateID,
|
||||
AIClipboardSkillCatalog.replyID,
|
||||
AIClipboardSkillCatalog.summarizeID,
|
||||
]
|
||||
)
|
||||
}
|
||||
|
||||
func testVisibleEmptyEnabledIDsShowsNoChips() {
|
||||
XCTAssertEqual(AIClipboardSkillCatalog.visible(enabledIDs: []).map(\.id), [])
|
||||
}
|
||||
|
||||
func testNONEAndEmptyProduceNoItems() {
|
||||
XCTAssertEqual(AITodoExtraction.items(from: "NONE"), [])
|
||||
XCTAssertEqual(AITodoExtraction.items(from: "没有待办事项"), [])
|
||||
XCTAssertEqual(AITodoExtraction.items(from: " \n "), [])
|
||||
XCTAssertEqual(AITodoExtraction.items(from: "no tasks"), [])
|
||||
}
|
||||
|
||||
func testStripsBulletsAndCapsAtTwenty() {
|
||||
let lines = (1...25).map { "- 任务\($0)" }.joined(separator: "\n")
|
||||
let items = AITodoExtraction.items(from: lines)
|
||||
XCTAssertEqual(items.count, 20)
|
||||
XCTAssertEqual(items.first, "任务1")
|
||||
}
|
||||
|
||||
func testSingleShortTaskIsKept() {
|
||||
XCTAssertEqual(AITodoExtraction.items(from: "买牛奶"), ["买牛奶"])
|
||||
}
|
||||
|
||||
func testWholeClipboardEchoIsRejected() {
|
||||
let source = String(repeating: "这是一段很长的会议纪要内容，包含许多句子。", count: 4)
|
||||
XCTAssertEqual(
|
||||
AITodoExtraction.items(from: source, sourceClipboard: source),
|
||||
[]
|
||||
)
|
||||
}
|
||||
|
||||
func testPendingShortcutPayloadExpires() {
|
||||
let old = AIAgentShortcutRunPayload(
|
||||
skillID: "extractTodos",
|
||||
titles: ["买牛奶"],
|
||||
createdAt: Date(timeIntervalSinceNow: -120)
|
||||
)
|
||||
let data = AIAgentShortcutRun.encode(old)!
|
||||
XCTAssertNil(AIAgentShortcutRun.decode(data))
|
||||
}
|
||||
|
||||
func testShortcutsRunURLEncodesNameAndText() {
|
||||
let url = AIAgentShortcutRun.shortcutsRunURL(name: "OSG · 提取待办", text: "买牛奶\n回邮件")
|
||||
XCTAssertEqual(url?.scheme, "shortcuts")
|
||||
XCTAssertEqual(url?.host, "run-shortcut")
|
||||
let items = URLComponents(url: url!, resolvingAgainstBaseURL: false)?.queryItems ?? []
|
||||
XCTAssertEqual(items.first { $0.name == "name" }?.value, "OSG · 提取待办")
|
||||
XCTAssertEqual(items.first { $0.name == "input" }?.value, "text")
|
||||
XCTAssertEqual(items.first { $0.name == "text" }?.value, "买牛奶\n回邮件")
|
||||
XCTAssertNil(items.first { $0.name == "x-success" })
|
||||
}
|
||||
|
||||
func testXCallbackRunURLUsesCallbackHost() {
|
||||
let url = AIAgentShortcutRun.shortcutsRunURL(
|
||||
name: "OSG · 提取待办",
|
||||
text: "买牛奶",
|
||||
xSuccess: "osgkeyboard://skill/shortcut-result?status=success",
|
||||
xError: "osgkeyboard://skill/shortcut-result?status=error",
|
||||
xCancel: "osgkeyboard://skill/shortcut-result?status=cancel"
|
||||
)
|
||||
XCTAssertEqual(url?.scheme, "shortcuts")
|
||||
XCTAssertEqual(url?.host, "x-callback-url")
|
||||
XCTAssertEqual(url?.path, "/run-shortcut")
|
||||
let items = URLComponents(url: url!, resolvingAgainstBaseURL: false)?.queryItems ?? []
|
||||
XCTAssertEqual(items.first { $0.name == "text" }?.value, "买牛奶")
|
||||
XCTAssertEqual(
|
||||
items.first { $0.name == "x-success" }?.value,
|
||||
"osgkeyboard://skill/shortcut-result?status=success"
|
||||
)
|
||||
}
|
||||
|
||||
func testPreviewEscapesNewlines() {
|
||||
XCTAssertEqual(AIAgentShortcutRun.preview("买牛奶\n回邮件"), "买牛奶\\n回邮件")
|
||||
}
|
||||
|
||||
func testExtractTodosUsesICloudShareLink() {
|
||||
let skill = AIClipboardSkillCatalog.skill(id: AIClipboardSkillCatalog.extractTodosID)
|
||||
XCTAssertEqual(
|
||||
skill?.shortcutICloudURL,
|
||||
AIClipboardSkillCatalog.extractTodosShortcutICloudURL
|
||||
)
|
||||
XCTAssertEqual(skill?.shortcutICloudURL?.host, "www.icloud.com")
|
||||
XCTAssertEqual(skill?.shortcutName, "OSG · 提取待办")
|
||||
}
|
||||
|
||||
func testICloudShareLinkMapsToShortcutsInstallURL() {
|
||||
let share = URL(string: "https://www.icloud.com/shortcuts/520317da7ae74759b64d5fb069c71f81")!
|
||||
let url = AIAgentShortcutRun.shortcutsInstallURL(from: share)
|
||||
XCTAssertEqual(url?.scheme, "shortcuts")
|
||||
XCTAssertEqual(url?.host, "shortcuts")
|
||||
XCTAssertEqual(url?.path, "/520317da7ae74759b64d5fb069c71f81")
|
||||
XCTAssertEqual(
|
||||
AIAgentShortcutRun.iCloudShareToken(from: share),
|
||||
"520317da7ae74759b64d5fb069c71f81"
|
||||
)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user