待ち行列への登録と状態表示を追加（新着からの登録、/queue、DATABASE_URL）

internal/web/store_test.go

@@ -0,0 +1,132 @@
+package web
+
+import (
+	"context"
+	"sort"
+	"time"
+
+	"gitea.kazu634.com/kazu634/tver/internal/store"
+	"github.com/jackc/pgx/v5"
+)
+
+type fakeJobStore struct {
+	jobs                      map[string]store.Job
+	nextID                    int64
+	lastEpisode               store.Episode
+	enqueueCalls, statusCalls int
+	statusIDs                 []string
+	listStatuses              []string
+	listLimits                []int
+	err                       error
+}
+
+func (f *fakeJobStore) ensure() {
+	if f.jobs == nil {
+		f.jobs = make(map[string]store.Job)
+	}
+}
+
+func (f *fakeJobStore) Enqueue(_ context.Context, ep store.Episode, source string) (store.Job, bool, error) {
+	f.enqueueCalls++
+	f.lastEpisode = ep
+	if f.err != nil {
+		return store.Job{}, false, f.err
+	}
+	f.ensure()
+	if job, ok := f.jobs[ep.ID]; ok {
+		return job, false, nil
+	}
+	f.nextID++
+	job := store.Job{ID: f.nextID, EpisodeID: ep.ID, SeriesTitle: ep.SeriesTitle,
+		Title: ep.Title, BroadcasterName: ep.BroadcasterName, Source: source,
+		Status: "queued", CreatedAt: time.Date(2026, 10, 3, 0, 0, 0, 0, time.UTC)}
+	f.jobs[ep.ID] = job
+	return job, true, nil
+}
+
+func (f *fakeJobStore) EpisodeStatuses(_ context.Context, ids []string) (map[string]string, error) {
+	f.statusCalls++
+	f.statusIDs = append([]string(nil), ids...)
+	if f.err != nil {
+		return nil, f.err
+	}
+	result := make(map[string]string)
+	for _, id := range ids {
+		if job, ok := f.jobs[id]; ok {
+			result[id] = job.Status
+		}
+	}
+	return result, nil
+}
+
+func (f *fakeJobStore) ListJobs(_ context.Context, status string, limit int) ([]store.Job, error) {
+	f.listStatuses = append(f.listStatuses, status)
+	f.listLimits = append(f.listLimits, limit)
+	if f.err != nil {
+		return nil, f.err
+	}
+	var result []store.Job
+	for _, job := range f.jobs {
+		if job.Status == status {
+			result = append(result, job)
+		}
+	}
+	sort.Slice(result, func(i, j int) bool { return result[i].ID > result[j].ID })
+	if len(result) > limit {
+		result = result[:limit]
+	}
+	return result, nil
+}
+
+func (f *fakeJobStore) GetJob(_ context.Context, id int64) (store.Job, error) {
+	if f.err != nil {
+		return store.Job{}, f.err
+	}
+	for _, job := range f.jobs {
+		if job.ID == id {
+			return job, nil
+		}
+	}
+	return store.Job{}, pgx.ErrNoRows
+}
+
+func (f *fakeJobStore) Retry(_ context.Context, id int64) error {
+	for ep, job := range f.jobs {
+		if job.ID == id && (job.Status == "failed" || job.Status == "canceled") {
+			job.Status = "queued"
+			f.jobs[ep] = job
+			return nil
+		}
+	}
+	return store.ErrInvalidTransition
+}
+
+func (f *fakeJobStore) RetryEpisode(_ context.Context, id string) error {
+	job, ok := f.jobs[id]
+	if !ok || (job.Status != "failed" && job.Status != "canceled") {
+		return store.ErrInvalidTransition
+	}
+	job.Status = "queued"
+	f.jobs[id] = job
+	return nil
+}
+
+func (f *fakeJobStore) Cancel(_ context.Context, id int64) error {
+	if f.err != nil {
+		return f.err
+	}
+	for ep, job := range f.jobs {
+		if job.ID == id && job.Status == "queued" {
+			job.Status = "canceled"
+			f.jobs[ep] = job
+			return nil
+		}
+	}
+	return store.ErrInvalidTransition
+}
+
+func (f *fakeJobStore) CancelQueued(ctx context.Context, id int64) error {
+	return f.Cancel(ctx, id)
+}
+
+var _ JobStore = (*fakeJobStore)(nil)