wehub-resource-sync
26f897c1ec
release / release-please (push) Failing after 1m49s
docs / build (push) Failing after 6m34s
release / build-and-upload (arm64, linux) (push) Has been cancelled
release / build-and-upload (arm64, windows) (push) Has been cancelled
release / build-darwin (amd64, darwin) (push) Has been cancelled
release / checksums (push) Has been cancelled
release / finalize (push) Has been cancelled
release / build-darwin (arm64, darwin) (push) Has been cancelled
release / build-and-upload (amd64, linux) (push) Has been cancelled
release / build-and-upload (amd64, windows) (push) Has been cancelled
docs / deploy (push) Has been cancelled
109 lines
3.0 KiB
Go
109 lines
3.0 KiB
Go
package db
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestUpdateRunIntent_RoundTrip(t *testing.T) {
|
|
d := openTestDB(t)
|
|
repo, _ := d.InsertRepo("/home/user/intent", "git@github.com:user/intent.git", "main")
|
|
run, _ := d.InsertRun(repo.ID, "feature", "abc", "def")
|
|
|
|
got, _ := d.GetRun(run.ID)
|
|
if got.Intent != nil || got.IntentSource != nil || got.IntentSessionID != nil || got.IntentScore != nil {
|
|
t.Fatalf("expected nil intent fields on fresh run, got %+v", got)
|
|
}
|
|
|
|
if err := d.UpdateRunIntent(run.ID, RunIntent{
|
|
Summary: "user wanted to add foo",
|
|
Source: "claude",
|
|
SessionID: "abc-123",
|
|
Score: 0.85,
|
|
}); err != nil {
|
|
t.Fatalf("update intent: %v", err)
|
|
}
|
|
|
|
got, _ = d.GetRun(run.ID)
|
|
if got.Intent == nil || *got.Intent != "user wanted to add foo" {
|
|
t.Errorf("intent = %v, want %q", got.Intent, "user wanted to add foo")
|
|
}
|
|
if got.IntentSource == nil || *got.IntentSource != "claude" {
|
|
t.Errorf("intent source = %v, want claude", got.IntentSource)
|
|
}
|
|
if got.IntentSessionID == nil || *got.IntentSessionID != "abc-123" {
|
|
t.Errorf("intent session = %v, want abc-123", got.IntentSessionID)
|
|
}
|
|
if got.IntentScore == nil || *got.IntentScore != 0.85 {
|
|
t.Errorf("intent score = %v, want 0.85", got.IntentScore)
|
|
}
|
|
}
|
|
|
|
func TestIntentCache_PutGet(t *testing.T) {
|
|
d := openTestDB(t)
|
|
|
|
got, err := d.GetIntentCache("missing")
|
|
if err != nil {
|
|
t.Fatalf("get missing: %v", err)
|
|
}
|
|
if got != nil {
|
|
t.Fatalf("expected nil for missing key, got %+v", got)
|
|
}
|
|
|
|
if err := d.PutIntentCache(IntentCacheEntry{
|
|
CacheKey: "k1",
|
|
Summary: "do the thing",
|
|
AgentName: "claude",
|
|
SessionID: "sess-1",
|
|
}); err != nil {
|
|
t.Fatalf("put: %v", err)
|
|
}
|
|
|
|
got, err = d.GetIntentCache("k1")
|
|
if err != nil {
|
|
t.Fatalf("get: %v", err)
|
|
}
|
|
if got == nil || got.Summary != "do the thing" || got.AgentName != "claude" {
|
|
t.Fatalf("got %+v", got)
|
|
}
|
|
if got.CreatedAt == 0 {
|
|
t.Errorf("expected created_at populated")
|
|
}
|
|
|
|
// Replace.
|
|
if err := d.PutIntentCache(IntentCacheEntry{
|
|
CacheKey: "k1",
|
|
Summary: "do the new thing",
|
|
AgentName: "claude",
|
|
SessionID: "sess-1",
|
|
}); err != nil {
|
|
t.Fatalf("replace: %v", err)
|
|
}
|
|
got, _ = d.GetIntentCache("k1")
|
|
if got.Summary != "do the new thing" {
|
|
t.Errorf("want replaced summary, got %q", got.Summary)
|
|
}
|
|
}
|
|
|
|
func TestIntentCache_Cleanup(t *testing.T) {
|
|
d := openTestDB(t)
|
|
|
|
now := time.Now().Unix()
|
|
d.PutIntentCache(IntentCacheEntry{CacheKey: "old", Summary: "x", AgentName: "claude", SessionID: "s", CreatedAt: now - int64((40 * 24 * time.Hour).Seconds())})
|
|
d.PutIntentCache(IntentCacheEntry{CacheKey: "new", Summary: "x", AgentName: "claude", SessionID: "s", CreatedAt: now})
|
|
|
|
deleted, err := d.CleanupOldIntentCache(30 * 24 * time.Hour)
|
|
if err != nil {
|
|
t.Fatalf("cleanup: %v", err)
|
|
}
|
|
if deleted != 1 {
|
|
t.Errorf("deleted = %d, want 1", deleted)
|
|
}
|
|
if got, _ := d.GetIntentCache("old"); got != nil {
|
|
t.Errorf("expected old entry gone")
|
|
}
|
|
if got, _ := d.GetIntentCache("new"); got == nil {
|
|
t.Errorf("expected new entry retained")
|
|
}
|
|
}
|