131 lines
5.0 KiB
Swift
131 lines
5.0 KiB
Swift
import Testing
|
||
@testable import 康康
|
||
|
||
/// 日记追问对话的纯逻辑：轮数计数、收尾判定、离线兜底草稿。
|
||
/// DiaryChatService 是 @MainActor struct,测试整体标 @MainActor 规避隔离。
|
||
@MainActor
|
||
struct DiaryChatServiceTests {
|
||
|
||
// MARK: roundsUsed —— 开场白不算追问(assistant 轮数 − 1),user 轮不计
|
||
|
||
@Test func roundsUsedIsZeroForEmptyTurns() {
|
||
#expect(DiaryChatService.roundsUsed(in: []) == 0)
|
||
}
|
||
|
||
@Test func roundsUsedIsZeroForOpeningOnly() {
|
||
// 仅 1 条 assistant(开场白)→ max(0, 1-1) = 0
|
||
let turns: [HealthExportDialogueTurn] = [.assistant("说说你今天的情况")]
|
||
#expect(DiaryChatService.roundsUsed(in: turns) == 0)
|
||
}
|
||
|
||
@Test func roundsUsedCountsAssistantRepliesAfterOpening() {
|
||
// 开场 + 2 条追问 = 3 条 assistant → max(0, 3-1) = 2
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("开场"),
|
||
.user("我头疼"),
|
||
.assistant("追问一"),
|
||
.user("还发烧"),
|
||
.assistant("追问二")
|
||
]
|
||
#expect(DiaryChatService.roundsUsed(in: turns) == 2)
|
||
}
|
||
|
||
@Test func roundsUsedIgnoresUserTurns() {
|
||
// 夹杂多条 user 不影响计数:仍是 3 条 assistant → 2
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("开场"),
|
||
.user("u1"), .user("u2"),
|
||
.assistant("追问一"),
|
||
.user("u3"), .user("u4"), .user("u5"),
|
||
.assistant("追问二"),
|
||
.user("u6")
|
||
]
|
||
#expect(DiaryChatService.roundsUsed(in: turns) == 2)
|
||
}
|
||
|
||
// MARK: isWrappedUp —— roundsUsed >= maxRounds(4)
|
||
|
||
@Test func isNotWrappedUpAtThreeRounds() {
|
||
// 4 条 assistant(开场 + 3 追问)→ roundsUsed = 3 < 4 → 未收尾
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("开场"),
|
||
.user("a"), .assistant("q1"),
|
||
.user("b"), .assistant("q2"),
|
||
.user("c"), .assistant("q3")
|
||
]
|
||
#expect(DiaryChatService.roundsUsed(in: turns) == 3)
|
||
#expect(DiaryChatService.isWrappedUp(turns: turns) == false)
|
||
}
|
||
|
||
@Test func isWrappedUpAtFourRounds() {
|
||
// 5 条 assistant(开场 + 4 追问)→ roundsUsed = 4 = maxRounds → 收尾
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("开场"),
|
||
.user("a"), .assistant("q1"),
|
||
.user("b"), .assistant("q2"),
|
||
.user("c"), .assistant("q3"),
|
||
.user("d"), .assistant("q4")
|
||
]
|
||
#expect(DiaryChatService.roundsUsed(in: turns) == DiaryChatPrompts.maxRounds)
|
||
#expect(DiaryChatService.isWrappedUp(turns: turns) == true)
|
||
}
|
||
|
||
// MARK: fallbackDraft —— 只拼 user 原话,assistant 绝不进草稿
|
||
|
||
@Test func fallbackDraftIsEmptyWhenNoUserTurns() {
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("开场"),
|
||
.assistant("再问一句")
|
||
]
|
||
#expect(DiaryChatService.fallbackDraft(turns: turns) == "")
|
||
}
|
||
|
||
@Test func fallbackDraftReturnsSingleUserVerbatimWithoutBullet() {
|
||
// 恰 1 条 user → 原话本身,无「· 」前缀;数值原样
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("说说你今天的情况"),
|
||
.user("今天头疼,量了血压 140/90")
|
||
]
|
||
let draft = DiaryChatService.fallbackDraft(turns: turns)
|
||
#expect(draft == "今天头疼,量了血压 140/90")
|
||
#expect(!draft.contains("· "))
|
||
}
|
||
|
||
@Test func fallbackDraftBulletsMultipleUserTurns() {
|
||
// ≥2 条 user → 每条前缀「· 」按换行拼接
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("开场"),
|
||
.user("今天头疼"),
|
||
.assistant("量血压了吗?"),
|
||
.user("量了血压 140/90")
|
||
]
|
||
let draft = DiaryChatService.fallbackDraft(turns: turns)
|
||
#expect(draft == "· 今天头疼\n· 量了血压 140/90")
|
||
}
|
||
|
||
@Test func fallbackDraftDropsBlankUserTurns() {
|
||
// 纯空格 / 换行空白的 user 轮被丢弃,剩两条有效仍带「· 」
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.user("第一条有效"),
|
||
.user(" "),
|
||
.user("\n \n"),
|
||
.user("第二条有效")
|
||
]
|
||
let draft = DiaryChatService.fallbackDraft(turns: turns)
|
||
#expect(draft == "· 第一条有效\n· 第二条有效")
|
||
}
|
||
|
||
@Test func fallbackDraftNeverIncludesAssistantTextAndKeepsNumbers() {
|
||
// assistant 文本绝不出现;140/90 原样保留
|
||
let turns: [HealthExportDialogueTurn] = [
|
||
.assistant("上次你记过头疼"),
|
||
.user("今天又头疼了,血压 140/90"),
|
||
.assistant("需要提醒复诊吗?")
|
||
]
|
||
let draft = DiaryChatService.fallbackDraft(turns: turns)
|
||
#expect(!draft.contains("上次你记过头疼"))
|
||
#expect(!draft.contains("需要提醒复诊吗?"))
|
||
#expect(draft.contains("140/90"))
|
||
}
|
||
}
|