64 lines
2.0 KiB
Go
64 lines
2.0 KiB
Go
package store_test
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"slices"
|
|
"testing"
|
|
"time"
|
|
|
|
"g.fc.run/misc/byre/internal/api"
|
|
"g.fc.run/misc/byre/internal/store"
|
|
)
|
|
|
|
func TestQueuedTurnKeepsImages(t *testing.T) {
|
|
ctx := context.Background()
|
|
st, err := store.Open(filepath.Join(t.TempDir(), "byre.db"), func(api.Event) {})
|
|
if err != nil {
|
|
t.Fatalf("open store: %v", err)
|
|
}
|
|
t.Cleanup(func() { st.Close() })
|
|
|
|
now := time.Now().UTC()
|
|
_, err = st.EnsureUser(ctx, "alice", "Alice")
|
|
if err != nil {
|
|
t.Fatalf("ensure user: %v", err)
|
|
}
|
|
err = st.CreateProject(ctx, api.Project{Id: "p1", Name: "demo", RepoPath: "/nonexistent", DefaultBranch: "main", Remote: "none", CreatedBy: "alice", CreatedAt: now})
|
|
if err != nil {
|
|
t.Fatalf("create project: %v", err)
|
|
}
|
|
err = st.CreateTask(ctx, api.Task{Id: "t1", ProjectId: "p1", Title: "t1", Branch: "t1", WorktreePath: "/nonexistent", BaseRef: "x", Status: api.TaskStatusActive, CreatedBy: "alice", CreatedAt: now})
|
|
if err != nil {
|
|
t.Fatalf("create task: %v", err)
|
|
}
|
|
err = st.CreateSession(ctx, api.AgentSession{Id: "s1", TaskId: "t1", Vendor: api.Claude, VendorSessionId: "s1", State: api.AgentSessionStateIdle, Role: api.Implementer, CreatedBy: "alice", CreatedAt: now}, "p1")
|
|
if err != nil {
|
|
t.Fatalf("create session: %v", err)
|
|
}
|
|
|
|
b, err := st.CreateBlob(ctx, "image/png", []byte("not really a png"), "alice")
|
|
if err != nil {
|
|
t.Fatalf("create blob: %v", err)
|
|
}
|
|
got, data, err := st.ReadBlob(ctx, b.Id)
|
|
if err != nil {
|
|
t.Fatalf("read blob: %v", err)
|
|
}
|
|
if got.ContentType != "image/png" || got.CreatedBy != "alice" || string(data) != "not really a png" {
|
|
t.Fatalf("blob round trip: %+v %q", got, data)
|
|
}
|
|
|
|
err = st.EnqueueTurn(ctx, store.QueuedTurn{SessionID: "s1", Actor: "alice", Text: "look", Images: []string{b.Id}, CreatedAt: now})
|
|
if err != nil {
|
|
t.Fatalf("enqueue: %v", err)
|
|
}
|
|
queued, err := st.QueuedTurns(ctx)
|
|
if err != nil {
|
|
t.Fatalf("queued turns: %v", err)
|
|
}
|
|
if len(queued) != 1 || !slices.Equal(queued[0].Images, []string{b.Id}) {
|
|
t.Fatalf("queued turn lost its images: %+v", queued)
|
|
}
|
|
}
|