Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/repository/image_task_store_test.go

@@ -0,0 +1,43 @@
+package repository
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/Wei-Shaw/sub2api/internal/service"
+	"github.com/alicebob/miniredis/v2"
+	"github.com/redis/go-redis/v9"
+	"github.com/stretchr/testify/require"
+)
+
+func TestImageTaskStoreRoundTripAndTTL(t *testing.T) {
+	mr := miniredis.RunT(t)
+	rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
+	t.Cleanup(func() { _ = rdb.Close() })
+	store := NewImageTaskStore(rdb)
+	task := &service.ImageTaskRecord{
+		ID:        "imgtask_123",
+		UserID:    7,
+		APIKeyID:  9,
+		Status:    service.ImageTaskStatusProcessing,
+		CreatedAt: 100,
+		ExpiresAt: 200,
+	}
+
+	require.NoError(t, store.Save(context.Background(), task, 24*time.Hour))
+	got, err := store.Get(context.Background(), task.ID)
+	require.NoError(t, err)
+	require.Equal(t, task, got)
+	require.Equal(t, 24*time.Hour, mr.TTL(imageTaskKey(task.ID)))
+}
+
+func TestImageTaskStoreMissing(t *testing.T) {
+	mr := miniredis.RunT(t)
+	rdb := redis.NewClient(&redis.Options{Addr: mr.Addr()})
+	t.Cleanup(func() { _ = rdb.Close() })
+	store := NewImageTaskStore(rdb)
+
+	_, err := store.Get(context.Background(), "imgtask_missing")
+	require.ErrorIs(t, err, service.ErrImageTaskNotFound)
+}