91 lines
2.8 KiB
Go
91 lines
2.8 KiB
Go
package store_test
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"path/filepath"
|
|
"slices"
|
|
"testing"
|
|
"time"
|
|
|
|
"g.fc.run/misc/byre/internal/api"
|
|
"g.fc.run/misc/byre/internal/store"
|
|
)
|
|
|
|
func TestSetPreviewReplacesNameAndPort(t *testing.T) {
|
|
ctx := context.Background()
|
|
events := []api.Event{}
|
|
st, err := store.Open(filepath.Join(t.TempDir(), "byre.db"), store.DefaultPortFirst, func(ev api.Event) { events = append(events, ev) })
|
|
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}, "alice")
|
|
if err != nil {
|
|
t.Fatalf("create task: %v", err)
|
|
}
|
|
|
|
set := func(name string, port int) []int {
|
|
t.Helper()
|
|
dropped, err := st.SetPreview(ctx, api.Preview{TaskId: "t1", Name: name, Port: port, CreatedAt: now}, "agent")
|
|
if err != nil {
|
|
t.Fatalf("set preview %s:%d: %v", name, port, err)
|
|
}
|
|
return dropped
|
|
}
|
|
list := func() []string {
|
|
t.Helper()
|
|
previews, err := st.ListPreviews(ctx, "t1")
|
|
if err != nil {
|
|
t.Fatalf("list previews: %v", err)
|
|
}
|
|
out := []string{}
|
|
for _, p := range previews {
|
|
out = append(out, fmt.Sprintf("%s:%d", p.Name, p.Port))
|
|
}
|
|
return out
|
|
}
|
|
|
|
if d := set("web", 41001); len(d) != 0 {
|
|
t.Fatalf("first preview dropped %v", d)
|
|
}
|
|
if d := set("api", 41002); len(d) != 0 {
|
|
t.Fatalf("second preview dropped %v", d)
|
|
}
|
|
if d := set("web", 41001); len(d) != 0 {
|
|
t.Fatalf("re-exposing the same name and port dropped %v", d)
|
|
}
|
|
if d := set("web", 41003); !slices.Equal(d, []int{41001}) {
|
|
t.Fatalf("moving web to 41003 dropped %v, want [41001]", d)
|
|
}
|
|
if d := set("storybook", 41002); len(d) != 0 {
|
|
t.Fatalf("renaming 41002 dropped %v, want none", d)
|
|
}
|
|
if got, want := list(), []string{"storybook:41002", "web:41003"}; !slices.Equal(got, want) {
|
|
t.Fatalf("previews = %v, want %v", got, want)
|
|
}
|
|
if last := events[len(events)-1]; last.Kind != "preview.exposed" || last.Payload["name"] != "storybook" {
|
|
t.Fatalf("last event = %s %v", last.Kind, last.Payload)
|
|
}
|
|
|
|
_, err = st.DiscardTask(ctx, "t1", "alice")
|
|
if err != nil {
|
|
t.Fatalf("discard: %v", err)
|
|
}
|
|
_, err = st.SetPreview(ctx, api.Preview{TaskId: "t1", Name: "late", Port: 41004, CreatedAt: now}, "agent")
|
|
if !errors.Is(err, store.ErrNotFound) {
|
|
t.Fatalf("set preview on a closed task: %v, want ErrNotFound", err)
|
|
}
|
|
}
|