mirror of https://github.com/openclaw/openclaw.git
96 lines
3.9 KiB
Swift
96 lines
3.9 KiB
Swift
import Foundation
|
||
import SwabbleKit
|
||
import Testing
|
||
@testable import OpenClaw
|
||
|
||
struct VoiceWakeRuntimeTests {
|
||
@Test func `trims after trigger keeps post speech`() {
|
||
let triggers = ["claude", "openclaw"]
|
||
let text = "hey Claude how are you"
|
||
#expect(VoiceWakeRuntime._testTrimmedAfterTrigger(text, triggers: triggers) == "how are you")
|
||
}
|
||
|
||
@Test func `trims after trigger returns original when no trigger`() {
|
||
let triggers = ["claude"]
|
||
let text = "good morning friend"
|
||
#expect(VoiceWakeRuntime._testTrimmedAfterTrigger(text, triggers: triggers) == text)
|
||
}
|
||
|
||
@Test func `trims after first matching trigger`() {
|
||
let triggers = ["buddy", "claude"]
|
||
let text = "hello buddy this is after trigger claude also here"
|
||
#expect(VoiceWakeRuntime
|
||
._testTrimmedAfterTrigger(text, triggers: triggers) == "this is after trigger claude also here")
|
||
}
|
||
|
||
@Test func `has content after trigger false when only trigger`() {
|
||
let triggers = ["openclaw"]
|
||
let text = "hey openclaw"
|
||
#expect(!VoiceWakeRuntime._testHasContentAfterTrigger(text, triggers: triggers))
|
||
}
|
||
|
||
@Test func `has content after trigger true when speech continues`() {
|
||
let triggers = ["claude"]
|
||
let text = "claude write a note"
|
||
#expect(VoiceWakeRuntime._testHasContentAfterTrigger(text, triggers: triggers))
|
||
}
|
||
|
||
@Test func `trims after chinese trigger keeps post speech`() {
|
||
let triggers = ["小爪", "openclaw"]
|
||
let text = "嘿 小爪 帮我打开设置"
|
||
#expect(VoiceWakeRuntime._testTrimmedAfterTrigger(text, triggers: triggers) == "帮我打开设置")
|
||
}
|
||
|
||
@Test func `trims after trigger handles width insensitive forms`() {
|
||
let triggers = ["openclaw"]
|
||
let text = "ＯｐｅｎＣｌａｗ 请帮我"
|
||
#expect(VoiceWakeRuntime._testTrimmedAfterTrigger(text, triggers: triggers) == "请帮我")
|
||
}
|
||
|
||
@Test func `gate requires gap between trigger and command`() {
|
||
let transcript = "hey openclaw do thing"
|
||
let segments = makeWakeWordSegments(
|
||
transcript: transcript,
|
||
words: [
|
||
("hey", 0.0, 0.1),
|
||
("openclaw", 0.2, 0.1),
|
||
("do", 0.35, 0.1),
|
||
("thing", 0.5, 0.1),
|
||
])
|
||
let config = WakeWordGateConfig(triggers: ["openclaw"], minPostTriggerGap: 0.3)
|
||
#expect(WakeWordGate.match(transcript: transcript, segments: segments, config: config) == nil)
|
||
}
|
||
|
||
@Test func `gate accepts gap and extracts command`() {
|
||
let transcript = "hey openclaw do thing"
|
||
let segments = makeWakeWordSegments(
|
||
transcript: transcript,
|
||
words: [
|
||
("hey", 0.0, 0.1),
|
||
("openclaw", 0.2, 0.1),
|
||
("do", 0.9, 0.1),
|
||
("thing", 1.1, 0.1),
|
||
])
|
||
let config = WakeWordGateConfig(triggers: ["openclaw"], minPostTriggerGap: 0.3)
|
||
#expect(WakeWordGate.match(transcript: transcript, segments: segments, config: config)?.command == "do thing")
|
||
}
|
||
|
||
@Test func `gate command text handles foreign string ranges`() {
|
||
let transcript = "hey openclaw do thing"
|
||
let other = "do thing"
|
||
let foreignRange = other.range(of: "do")
|
||
let segments = [
|
||
WakeWordSegment(text: "hey", start: 0.0, duration: 0.1, range: transcript.range(of: "hey")),
|
||
WakeWordSegment(text: "openclaw", start: 0.2, duration: 0.1, range: transcript.range(of: "openclaw")),
|
||
WakeWordSegment(text: "do", start: 0.9, duration: 0.1, range: foreignRange),
|
||
WakeWordSegment(text: "thing", start: 1.1, duration: 0.1, range: nil),
|
||
]
|
||
|
||
#expect(
|
||
WakeWordGate.commandText(
|
||
transcript: transcript,
|
||
segments: segments,
|
||
triggerEndTime: 0.3) == "do thing")
|
||
}
|
||
}
|