mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 16:30:47 +08:00
oliver
dbbe3add6a
同时收敛启动与运维脚本默认行为（含 wiki worker）、更新 Admin 可观测性与相关测试，降低首轮时延并提高运行稳定性。 Made-with: Cursor
98 lines
3.2 KiB
Swift
98 lines
3.2 KiB
Swift
import Foundation
|
|
import OpenClawProtocol
|
|
import Testing
|
|
@testable import OpenClaw
|
|
|
|
@MainActor
|
|
struct WorkActivityStoreTests {
|
|
@Test func `main session job preempts other`() {
|
|
let store = WorkActivityStore()
|
|
|
|
store.handleJob(sessionKey: "discord:group:1", state: "started")
|
|
#expect(store.iconState == .workingOther(.job))
|
|
#expect(store.current?.sessionKey == "discord:group:1")
|
|
|
|
store.handleJob(sessionKey: "main", state: "started")
|
|
#expect(store.iconState == .workingMain(.job))
|
|
#expect(store.current?.sessionKey == "main")
|
|
|
|
store.handleJob(sessionKey: "main", state: "finished")
|
|
#expect(store.iconState == .workingOther(.job))
|
|
#expect(store.current?.sessionKey == "discord:group:1")
|
|
|
|
store.handleJob(sessionKey: "discord:group:1", state: "finished")
|
|
#expect(store.iconState == .idle)
|
|
#expect(store.current == nil)
|
|
}
|
|
|
|
@Test func `job stays working after tool result grace`() async {
|
|
let store = WorkActivityStore()
|
|
|
|
store.handleJob(sessionKey: "main", state: "started")
|
|
#expect(store.iconState == .workingMain(.job))
|
|
|
|
store.handleTool(
|
|
sessionKey: "main",
|
|
phase: "start",
|
|
name: "read",
|
|
meta: nil,
|
|
args: ["path": AnyCodable("/tmp/file.txt")])
|
|
#expect(store.iconState == .workingMain(.tool(.read)))
|
|
|
|
store.handleTool(
|
|
sessionKey: "main",
|
|
phase: "result",
|
|
name: "read",
|
|
meta: nil,
|
|
args: ["path": AnyCodable("/tmp/file.txt")])
|
|
|
|
for _ in 0..<50 {
|
|
if store.iconState == .workingMain(.job) { break }
|
|
try? await Task.sleep(nanoseconds: 100_000_000)
|
|
}
|
|
#expect(store.iconState == .workingMain(.job))
|
|
|
|
store.handleJob(sessionKey: "main", state: "done")
|
|
#expect(store.iconState == .idle)
|
|
}
|
|
|
|
@Test func `tool label extracts first line and shortens home`() {
|
|
let store = WorkActivityStore()
|
|
let home = NSHomeDirectory()
|
|
|
|
store.handleTool(
|
|
sessionKey: "main",
|
|
phase: "start",
|
|
name: "bash",
|
|
meta: nil,
|
|
args: [
|
|
"command": AnyCodable("echo hi\necho bye"),
|
|
"path": AnyCodable("\(home)/Projects/openclaw"),
|
|
])
|
|
|
|
#expect(store.current?.label == "bash: echo hi")
|
|
#expect(store.iconState == .workingMain(.tool(.bash)))
|
|
|
|
store.handleTool(
|
|
sessionKey: "main",
|
|
phase: "start",
|
|
name: "read",
|
|
meta: nil,
|
|
args: ["path": AnyCodable("\(home)/secret.txt")])
|
|
|
|
#expect(store.current?.label == "read: ~/secret.txt")
|
|
#expect(store.iconState == .workingMain(.tool(.read)))
|
|
}
|
|
|
|
@Test func `resolve icon state honors override selection`() {
|
|
let store = WorkActivityStore()
|
|
store.handleJob(sessionKey: "main", state: "started")
|
|
#expect(store.iconState == .workingMain(.job))
|
|
|
|
store.resolveIconState(override: .idle)
|
|
#expect(store.iconState == .idle)
|
|
|
|
store.resolveIconState(override: .otherEdit)
|
|
#expect(store.iconState == .overridden(.tool(.edit)))
|
|
}
|
|
}
|