feat(gateway): 补齐桌面端高级媒体直连接口

新增图片矢量化、视频超分、每日用量、计价与任务隔离能力，并通过环境变量解析平台凭据。已通过 Go 全量门禁、迁移检查、镜像构建以及 Vectorizer 五格式和 Topaz 3 秒视频真实 DEV 验收。
2026-07-22 14:02:53 +08:00
parent cbebfd7baa
commit 3056cf8fca
37 changed files with 3336 additions and 29 deletions
@@ -0,0 +1,62 @@
+package httpapi
+
+import (
+	"testing"
+	"time"
+
+	"github.com/easyai/easyai-ai-gateway/apps/api/internal/auth"
+	"github.com/easyai/easyai-ai-gateway/apps/api/internal/store"
+)
+
+func TestAdvancedMediaScopeAliases(t *testing.T) {
+	tests := []struct {
+		kind  string
+		scope string
+	}{
+		{kind: "images.vectorize", scope: "image_vectorize"},
+		{kind: "images.vectorize", scope: "image"},
+		{kind: "images.vectorize", scope: "vectorize"},
+		{kind: "videos.upscales", scope: "video_enhance"},
+		{kind: "videos.upscales", scope: "video"},
+		{kind: "videos.upscales", scope: "video_upscale"},
+	}
+	for _, test := range tests {
+		user := &auth.User{APIKeyID: "key", APIKeyScopes: []string{test.scope}}
+		if !apiKeyScopeAllowed(user, test.kind) {
+			t.Fatalf("scope %q should allow %q", test.scope, test.kind)
+		}
+	}
+	if apiKeyScopeAllowed(&auth.User{APIKeyID: "key", APIKeyScopes: []string{"chat"}}, "videos.upscales") {
+		t.Fatal("chat scope must not allow video upscale")
+	}
+}
+
+func TestFillDailyTokenUsageDaysKeepsGapsAndStreaks(t *testing.T) {
+	location, err := time.LoadLocation("Asia/Shanghai")
+	if err != nil {
+		t.Fatal(err)
+	}
+	from := time.Date(2026, 7, 20, 0, 0, 0, 0, location)
+	to := time.Date(2026, 7, 24, 0, 0, 0, 0, location)
+	items, summary := fillDailyTokenUsageDays(map[string]store.DailyTokenUsage{
+		"2026-07-20": {Date: "2026-07-20", TotalTokens: 10, TaskCount: 1},
+		"2026-07-22": {Date: "2026-07-22", TotalTokens: 20, TaskCount: 1},
+		"2026-07-23": {Date: "2026-07-23", TotalTokens: 30, TaskCount: 2},
+		"2026-07-24": {Date: "2026-07-24", TotalTokens: 40, TaskCount: 1},
+	}, from, to)
+	if len(items) != 5 || items[1].Date != "2026-07-21" || items[1].TaskCount != 0 {
+		t.Fatalf("daily usage should contain continuous zero days: %+v", items)
+	}
+	if summary.CumulativeTokens != 100 || summary.PeakDailyTokens != 40 || summary.CurrentStreakDays != 3 || summary.LongestStreakDays != 3 {
+		t.Fatalf("unexpected daily usage summary: %+v", summary)
+	}
+}
+
+func TestAdvancedMediaDefaultModels(t *testing.T) {
+	if got := canonicalTaskModelName("images.vectorize", ""); got != "easy-image-vectorizer-1" {
+		t.Fatalf("vectorizer default model=%q", got)
+	}
+	if got := canonicalTaskModelName("videos.upscales", ""); got != "easy-proteus-standard-4" {
+		t.Fatalf("Topaz default model=%q", got)
+	}
+}