70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
package ctxmgr
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// TestSweepExpiredProcessingTurns_Memory 验证 T20-B5 孤儿批量扫描：
|
|
// 过租约的 processing 回合被标 failed；未过期/已终态不受影响；幂等可重复。
|
|
func TestSweepExpiredProcessingTurns_Memory(t *testing.T) {
|
|
mgr := NewContextManager(nil, nil, DefaultTimeDecayConfig())
|
|
ctx := context.Background()
|
|
|
|
// 直接在内存 map 构造三个状态：过期 processing / 新鲜 processing / 已完成
|
|
past := time.Now().Add(-1 * time.Minute)
|
|
future := time.Now().Add(5 * time.Minute)
|
|
mgr.mu.Lock()
|
|
mgr.turns[turnMemKey("sess-sweep", "turn-expired")] = &ChatTurnRecord{
|
|
SessionID: "sess-sweep",
|
|
ClientTurnID: "turn-expired",
|
|
Status: TurnStatusProcessing,
|
|
LeaseExpiresAt: &past,
|
|
}
|
|
mgr.turns[turnMemKey("sess-sweep", "turn-fresh")] = &ChatTurnRecord{
|
|
SessionID: "sess-sweep",
|
|
ClientTurnID: "turn-fresh",
|
|
Status: TurnStatusProcessing,
|
|
LeaseExpiresAt: &future,
|
|
}
|
|
done := time.Now()
|
|
mgr.turns[turnMemKey("sess-sweep", "turn-done")] = &ChatTurnRecord{
|
|
SessionID: "sess-sweep",
|
|
ClientTurnID: "turn-done",
|
|
Status: TurnStatusCompleted,
|
|
CompletedAt: &done,
|
|
}
|
|
mgr.mu.Unlock()
|
|
|
|
n, err := mgr.SweepExpiredProcessingTurns(ctx, 100)
|
|
if err != nil {
|
|
t.Fatalf("sweep: %v", err)
|
|
}
|
|
if n != 1 {
|
|
t.Fatalf("expected exactly 1 swept, got %d", n)
|
|
}
|
|
|
|
if rec, err := mgr.GetChatTurn(ctx, "sess-sweep", "turn-expired"); err != nil {
|
|
t.Fatalf("get after sweep: %v", err)
|
|
} else if rec.Status != TurnStatusFailed {
|
|
t.Fatalf("expired turn must be failed, got %s", rec.Status)
|
|
}
|
|
if rec, err := mgr.GetChatTurn(ctx, "sess-sweep", "turn-fresh"); err != nil {
|
|
t.Fatalf("get fresh: %v", err)
|
|
} else if rec.Status != TurnStatusProcessing {
|
|
t.Fatalf("fresh turn must stay processing, got %s", rec.Status)
|
|
}
|
|
if rec, err := mgr.GetChatTurn(ctx, "sess-sweep", "turn-done"); err != nil {
|
|
t.Fatalf("get done: %v", err)
|
|
} else if rec.Status != TurnStatusCompleted {
|
|
t.Fatalf("done turn must stay completed, got %s", rec.Status)
|
|
}
|
|
|
|
// 幂等：再扫一次无新变化
|
|
n2, _ := mgr.SweepExpiredProcessingTurns(ctx, 100)
|
|
if n2 != 0 {
|
|
t.Fatalf("second sweep expect 0, got %d", n2)
|
|
}
|
|
}
|