65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
|
package memory
|
||
|
|
|
||
|
|
import (
|
||
|
|
"path/filepath"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCapabilityGapStoreAndLoad(t *testing.T) {
|
||
|
|
dbPath := filepath.Join(t.TempDir(), "test.db")
|
||
|
|
store, err := NewSQLiteStore(dbPath, nil)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("NewSQLiteStore error: %v", err)
|
||
|
|
}
|
||
|
|
defer store.Close()
|
||
|
|
|
||
|
|
if err := store.SaveCapabilityGap("c1", "u1", "intent-a", "reason-a"); err != nil {
|
||
|
|
t.Fatalf("SaveCapabilityGap error: %v", err)
|
||
|
|
}
|
||
|
|
if err := store.SaveCapabilityGap("c1", "u1", "intent-b", "reason-b"); err != nil {
|
||
|
|
t.Fatalf("SaveCapabilityGap error: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
items, err := store.TopCapabilityGaps(10)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("TopCapabilityGaps error: %v", err)
|
||
|
|
}
|
||
|
|
if len(items) != 2 {
|
||
|
|
t.Fatalf("expected 2 items, got %d", len(items))
|
||
|
|
}
|
||
|
|
if items[0].Intent != "intent-b" {
|
||
|
|
t.Fatalf("expected newest first, got first intent=%s", items[0].Intent)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTopCapabilityGapClusters(t *testing.T) {
|
||
|
|
dbPath := filepath.Join(t.TempDir(), "cluster.db")
|
||
|
|
store, err := NewSQLiteStore(dbPath, nil)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("NewSQLiteStore error: %v", err)
|
||
|
|
}
|
||
|
|
defer store.Close()
|
||
|
|
|
||
|
|
if err := store.SaveCapabilityGap("c1", "u1", "帮我查询 data 目录", "no_skill_matched"); err != nil {
|
||
|
|
t.Fatalf("SaveCapabilityGap error: %v", err)
|
||
|
|
}
|
||
|
|
if err := store.SaveCapabilityGap("c1", "u2", "帮我 查询 data 目录", "no_skill_matched"); err != nil {
|
||
|
|
t.Fatalf("SaveCapabilityGap error: %v", err)
|
||
|
|
}
|
||
|
|
if err := store.SaveCapabilityGap("c2", "u3", "读取配置文件内容", "tool_call_failed:file"); err != nil {
|
||
|
|
t.Fatalf("SaveCapabilityGap error: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
clusters, err := store.TopCapabilityGapClusters(10, time.Now().UTC().Add(-1*time.Hour))
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("TopCapabilityGapClusters error: %v", err)
|
||
|
|
}
|
||
|
|
if len(clusters) == 0 {
|
||
|
|
t.Fatalf("expected non-empty clusters")
|
||
|
|
}
|
||
|
|
if clusters[0].Count < 2 {
|
||
|
|
t.Fatalf("expected first cluster count >= 2, got %d", clusters[0].Count)
|
||
|
|
}
|
||
|
|
}
|