byre/internal/store/ports_test.go

package store_test

import (
	"context"
	"path/filepath"
	"testing"
	"time"

	"g.fc.run/misc/byre/internal/api"
	"g.fc.run/misc/byre/internal/store"
)

func TestPortBlocksStartAtPortFirst(t *testing.T) {
	ctx := context.Background()
	st, err := store.Open(filepath.Join(t.TempDir(), "byre.db"), 51000, 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)
	}

	for i, want := range []int{51000, 51010} {
		id := []string{"t1", "t2"}[i]
		err = st.CreateTask(ctx, api.Task{Id: id, ProjectId: "p1", Title: id, Branch: id, WorktreePath: "/nonexistent", BaseRef: "x", Status: api.TaskStatusActive, CreatedBy: "alice", CreatedAt: now}, "alice")
		if err != nil {
			t.Fatalf("create task %s: %v", id, err)
		}
		task, err := st.GetTask(ctx, id)
		if err != nil {
			t.Fatalf("get task %s: %v", id, err)
		}
		if task.PortBase != want {
			t.Fatalf("task %s port_base = %d, want %d", id, task.PortBase, want)
		}
	}

	_, err = st.DiscardTask(ctx, "t1", "alice")
	if err != nil {
		t.Fatalf("discard: %v", err)
	}
	err = st.CreateTask(ctx, api.Task{Id: "t3", ProjectId: "p1", Title: "t3", Branch: "t3", WorktreePath: "/nonexistent", BaseRef: "x", Status: api.TaskStatusActive, CreatedBy: "alice", CreatedAt: now}, "alice")
	if err != nil {
		t.Fatalf("create task t3: %v", err)
	}
	reused, err := st.GetTask(ctx, "t3")
	if err != nil || reused.PortBase != 51000 {
		t.Fatalf("t3 port_base = %d, want the freed 51000: %v", reused.PortBase, err)
	}
}